testsuite: improve coverage of shared/util.h

[thirdparty/kmod.git] / testsuite / test-util.c

diff --git a/testsuite/test-util.c b/testsuite/test-util.c
index 5fa8c44e5692ae1c9e073b296a2e5038e465570e..b0c4ce6c8dca7919c4e588fbb523569e63a50071 100644 (file)
--- a/testsuite/test-util.c
+++ b/testsuite/test-util.c
@@ -16,9 +16,14 @@
  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <fcntl.h>
+#include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <sys/types.h>
 
 #include <shared/util.h>
 
@@ -96,4 +101,110 @@ DEFINE_TEST(test_freadline_wrapped,
                .out = TESTSUITE_ROOTFS "test-util/freadline_wrapped-correct.txt",
        });
 
+static int test_strchr_replace(const struct test *t)
+{
+       _cleanup_free_ char *s = strdup("this is a test string");
+       const char *res = "thiC iC a teCt Ctring";
+
+       strchr_replace(s, 's', 'C');
+       assert_return(streq(s, res), EXIT_FAILURE);
+
+       return EXIT_SUCCESS;
+}
+DEFINE_TEST(test_strchr_replace,
+       .description = "check implementation of strchr_replace()",
+       .need_spawn = false,
+       );
+
+static int test_underscores(const struct test *t)
+{
+       struct teststr {
+               char *val;
+               const char *res;
+       } teststr[] = {
+               { strdup("aa-bb-cc_"), "aa_bb_cc_" },
+               { strdup("-aa-bb-cc-"), "_aa_bb_cc_" },
+               { strdup("-aa[-bb-]cc-"), "_aa[-bb-]cc_" },
+               { strdup("-aa-[bb]-cc-"), "_aa_[bb]_cc_" },
+               { strdup("-aa-[b-b]-cc-"), "_aa_[b-b]_cc_" },
+               { strdup("-aa-b[-]b-cc"), "_aa_b[-]b_cc" },
+               { }
+       }, *iter;
+
+       for (iter = &teststr[0]; iter->val != NULL; iter++) {
+               _cleanup_free_ char *val = iter->val;
+               underscores(val);
+               assert_return(streq(val, iter->res), EXIT_FAILURE);
+       }
+
+       return EXIT_SUCCESS;
+}
+DEFINE_TEST(test_underscores,
+       .description = "check implementation of underscores()",
+       .need_spawn = false,
+       );
+
+static int test_path_ends_with_kmod_ext(const struct test *t)
+{
+       struct teststr {
+               const char *val;
+               bool res;
+       } teststr[] = {
+               { "/bla.ko", true },
+#ifdef ENABLE_ZLIB
+               { "/bla.ko.gz", true },
+#endif
+#ifdef ENABLE_XZ
+               { "/bla.ko.xz", true },
+#endif
+               { "/bla.ko.x", false },
+               { "/bla.ko.", false },
+               { "/bla.koz", false },
+               { "/b", false },
+               { }
+       }, *iter;
+
+       for (iter = &teststr[0]; iter->val != NULL; iter++) {
+               assert_return(path_ends_with_kmod_ext(iter->val,
+                                                     strlen(iter->val)) == iter->res,
+                             EXIT_FAILURE);
+       }
+
+       return EXIT_SUCCESS;
+}
+DEFINE_TEST(test_path_ends_with_kmod_ext,
+       .description = "check implementation of path_ends_with_kmod_ext()",
+       .need_spawn = false,
+       );
+
+#define TEST_WRITE_STR_SAFE_FILE "/write-str-safe"
+#define TEST_WRITE_STR_SAFE_PATH TESTSUITE_ROOTFS "test-util2/" TEST_WRITE_STR_SAFE_FILE
+static int test_write_str_safe(const struct test *t)
+{
+       const char *s = "test";
+       int fd;
+
+       fd = open(TEST_WRITE_STR_SAFE_FILE ".txt", O_CREAT|O_TRUNC|O_WRONLY, 0644);
+       assert_return(fd >= 0, EXIT_FAILURE);
+
+       write_str_safe(fd, s, strlen(s));
+       close(fd);
+
+       return EXIT_SUCCESS;
+}
+DEFINE_TEST(test_write_str_safe,
+       .description = "check implementation of write_str_safe()",
+       .config = {
+               [TC_ROOTFS] = TESTSUITE_ROOTFS "test-util2/",
+       },
+       .need_spawn = true,
+       .output = {
+               .files = (const struct keyval[]) {
+                       { TEST_WRITE_STR_SAFE_PATH ".txt",
+                         TEST_WRITE_STR_SAFE_PATH "-correct.txt" },
+                       { }
+               },
+       });
+
+
 TESTSUITE_MAIN();