unit: Add test-argconfig-parse

Signed-off-by: Tokunori Ikegami <ikegami.t@gmail.com>

diff --git a/unit/meson.build b/unit/meson.build
index 296f763796e151212f533f90c644708cfc647feb..7e0e87848f983d9270b29b435596f42ebaf183fe 100644 (file)
--- a/unit/meson.build
+++ b/unit/meson.build
@@ -35,3 +35,12 @@ test_uint128_si = executable(
 )
 
 test('uint128-si', test_uint128_si)
+
+test_argconfig_parse = executable(
+    'test-argconfig-parse',
+    ['test-argconfig-parse.c', '../util/argconfig.c', '../util/suffix.c'],
+    include_directories: [incdir, '..'],
+    dependencies: [libnvme_dep],
+)
+
+test('argconfig_parse', test_argconfig_parse)

diff --git a/unit/test-argconfig-parse.c b/unit/test-argconfig-parse.c
new file mode 100644 (file)
index 0000000..23c8d4f
--- /dev/null
+++ b/unit/test-argconfig-parse.c
@@ -0,0 +1,164 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+#include <string.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <errno.h>
+#include <locale.h>
+
+#include "../util/argconfig.h"
+#include "nvme/types.h"
+
+#define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0]))
+
+static int test_rc;
+
+union val {
+       bool flag;
+       __u64 suffix;
+       __u32 uint;
+       int int_val;
+       __u64 long_val;
+       double double_val;
+       __u8 byte;
+       __u16 shrt;
+       int incr;
+       char *string;
+       char *fmt;
+       char *file;
+       char *list;
+       char *str;
+};
+
+struct toval_test {
+       char *arg;
+       void *val;
+       union val exp;
+       int size;
+       int ret;
+};
+
+static void check_val(const char *arg, void *exp, void *val, int size)
+{
+       if ((size && !memcmp(exp, val, size)) ||
+           (!size && !strcmp(*(char **)exp, *(char **)val)))
+               return;
+
+       switch (size) {
+       case 0:
+               printf("ERROR: printing {%s}, got '%s', expected '%s'\n",
+                      arg, *(char **)val, *(char **)exp);
+               break;
+       default:
+               printf("ERROR: printing {%s}, got '%llu', expected '%llu'\n",
+                      arg, *(unsigned long long *)val, *(unsigned long long *)exp);
+               break;
+       }
+
+       test_rc = 1;
+}
+
+struct cfg {
+       bool flag;
+       __u64 suffix;
+       __u32 uint;
+       int int_val;
+       __u64 long_val;
+       double double_val;
+       __u8 byte;
+       __u16 shrt;
+       int incr;
+       char *string;
+       char *fmt;
+       char *file;
+       char *list;
+       char *str;
+};
+
+static struct cfg cfg;
+
+#define VAL_TEST(a, c, v, l, r) \
+       { a, &cfg.c, { .c = v }, l ? sizeof(cfg.c) : 0, r }
+
+static struct toval_test toval_tests[] = {
+       VAL_TEST("--flag", flag, true, true, 0),
+       VAL_TEST("--flag=1", flag, false, true, -EINVAL),
+       VAL_TEST("--suffix=0", suffix, 0, true, 0),
+       VAL_TEST("--suffix=1", suffix, 1, true, 0),
+       VAL_TEST("--suffix=1234", suffix, 1234, true, 0),
+       VAL_TEST("--suffix=4096", suffix, 4096, true, 0),
+       VAL_TEST("--suffix=1Ki", suffix, 1024, true, 0),
+       VAL_TEST("--suffix=34Gi", suffix, 36507222016, true, 0),
+       VAL_TEST("--suffix=34.9Ki", suffix, 0, true, -EINVAL),
+       VAL_TEST("--suffix=32Gii", suffix, 0, true, -EINVAL),
+       VAL_TEST("--uint=1", uint, 1, true, 0),
+       VAL_TEST("--int=1", int_val, 1, true, 0),
+       VAL_TEST("--long=1", long_val, 1, true, 0),
+       VAL_TEST("--double=1", double_val, 1, true, 0),
+       VAL_TEST("--byte=1", byte, 1, true, 0),
+       VAL_TEST("--byte=256", byte, 0, true, -EINVAL),
+       VAL_TEST("--shrt=1", shrt, 1, true, 0),
+       VAL_TEST("--incr", incr, 1, true, 0),
+       VAL_TEST("--incr=1", incr, 0, true, -EINVAL),
+       VAL_TEST("--string=string", string, "string", false, 0),
+       VAL_TEST("--fmt=fmt", fmt, "fmt", false, 0),
+       VAL_TEST("--file=file", file, "file", false, 0),
+       VAL_TEST("--list=list", list, "list", false, 0),
+       VAL_TEST("--str=str", str, "str", false, 0),
+};
+
+void toval_test(struct toval_test *test)
+{
+       const char *desc = "Test argconfig parse";
+       int ret;
+       char *argv[] = { "test-argconfig", test->arg };
+
+       OPT_ARGS(opts) = {
+               OPT_FLAG("flag",'f', &cfg.flag, "flag"),
+               OPT_SUFFIX("suffix", 's', &cfg.suffix, "suffix"),
+               OPT_UINT("uint", 'u', &cfg.uint, "uint"),
+               OPT_INT("int", 'i', &cfg.int_val, "int"),
+               OPT_LONG("long", 'l', &cfg.long_val, "long"),
+               OPT_DOUBLE("double", 'd', &cfg.double_val, "double"),
+               OPT_BYTE("byte", 'b', &cfg.byte, "byte"),
+               OPT_SHRT("shrt", 'S', &cfg.shrt, "shrt"),
+               OPT_INCR("incr", 'I', &cfg.incr, "incr"),
+               OPT_STRING("string", 't', "STRING", &cfg.string, "string"),
+               OPT_FMT("fmt", 'F', &cfg.fmt, "fmt"),
+               OPT_FILE("file", 'L', &cfg.file, "file"),
+               OPT_LIST("list", 'T', &cfg.list, "list"),
+               OPT_STR("str", 'r', &cfg.str, "str"),
+               OPT_END()
+       };
+
+       ret = argconfig_parse(2, argv, desc, opts);
+       if (ret != test->ret) {
+               printf("ERROR: converting {%s} failed\n", test->arg);
+               test_rc = 1;
+               return;
+       }
+       if (ret)
+               return;
+
+       check_val(test->arg, &test->exp, test->val, test->size);
+}
+
+int main(void)
+{
+       unsigned int i;
+       FILE *f;
+
+       test_rc = 0;
+       setlocale(LC_NUMERIC, "C");
+       f = freopen("/dev/null", "w", stderr);
+       if (!f)
+               printf("ERROR: reopening stderr failed: %s\n", strerror(errno));
+
+       for (i = 0; i < ARRAY_SIZE(toval_tests); i++)
+               toval_test(&toval_tests[i]);
+
+       if (f)
+               fclose(f);
+
+       return test_rc ? EXIT_FAILURE : EXIT_SUCCESS;
+}