42 lines
764 B
C
42 lines
764 B
C
/* 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);
|
|
}
|