tests/unit/test_atoi_strtoi.c: Test strtou_noneg()

Signed-off-by: Alejandro Colomar <alx@kernel.org>

diff --git a/tests/unit/test_atoi_strtoi.c b/tests/unit/test_atoi_strtoi.c
index c82b17d7bcd756838d25390cea0d4cbbe5915219..535b6ab74578cd5fe153a9b33dafa8d510355963 100644 (file)
--- a/tests/unit/test_atoi_strtoi.c
+++ b/tests/unit/test_atoi_strtoi.c
@@ -15,10 +15,12 @@
 #include <cmocka.h>
 
 #include "atoi/strtoi.h"
+#include "atoi/strtou_noneg.h"
 
 
 static void test_strtoi(void **state);
 static void test_strtou(void **state);
+static void test_strtou_noneg(void **state);
 
 
 int
@@ -27,6 +29,7 @@ main(void)
     const struct CMUnitTest  tests[] = {
         cmocka_unit_test(test_strtoi),
         cmocka_unit_test(test_strtou),
+        cmocka_unit_test(test_strtou_noneg),
     };
 
     return cmocka_run_group_tests(tests, NULL, NULL);
@@ -107,3 +110,48 @@ test_strtou(void **state)
 
        assert_true(errno == 0);
 }
+
+
+static void
+test_strtou_noneg(void **state)
+{
+       int   status;
+       char  *end;
+
+       errno = 0;
+       assert_true(strtou_noneg("42", NULL, -1, 1, 2, &status)
+                   == 1);
+       assert_true(status == EINVAL);
+
+       assert_true(strtou_noneg("40", &end, 5, 0, UINTMAX_MAX, &status)
+                   == 20);
+       assert_true(status == 0);
+       assert_true(strcmp(end, "") == 0);
+
+       assert_true(strtou_noneg("-40", &end, 0, 2, UINTMAX_MAX, &status)
+                   == 2);
+       assert_true(status == ERANGE);
+       assert_true(strcmp(end, "") == 0);
+
+       assert_true(strtou_noneg("z", &end, 0, 0, UINTMAX_MAX, &status)
+                   == 0);
+       assert_true(status == ECANCELED);
+       assert_true(strcmp(end, "z") == 0);
+
+       assert_true(strtou_noneg(" 5", &end, 0, 3, 4, &status)
+                   == 4);
+       assert_true(status == ERANGE);
+       assert_true(strcmp(end, "") == 0);
+
+       assert_true(strtou_noneg("5z", &end, 0, 0, UINTMAX_MAX, &status)
+                   == 5);
+       assert_true(status == ENOTSUP);
+       assert_true(strcmp(end, "z") == 0);
+
+       assert_true(strtou_noneg("5z", &end, 0, 0, 4, &status)
+                   == 4);
+       assert_true(status == ERANGE);
+       assert_true(strcmp(end, "z") == 0);
+
+       assert_true(errno == 0);
+}