7 #include "ccan/tap/tap.h"
8 #include "ccan/talloc/talloc.h"
9 #include "ccan/string/string.h"
12 /* FIXME: Use build bug later. */
13 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
15 static struct test *tests = NULL;
16 static struct obj *objs = NULL;
22 void (*testfn)(const char *dir, struct test_type *t, const char *name);
28 struct test_type *type;
38 static char *output_name(const char *name)
42 assert(strends(name, ".c"));
44 ret = talloc_strdup(name, name);
45 ret[strlen(ret) - 2] = '\0';
49 static char *obj_list(void)
51 char *list = talloc_strdup(objs, "");
54 for (i = objs; i; i = i->next)
55 list = talloc_asprintf_append(list, "%s ", i->name);
58 list = talloc_asprintf_append(list, "ccan/tap/tap.o");
62 static void compile_objs(void)
66 for (i = objs; i; i = i->next) {
67 char *cmd = talloc_asprintf(i, "gcc " CFLAGS " -o %s.o -c %s%s",
68 output_name(i->name), i->name,
69 verbose ? "" : "> /dev/null 2>&1");
70 ok(system(cmd) == 0, "%s", cmd);
74 static void cleanup_objs(void)
78 for (i = objs; i; i = i->next)
79 unlink(talloc_asprintf(i, "%s.o", output_name(i->name)));
82 static void add_test(const char *testdir, const char *name, struct test_type *t)
84 struct test *test = talloc(testdir, struct test);
88 test->name = talloc_asprintf(test, "%s/%s", testdir, name);
92 static void add_obj(const char *testdir, const char *name)
94 struct obj *obj = talloc(testdir, struct obj);
97 obj->name = talloc_asprintf(obj, "%s/%s", testdir, name);
101 static int build(const char *dir, const char *name, int fail)
105 char *externals = talloc_strdup(name, "");
108 for (deps = get_deps(objs, dir); *deps; deps++) {
109 if (!strstarts(*deps, "ccan/"))
112 /* ccan/foo -> ccan/foo.o */
113 externals = talloc_asprintf_append(externals, " %s.o", *deps);
116 cmd = talloc_asprintf(name, "gcc " CFLAGS " %s -o %s %s %s%s%s",
117 fail ? "-DFAIL" : "",
118 output_name(name), name, obj_list(), externals,
119 verbose ? "" : "> /dev/null 2>&1");
122 fprintf(stderr, "Running %s\n", cmd);
126 diag("cmd '%s' failed to execute", cmd);
131 static void compile_ok(const char *dir, struct test_type *t, const char *name)
133 ok(build(dir, name, 0) == 0, "%s %s", t->name, name);
136 static void compile_fail(const char *dir, struct test_type *t, const char *name)
138 if (build(dir, name, 0) != 0)
139 fail("non-FAIL build %s", name);
141 ok(build(dir, name, 1) > 0, "%s %s", t->name, name);
144 static void run(const char *name)
146 if (system(output_name(name)) == -1)
147 fail("running %s had error %m", name);
150 static void cleanup(const char *name)
152 unlink(output_name(name));
155 static struct test_type test_types[] = {
156 { "compile_ok", compile_ok },
157 { "compile_fail", compile_fail },
158 { "run", compile_ok },
161 int main(int argc, char *argv[])
167 unsigned int num_tests = 0, num_objs = 0;
169 if (argc > 1 && streq(argv[1], "--verbose")) {
176 errx(1, "Usage: run_tests [--verbose] <dir>");
178 testdir = talloc_asprintf(NULL, "%s/test", argv[1]);
179 dir = opendir(testdir);
181 err(1, "Opening '%s'", testdir);
183 while ((d = readdir(dir)) != NULL) {
185 if (d->d_name[0] == '.' || !strends(d->d_name, ".c"))
188 for (i = 0; i < ARRAY_SIZE(test_types); i++) {
189 if (strstarts(d->d_name, test_types[i].name)) {
190 add_test(testdir, d->d_name, &test_types[i]);
195 if (i == ARRAY_SIZE(test_types)) {
196 add_obj(testdir, d->d_name);
201 plan_tests(num_tests + num_objs + (num_objs ? 1 : 0));
202 /* First all the extra object compilations. */
205 /* Do all the test compilations. */
206 for (test = tests; test; test = test->next)
207 test->type->testfn(argv[1], test->type, test->name);
211 /* Now run all the ones which wanted to run. */
212 for (test = tests; test; test = test->next) {
213 if (streq(test->type->name, "run"))