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