support limit

src/lib/testsuite/test.c

@@ -1,3 +1,4 @@
+#include <argp.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -16,26 +17,94 @@ extern void test_preason(void);
 extern void test_ordq(void);
 extern void test_delay(void);
 extern void test_kvmgr(void);
+extern void test_group(void);
+extern void test_xlist(void);
+extern void test_limit(void);
 
 void setUp(void) { return; }
 
 void tearDown(void) { return; }
 
+static char doc[]      = "xcelib_test";
+static char args_doc[] = "";
+
+#define MAX_TESTS (256)
+
+struct arguments {
+    int   ntests;
+    char* tests[MAX_TESTS];
+};
+
+static struct argp_option options[] = {
+    { "run", 't', "STR", 0, "run test" },
+    { 0 }
+};
+
+static error_t parse_opt(int key, char* arg, struct argp_state* state)
+{
+    struct arguments* arguments = state->input;
+
+    switch (key) {
+    case 't':
+        if (arguments->ntests >= MAX_TESTS) {
+            fprintf(stderr, "Too many tests\n");
+            exit(-1);
+        }
+        arguments->tests[arguments->ntests] = arg;
+        arguments->ntests++;
+        break;
+    default:
+        return ARGP_ERR_UNKNOWN;
+    }
+
+    return 0;
+}
+
+static struct argp argp = { options, parse_opt, args_doc, doc };
+struct arguments   arguments;
+
+static int check_test(struct arguments* args, char* x)
+{
+    int i;
+
+    if (args->ntests == 0) {
+        return 1;
+    }
+    for (i = 0; i < args->ntests; i++) {
+        if (strcmp(x, args->tests[i]) == 0) {
+            return 1;
+        }
+    }
+    return 0;
+}
+
+#define TEST(x)                                                                \
+    do {                                                                       \
+        if (check_test(&arguments, #x)) {                                      \
+            x();                                                               \
+        }                                                                      \
+    } while (0)
+
 int main(int argc, char* argv[])
 {
+    argp_parse(&argp, argc, argv, 0, 0, &arguments);
+
     UNITY_BEGIN();
-    test_config();
-    test_idgen();
-    test_ht_string();
-    test_ht_int();
-    test_zarray();
-    test_xstrings();
-    test_error();
-    test_utils();
-    test_preason();
-    test_ordq();
-    test_delay();
-    test_kvmgr();
+    TEST(test_config);
+    TEST(test_idgen);
+    TEST(test_ht_string);
+    TEST(test_ht_int);
+    TEST(test_zarray);
+    TEST(test_xstrings);
+    TEST(test_error);
+    TEST(test_utils);
+    TEST(test_preason);
+    TEST(test_ordq);
+    TEST(test_delay);
+    TEST(test_kvmgr);
+    TEST(test_group);
+    TEST(test_xlist);
+    TEST(test_limit);
 
     return UNITY_END();
 }