9 #include "ccan/tap/tap.h"
10 #include "ccan/talloc/talloc.h"
11 #include "ccan/str/str.h"
12 #include "ccan/array_size/array_size.h"
15 static struct test *tests = NULL;
16 static struct obj *objs = NULL;
21 void (*buildfn)(const char *dir, struct test_type *t, const char *name,
22 const char *apiobj, const char *libs);
23 void (*runfn)(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(const char *dir)
51 char *list = talloc_strdup(objs, "");
54 for (i = objs; i; i = i->next)
55 list = talloc_asprintf_append(list, "%s ", i->name);
58 if (!streq(dir, "tap") && !strends(dir, "/tap"))
59 list = talloc_asprintf_append(list, "ccan/tap/tap.o");
63 static void compile_objs(void)
67 for (i = objs; i; i = i->next) {
68 char *cmd = talloc_asprintf(i, "gcc " CFLAGS " -o %s.o -c %s%s",
69 output_name(i->name), i->name,
70 verbose ? "" : "> /dev/null 2>&1");
71 ok(system(cmd) == 0, "%s", cmd);
75 static void cleanup_objs(void)
79 for (i = objs; i; i = i->next) {
82 unlink(talloc_asprintf(i, "%s.o", output_name(i->name)));
86 static void add_test(const char *testdir, const char *name, struct test_type *t)
88 struct test *test = talloc(testdir, struct test);
92 test->name = talloc_asprintf(test, "%s/%s", testdir, name);
96 static void add_obj(const char *testdir, const char *name, bool generate)
98 struct obj *obj = talloc(testdir, struct obj);
101 obj->name = talloc_asprintf(obj, "%s/%s", testdir, name);
102 obj->generate = generate;
106 static int build(const char *dir, const char *name, const char *apiobj,
107 const char *libs, int fail)
112 cmd = talloc_asprintf(name, "gcc " CFLAGS " %s -o %s %s %s %s%s %s",
113 fail ? "-DFAIL" : "",
114 output_name(name), name, apiobj, obj_list(dir),
115 libs, verbose ? "" : "> /dev/null 2>&1");
118 fprintf(stderr, "Running %s\n", cmd);
122 diag("cmd '%s' failed to execute", cmd);
127 static void compile_ok(const char *dir, struct test_type *t, const char *name,
128 const char *apiobj, const char *libs)
130 ok(build(dir, name, "", libs, 0) == 0, "%s %s", t->name, name);
133 /* api tests get the API obj linked in as well. */
134 static void compile_api_ok(const char *dir, struct test_type *t,
135 const char *name, const char *apiobj,
138 ok(build(dir, name, apiobj, libs, 0) == 0, "%s %s", t->name, name);
141 static void compile_fail(const char *dir, struct test_type *t, const char *name,
142 const char *apiobj, const char *libs)
144 if (build(dir, name, "", libs, 0) != 0)
145 fail("non-FAIL build %s", name);
147 ok(build(dir, name, "", libs, 1) > 0, "%s %s", t->name, name);
150 static void no_run(const char *name)
154 static void run(const char *name)
156 if (system(output_name(name)) != 0)
157 fail("running %s had error", name);
160 static void cleanup(const char *name)
162 unlink(output_name(name));
165 static struct test_type test_types[] = {
166 { "compile_ok", compile_ok, no_run },
167 { "compile_fail", compile_fail, no_run },
168 { "run", compile_ok, run },
169 { "api", compile_api_ok, run },
172 int main(int argc, char *argv[])
177 const char *apiobj = "";
178 char *libs = talloc_strdup(NULL, "");
180 unsigned int num_tests = 0, num_objs = 0, i;
182 if (argc > 1 && streq(argv[1], "--verbose")) {
188 while (argc > 1 && strstarts(argv[1], "--lib=")) {
189 libs = talloc_asprintf_append(libs, " -l%s",
190 argv[1] + strlen("--lib="));
195 if (argc > 1 && strstarts(argv[1], "--apiobj=")) {
196 apiobj = argv[1] + strlen("--apiobj=");
202 errx(1, "Usage: run_tests [--verbose] [--apiobj=<obj>] <dir> [<extra-objs>...]");
204 testdir = talloc_asprintf(NULL, "%s/test", argv[1]);
205 dir = opendir(testdir);
207 err(1, "Opening '%s'", testdir);
209 while ((d = readdir(dir)) != NULL) {
210 if (d->d_name[0] == '.' || !strends(d->d_name, ".c"))
213 for (i = 0; i < ARRAY_SIZE(test_types); i++) {
214 if (strstarts(d->d_name, test_types[i].name)) {
215 add_test(testdir, d->d_name, &test_types[i]);
220 if (i == ARRAY_SIZE(test_types)) {
221 add_obj(testdir, d->d_name, true);
226 plan_tests(num_tests + num_objs + (num_objs ? 1 : 0));
227 /* First all the extra object compilations. */
230 /* Now add any object files from the command line */
231 cwd = talloc_strdup(testdir, ".");
232 for (i = 2; i < argc; i++)
233 add_obj(cwd, argv[i], false);
235 /* Do all the test compilations. */
236 for (test = tests; test; test = test->next)
237 test->type->buildfn(argv[1], test->type, test->name,
242 /* Now run all the ones which wanted to run. */
243 for (test = tests; test; test = test->next) {
244 test->type->runfn(test->name);