1/*	$NetBSD: tinytest_macros.h,v 1.6 2021/04/10 19:18:45 rillig Exp $	*/
2
3/* tinytest_macros.h -- Copyright 2009-2012 Nick Mathewson
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 *    derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28#ifndef TINYTEST_MACROS_H_INCLUDED_
29#define TINYTEST_MACROS_H_INCLUDED_
30
31/* Helpers for defining statement-like macros */
32#define TT_STMT_BEGIN do {
33#define TT_STMT_END } while (0)
34
35/* Redefine this if your test functions want to abort with something besides
36 * "goto end;" */
37#ifndef TT_EXIT_TEST_FUNCTION
38#define TT_EXIT_TEST_FUNCTION TT_STMT_BEGIN goto end; TT_STMT_END
39#endif
40
41/* Redefine this if you want to note success/failure in some different way. */
42#ifndef TT_DECLARE
43#define TT_DECLARE(prefix, args)				\
44	TT_STMT_BEGIN						\
45	printf("\n  %s %s:%d: ",prefix,__FILE__,__LINE__);	\
46	printf args ;						\
47	TT_STMT_END
48#endif
49
50/* Announce a failure. Args are parenthesized printf args. */
51#define TT_GRIPE(args) TT_DECLARE("FAIL", args)
52
53/* Announce a non-failure if we're verbose. */
54#define TT_BLATHER(args)						\
55	TT_STMT_BEGIN							\
56	if (tinytest_get_verbosity_()>1) TT_DECLARE("  OK", args);	\
57	TT_STMT_END
58
59#define TT_DIE(args)						\
60	TT_STMT_BEGIN						\
61	tinytest_set_test_failed_();				\
62	TT_GRIPE(args);						\
63	TT_EXIT_TEST_FUNCTION;					\
64	TT_STMT_END
65
66#define TT_FAIL(args)				\
67	TT_STMT_BEGIN						\
68	tinytest_set_test_failed_();				\
69	TT_GRIPE(args);						\
70	TT_STMT_END
71
72/* Fail and abort the current test for the reason in msg */
73#define tt_abort_printf(msg) TT_DIE(msg)
74#define tt_abort_perror(op) TT_DIE(("%s: %s [%d]",(op),strerror(errno), errno))
75#define tt_abort_msg(msg) TT_DIE(("%s", msg))
76#define tt_abort() TT_DIE(("%s", "(Failed.)"))
77
78/* Fail but do not abort the current test for the reason in msg. */
79#define tt_failprint_f(msg) TT_FAIL(msg)
80#define tt_fail_perror(op) TT_FAIL(("%s: %s [%d]",(op),strerror(errno), errno))
81#define tt_fail_msg(msg) TT_FAIL(("%s", msg))
82#define tt_fail() TT_FAIL(("%s", "(Failed.)"))
83
84/* End the current test, and indicate we are skipping it. */
85#define tt_skip()						\
86	TT_STMT_BEGIN						\
87	tinytest_set_test_skipped_();				\
88	TT_EXIT_TEST_FUNCTION;					\
89	TT_STMT_END
90
91#define tt_want_(b, msg, fail)				\
92	TT_STMT_BEGIN					\
93	if (!(b)) {					\
94		tinytest_set_test_failed_();		\
95		TT_GRIPE(("%s",msg));			\
96		fail;					\
97	} else {					\
98		TT_BLATHER(("%s",msg));			\
99	}						\
100	TT_STMT_END
101
102/* Assert b, but do not stop the test if b fails.  Log msg on failure. */
103#define tt_want_msg(b, msg)			\
104	tt_want_(b, msg, );
105
106/* Assert b and stop the test if b fails.  Log msg on failure. */
107#define tt_assert_msg(b, msg)			\
108	tt_want_(b, msg, TT_EXIT_TEST_FUNCTION);
109
110/* Assert b, but do not stop the test if b fails. */
111#define tt_want(b)   tt_want_msg( (b), "want("#b")")
112/* Assert b, and stop the test if b fails. */
113#define tt_assert(b) tt_assert_msg((b), "assert("#b")")
114
115#define tt_assert_test_fmt_type(a,b,str_test,type,test,printf_type,printf_fmt, \
116    setup_block,cleanup_block,die_on_fail)				\
117	TT_STMT_BEGIN							\
118	type val1_ = (type)(a);						\
119	type val2_ = (type)(b);						\
120	int tt_status_ = (test);					\
121	if (!tt_status_ || tinytest_get_verbosity_()>1)	{		\
122		printf_type print_;					\
123		printf_type print1_;					\
124		printf_type print2_;					\
125		type value_ = val1_;					\
126		setup_block;						\
127		print1_ = print_;					\
128		value_ = val2_;						\
129		setup_block;						\
130		print2_ = print_;					\
131		TT_DECLARE(tt_status_?"	 OK":"FAIL",			\
132			   ("assert(%s): "printf_fmt" vs "printf_fmt,	\
133			    str_test, print1_, print2_));		\
134		print_ = print1_;					\
135		cleanup_block;						\
136		print_ = print2_;					\
137		cleanup_block;						\
138		if (!tt_status_) {					\
139			tinytest_set_test_failed_();			\
140			die_on_fail ;					\
141		}							\
142	}								\
143	TT_STMT_END
144
145#define tt_assert_test_type(a,b,str_test,type,test,fmt,die_on_fail)	\
146	tt_assert_test_fmt_type(a,b,str_test,type,test,type,fmt,	\
147	    {print_=value_;},{},die_on_fail)
148
149#define tt_assert_test_type_opt(a,b,str_test,type,test,fmt,die_on_fail)	\
150	tt_assert_test_fmt_type(a,b,str_test,type,test,type,fmt,	\
151            {print_=value_?value_:"<NULL>";},{},die_on_fail)
152
153/* Helper: assert that a op b, when cast to type.  Format the values with
154 * printf format fmt on failure. */
155#define tt_assert_op_type(a,op,b,type,fmt)				\
156	tt_assert_test_type(a,b,#a" "#op" "#b,type,(val1_ op val2_),fmt, \
157	    TT_EXIT_TEST_FUNCTION)
158
159#define tt_int_op(a,op,b)			\
160	tt_assert_test_type(a,b,#a" "#op" "#b,long,(val1_ op val2_), \
161	    "%ld",TT_EXIT_TEST_FUNCTION)
162
163/** To compare SOCKET(windows)/fd */
164#define tt_fd_op(a,op,b) do {                                          \
165	int _a = (int)(a);                                             \
166	int _b = (int)(b);                                             \
167	tt_assert_test_type(_a,_b,#a" "#op" "#b,long,(val1_ op val2_), \
168	    "%ld",TT_EXIT_TEST_FUNCTION);                              \
169} while (0)
170
171#define tt_uint_op(a,op,b)						\
172	tt_assert_test_type(a,b,#a" "#op" "#b,unsigned long,		\
173	    (val1_ op val2_),"%lu",TT_EXIT_TEST_FUNCTION)
174
175#define tt_ptr_op(a,op,b)						\
176	tt_assert_test_type(a,b,#a" "#op" "#b,const void*,              \
177	    (val1_ op val2_),"%p",TT_EXIT_TEST_FUNCTION)
178
179/** XXX: have some issues with printing this non-NUL terminated strings */
180#define tt_nstr_op(n,a,op,b)						\
181	tt_assert_test_type_opt(a,b,#a" "#op" "#b,const char *,		\
182	    (val1_ && val2_ && strncmp(val1_,val2_,(n)) op 0),"<%s>",	\
183	    TT_EXIT_TEST_FUNCTION)
184
185#define tt_str_op(a,op,b)						\
186	tt_assert_test_type_opt(a,b,#a" "#op" "#b,const char *,		\
187	    (val1_ && val2_ && strcmp(val1_,val2_) op 0),"<%s>",	\
188	    TT_EXIT_TEST_FUNCTION)
189
190#define tt_mem_op(expr1, op, expr2, len)                                \
191  tt_assert_test_fmt_type(expr1,expr2,#expr1" "#op" "#expr2,            \
192			  const void *,                                 \
193			  (val1_ && val2_ && memcmp(val1_, val2_, len) op 0), \
194			  char *, "%s",					\
195			  { print_ = tinytest_format_hex_(value_, (len)); }, \
196			  { if (print_) free(print_); },		\
197			  TT_EXIT_TEST_FUNCTION				\
198                          );
199
200#define tt_want_int_op(a,op,b)						\
201	tt_assert_test_type(a,b,#a" "#op" "#b,long,(val1_ op val2_),"%ld",(void)0)
202
203#define tt_want_uint_op(a,op,b)						\
204	tt_assert_test_type(a,b,#a" "#op" "#b,unsigned long,		\
205	    (val1_ op val2_),"%lu",(void)0)
206
207#define tt_want_ptr_op(a,op,b)						\
208  tt_assert_test_type(a,b,#a" "#op" "#b,const void*,			\
209	    (val1_ op val2_),"%p",(void)0)
210
211#define tt_want_str_op(a,op,b)						\
212	tt_assert_test_type(a,b,#a" "#op" "#b,const char *,		\
213	    (strcmp(val1_,val2_) op 0),"<%s>",(void)0)
214
215#endif
216