summaryrefslogtreecommitdiff
path: root/tests/commonlib/bsd/helpers-test.c
blob: 324c901b127ff17e3a4d86801988eab29daee701 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
/* 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_genmask(void **state)
{
	assert_int_equal(GENMASK(4, 4), 0x10);
	assert_int_equal(GENMASK(4, 3), 0x18);
	assert_int_equal(GENMASK(4, 0), 0x1f);
	/* Edge cases */
	assert_int_equal(GENMASK(0, 0), 1);
	assert_int_equal(GENMASK(31, 31), 0x80000000);
	assert_int_equal(GENMASK(31, 0), 0xffffffff);
	assert_int_equal(GENMASK(63, 63), 0x8000000000000000);
	assert_int_equal(GENMASK(63, 0), 0xffffffffffffffff);
}

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_genmask),
		cmocka_unit_test(test_retry),
	};

	return cmocka_run_group_tests(tests, NULL, NULL);
}