tests: Add tests for assert_uint_in_set()

3 files changed, 33 insertions, 2 deletions

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 83f1b79..1d1649e 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -97,7 +97,7 @@ set_tests_properties(
     test_assert_set_fail
         PROPERTIES
         PASS_REGULAR_EXPRESSION
-        "\\[  FAILED  \\] set_fail_tests: 2 test"
+        "\\[  FAILED  \\] set_fail_tests: 3 test"
 )
 
 # test_expect_check_fail
diff --git a/tests/test_assert_set.c b/tests/test_assert_set.c
index 7f27a01..43d28f4 100644
--- a/tests/test_assert_set.c
+++ b/tests/test_assert_set.c
@@ -72,6 +72,26 @@ static void test_assert_int_not_in_set_64(void **state)
     assert_int_not_in_set(INT64_MAX, set, ARRAY_SIZE(set));
 }
 
+static void test_assert_uint_in_set_32(void **state)
+{
+    uint32_t set[] = {0, 1, 2, 3, UINT32_MAX};
+
+    (void)state; /* unused */
+
+    assert_int_in_set(0, set, ARRAY_SIZE(set));
+    assert_int_in_set(1, set, ARRAY_SIZE(set));
+    assert_int_in_set(UINT32_MAX, set, ARRAY_SIZE(set));
+}
+
+static void test_assert_uint_in_set_64(void **state)
+{
+    uint64_t set[] = {1, 2, 3, UINT64_MAX};
+
+    (void)state; /* unused */
+
+    assert_int_in_set(UINT64_MAX, set, ARRAY_SIZE(set));
+}
+
 int main(void) {
     const struct CMUnitTest set_tests[] = {
         cmocka_unit_test(test_assert_int_in_set_8),
@@ -80,6 +100,8 @@ int main(void) {
         cmocka_unit_test(test_assert_int_in_set_64),
         cmocka_unit_test(test_assert_int_not_in_set_32),
         cmocka_unit_test(test_assert_int_not_in_set_64),
+        cmocka_unit_test(test_assert_uint_in_set_32),
+        cmocka_unit_test(test_assert_uint_in_set_64),
     };
 
     return cmocka_run_group_tests(set_tests, NULL, NULL);
diff --git a/tests/test_assert_set_fail.c b/tests/test_assert_set_fail.c
index e10f333..1f8a03a 100644
--- a/tests/test_assert_set_fail.c
+++ b/tests/test_assert_set_fail.c
@@ -25,12 +25,21 @@ static void test_assert_int_not_in_set_fail(void **state)
     assert_int_not_in_set(3, set, ARRAY_SIZE(set));
 }
 
+static void test_assert_uint_in_set_fail(void **state)
+{
+    uint32_t set[] = {1, 2, 3, UINT32_MAX};
+
+    (void)state; /* unused */
+
+    assert_int_in_set(0, set, ARRAY_SIZE(set));
+}
+
 int main(void) {
     const struct CMUnitTest set_fail_tests[] = {
         cmocka_unit_test(test_assert_int_in_set_fail),
         cmocka_unit_test(test_assert_int_not_in_set_fail),
+        cmocka_unit_test(test_assert_uint_in_set_fail),
     };
 
     return cmocka_run_group_tests(set_fail_tests, NULL, NULL);
 }
-