/* SPDX-License-Identifier: GPL-2.0-only */ #include <commonlib/bsd/helpers.h> #include <tests/test.h> static void func(void) { function_called(); } static void test_retry(void **state) { int count; /* 2-argument form */ count = 0; assert_true(retry(3, ++count == 1)); count = 0; assert_true(retry(3, ++count == 3)); count = 0; assert_false(retry(3, ++count == 4)); /* 3-argument form */ expect_function_calls(func, 9); assert_null(retry(10, NULL, func())); assert_int_equal(retry(10, 999, func()), 999); count = 0; expect_function_calls(func, 3); assert_true(retry(10, ++count == 4, func())); } int main(void) { const struct CMUnitTest tests[] = { cmocka_unit_test(test_retry), }; return cmocka_run_group_tests(tests, NULL, NULL); }