1 /* SPDX-License-Identifier: GPL-2.0-only */
2
3 #include <commonlib/bsd/helpers.h>
4 #include <tests/test.h>
5
func(void)6 static void func(void)
7 {
8 function_called();
9 }
10
test_genmask(void ** state)11 static void test_genmask(void **state)
12 {
13 assert_int_equal(GENMASK(4, 4), 0x10);
14 assert_int_equal(GENMASK(4, 3), 0x18);
15 assert_int_equal(GENMASK(4, 0), 0x1f);
16 /* Edge cases */
17 assert_int_equal(GENMASK(0, 0), 1);
18 assert_int_equal(GENMASK(31, 31), 0x80000000);
19 assert_int_equal(GENMASK(31, 0), 0xffffffff);
20 assert_int_equal(GENMASK(63, 63), 0x8000000000000000);
21 assert_int_equal(GENMASK(63, 0), 0xffffffffffffffff);
22 }
23
test_retry(void ** state)24 static void test_retry(void **state)
25 {
26 int count;
27
28 /* 2-argument form */
29 count = 0;
30 assert_true(retry(3, ++count == 1));
31 count = 0;
32 assert_true(retry(3, ++count == 3));
33 count = 0;
34 assert_false(retry(3, ++count == 4));
35
36 /* 3-argument form */
37 expect_function_calls(func, 9);
38 assert_null(retry(10, NULL, func()));
39
40 assert_int_equal(retry(10, 999, func()), 999);
41
42 count = 0;
43 expect_function_calls(func, 3);
44 assert_true(retry(10, ++count == 4, func()));
45 }
46
main(void)47 int main(void)
48 {
49 const struct CMUnitTest tests[] = {
50 cmocka_unit_test(test_genmask),
51 cmocka_unit_test(test_retry),
52 };
53
54 return cb_run_group_tests(tests, NULL, NULL);
55 }
56