diff --git a/include/criterion/options.h b/include/criterion/options.h index d04db21..94829cf 100644 --- a/include/criterion/options.h +++ b/include/criterion/options.h @@ -32,6 +32,7 @@ struct criterion_options { bool enable_tap_format; bool no_early_exit; bool always_succeed; + bool use_ascii; }; extern struct criterion_options criterion_options; diff --git a/src/main.c b/src/main.c index ab362ea..3911cd7 100644 --- a/src/main.c +++ b/src/main.c @@ -1,8 +1,11 @@ #define _GNU_SOURCE #include #include +#include #include #include +#include +#include "runner.h" # define VERSION "v1.0.0" # define VERSION_MSG "Tests compiled with Criterion " VERSION "\n" @@ -14,6 +17,9 @@ " -h or --help: prints this message\n" \ " -v or --version: prints the version of criterion " \ "these tests have been linked against\n" \ + " -l or --list: prints all the tests in a list\n" \ + " --ascii: don't use fancy unicode symbols " \ + "or colors in the output\n" \ " --tap: enables TAP formatting\n" \ " --always-succeed: always exit with 0\n" \ " --no-early-exit: do not exit the test worker " \ @@ -31,12 +37,54 @@ int print_version(void) { return 0; } +# define UTF8_TREE_NODE "├" +# define UTF8_TREE_END "└" +# define UTF8_TREE_JOIN "──" + +# define ASCII_TREE_NODE "|" +# define ASCII_TREE_END "`" +# define ASCII_TREE_JOIN "--" + +bool is_disabled(struct criterion_suite *s, struct criterion_test *t) { + return (s->data && s->data->disabled) || t->data->disabled; +} + +int list_tests(bool unicode) { + smart struct criterion_test_set *set = criterion_init(); + + const char *node = unicode ? UTF8_TREE_NODE : ASCII_TREE_NODE; + const char *join = unicode ? UTF8_TREE_JOIN : ASCII_TREE_JOIN; + const char *end = unicode ? UTF8_TREE_END : ASCII_TREE_END; + + FOREACH_SET(struct criterion_suite_set *s, set->suites) { + size_t tests = s->tests ? s->tests->size : 0; + if (!tests) + continue; + + printf("%s: " SIZE_T_FORMAT " test%s\n", + s->suite.name, + tests, + tests == 1 ? "" : "s"); + + FOREACH_SET(struct criterion_test *t, s->tests) { + printf("%s%s %s%s\n", + --tests == 0 ? end : node, + join, + t->name, + is_disabled(&s->suite, t) ? "(disabled)" : ""); + } + } + return 0; +} + int main(int argc, char *argv[]) { static struct option opts[] = { {"verbose", optional_argument, 0, 'b'}, {"version", no_argument, 0, 'v'}, {"tap", no_argument, 0, 't'}, {"help", no_argument, 0, 'h'}, + {"list", no_argument, 0, 'l'}, + {"ascii", no_argument, 0, 'k'}, {"always-succeed", no_argument, 0, 'y'}, {"no-early-exit", no_argument, 0, 'z'}, {0, 0, 0, 0 } @@ -49,17 +97,28 @@ int main(int argc, char *argv[]) { .logging_threshold = atoi(getenv("CRITERION_VERBOSITY_LEVEL") ?: "2"), }; - for (int c; (c = getopt_long(argc, argv, "hv", opts, NULL)) != -1;) { + bool do_list_tests = false; + bool do_print_version = false; + bool do_print_usage = false; + for (int c; (c = getopt_long(argc, argv, "hvl", opts, NULL)) != -1;) { switch (c) { case 'b': criterion_options.logging_threshold = atoi(optarg ?: "1"); break; case 't': criterion_options.enable_tap_format = true; break; case 'y': criterion_options.always_succeed = true; break; case 'z': criterion_options.no_early_exit = true; break; - case 'v': return print_version(); + case 'k': criterion_options.use_ascii = true; break; + case 'l': do_list_tests = true; break; + case 'v': do_print_version = true; break; case 'h': - default : return print_usage(argv[0]); + default : do_print_usage = true; break; } } + if (do_print_usage) + return print_usage(argv[0]); + if (do_print_version) + return print_version(); + if (do_list_tests) + return list_tests(!criterion_options.use_ascii); return !criterion_run_all_tests(); } diff --git a/src/runner.c b/src/runner.c index baaeee0..231bfc9 100644 --- a/src/runner.c +++ b/src/runner.c @@ -59,7 +59,7 @@ static void dtor_test_set(void *ptr, UNUSED void *meta) { sfree(t->suites); } -static struct criterion_test_set *criterion_init(void) { +struct criterion_test_set *criterion_init(void) { struct criterion_ordered_set *suites = new_ordered_set(cmp_suite, dtor_suite_set); FOREACH_SUITE_SEC(s) { diff --git a/src/runner.h b/src/runner.h index ca243f0..64cc668 100644 --- a/src/runner.h +++ b/src/runner.h @@ -29,6 +29,8 @@ DECL_SECTION_LIMITS(struct criterion_test, criterion_tests); DECL_SECTION_LIMITS(struct criterion_suite, crit_suites); +struct criterion_test_set *criterion_init(void); + # define FOREACH_TEST_SEC(Test) \ for (struct criterion_test *Test = SECTION_START(criterion_tests); \ Test < SECTION_END(criterion_tests); \