1 #include <stdarg.h>
2 #include <stddef.h>
3 #include <setjmp.h>
4 #include <assert.h>
5 #include <cmocka.h>
6 /* cmocka < 1.0 didn't support these features we need */
7 #ifndef assert_ptr_equal
8 #define assert_ptr_equal(a, b) \
9 _assert_int_equal(cast_ptr_to_largest_integral_type(a), \
10 cast_ptr_to_largest_integral_type(b), \
11 __FILE__, __LINE__)
12 #define CMUnitTest UnitTest
13 #define cmocka_unit_test unit_test
14 #define cmocka_run_group_tests(t, setup, teardown) run_tests(t)
15 #endif
16
17
18 extern void mock_assert(const int result, const char* const expression,
19 const char * const file, const int line);
20 #undef assert
21 #define assert(expression) \
22 mock_assert((int)(expression), #expression, __FILE__, __LINE__);
23
24 #include "list.h"
25
26 /* remap exit -> assert, then use cmocka's mock_assert
27 (compile with `--wrap=exit`) */
28 extern void exit(int status);
29 extern void __real_exit(int status);
30 //void __wrap_exit(int status);
__wrap_exit(int status)31 void __wrap_exit(int status) {
32 (void)status;
33 assert(0);
34 }
35
36 /* ignore all printfs */
37 #undef printf
38 extern int printf(const char *format, ...);
39 //extern int __real_printf(const char *format, ...);
40 int __wrap_printf(const char *format, ...);
__wrap_printf(const char * format,...)41 int __wrap_printf(const char *format, ...) {
42 (void)format;
43 return 1;
44 }
45
46 static list_t testlist = {.element_prealloc_count = 0};
47
test_contains(void ** state)48 static void test_contains(void **state) {
49 (void)state;
50
51 u32 one = 1;
52 u32 two = 2;
53
54 list_append(&testlist, &one);
55 assert_true(list_contains(&testlist, &one));
56 assert_false(list_contains(&testlist, &two));
57 list_remove(&testlist, &one);
58 assert_false(list_contains(&testlist, &one));
59 }
60
test_foreach(void ** state)61 static void test_foreach(void **state) {
62 (void)state;
63
64 u32 one = 1;
65 u32 two = 2;
66 u32 result = 0;
67
68 list_append(&testlist, &one);
69 list_append(&testlist, &two);
70 list_append(&testlist, &one);
71
72 /* The list is for pointers, so int doesn't work as type directly */
73 LIST_FOREACH(&testlist, u32, {
74 result += *el;
75 });
76
77 assert_int_equal(result, 4);
78
79 }
80
test_long_list(void ** state)81 static void test_long_list(void **state) {
82 (void)state;
83
84 u32 result1 = 0;
85 u32 result2 = 0;
86 u32 i;
87
88 u32 vals[100];
89
90 for (i = 0; i < 100; i++) {
91 vals[i] = i;
92 }
93
94 LIST_FOREACH_CLEAR(&testlist, void, {});
95 for (i = 0; i < 100; i++) {
96 list_append(&testlist, &vals[i]);
97 }
98 LIST_FOREACH(&testlist, u32, {
99 result1 += *el;
100 });
101 //printf("removing %d\n", vals[50]);
102 list_remove(&testlist, &vals[50]);
103
104 LIST_FOREACH(&testlist, u32, {
105 // printf("var: %d\n", *el);
106 result2 += *el;
107 });
108 assert_int_not_equal(result1, result2);
109 assert_int_equal(result1, result2 + 50);
110
111 result1 = 0;
112 LIST_FOREACH_CLEAR(&testlist, u32, {
113 result1 += *el;
114 });
115 assert_int_equal(result1, result2);
116
117 result1 = 0;
118 LIST_FOREACH(&testlist, u32, {
119 result1 += *el;
120 });
121 assert_int_equal(result1, 0);
122
123 }
124
main(int argc,char ** argv)125 int main(int argc, char **argv) {
126 (void)argc;
127 (void)argv;
128
129 const struct CMUnitTest tests[] = {
130 cmocka_unit_test(test_contains),
131 cmocka_unit_test(test_foreach),
132 cmocka_unit_test(test_long_list),
133 };
134
135 //return cmocka_run_group_tests (tests, setup, teardown);
136 __real_exit( cmocka_run_group_tests (tests, NULL, NULL) );
137
138 // fake return for dumb compilers
139 return 0;
140 }
141