From: Tim Gore <tim.gore@xxxxxxxxx> igt_subtest_init mainly does stuff that we also want for simple/single tests, such as looking for --list-subtests and --help options and calling common_init. So just call this from igt_simple_init and then set tests_with_subtests to false. NOTE that this means that check_igt_exit is now registered as an exit handler for single tests, so need to make sure that ALL tests exit via igt_exit. Signed-off-by: Tim Gore <tim.gore@xxxxxxxxx> --- lib/igt_core.c | 37 +++++++++++++++++++++---------------- lib/igt_core.h | 4 ++-- tests/gem_gtt_hog.c | 2 +- tests/igt_simulation.c | 4 ++-- 4 files changed, 26 insertions(+), 21 deletions(-) diff --git a/lib/igt_core.c b/lib/igt_core.c index b822bc9..7871360 100644 --- a/lib/igt_core.c +++ b/lib/igt_core.c @@ -457,14 +457,21 @@ void igt_subtest_init(int argc, char **argv) * If there's not a reason to the contrary it's less error prone to just use an * #igt_simple_main block instead of stitching the tests's main() function together * manually. + * We now call through to the igt_subtest_init function since we want most of its + * functionality. Note that this will install the igt_exit_handler so you need to + * exit via igt_exit(), Dont call exit() + * */ -void __igt_simple_init(void) +void __igt_simple_init(int argc, char **argv) { - print_version(); - - oom_adjust_for_doom(); - - common_init(); + /* use igt_subtest_init, it does mostly what we want */ + igt_subtest_init(argc, argv); + /* but of course we dont have subtests - so mark this */ + test_with_subtests = false; + if (list_subtests) + igt_exit(); + if (run_single_subtest) + igt_fail(1); /* we dont have subtests ! */ } /* @@ -565,7 +572,7 @@ void igt_skip(const char *f, ...) assert(in_fixture); __igt_fixture_end(); } else { - exit(IGT_EXIT_SKIP); + igt_exit(); } } @@ -655,7 +662,7 @@ void igt_fail(int exitcode) __igt_fixture_end(); } - exit(exitcode); + igt_exit(); } } @@ -713,18 +720,16 @@ void igt_exit(void) if (igt_only_list_subtests()) exit(IGT_EXIT_SUCCESS); - if (!test_with_subtests) - exit(IGT_EXIT_SUCCESS); - - /* Calling this without calling one of the above is a failure */ - assert(skipped_one || succeeded_one || failed_one); + if (test_with_subtests) + /* Calling this without calling one of the above is a failure */ + assert(skipped_one || succeeded_one || failed_one); if (failed_one) exit(igt_exitcode); - else if (succeeded_one) - exit(IGT_EXIT_SUCCESS); - else + else if (skipped_one) exit(IGT_EXIT_SKIP); + else + exit(IGT_EXIT_SUCCESS); } /* fork support code */ diff --git a/lib/igt_core.h b/lib/igt_core.h index 75ee60c..448beed 100644 --- a/lib/igt_core.h +++ b/lib/igt_core.h @@ -166,7 +166,7 @@ bool igt_only_list_subtests(void); * * Init for simple tests without subtests */ -void __igt_simple_init(void); +void __igt_simple_init(int argc, char **argv); /** * igt_simple_main: @@ -181,7 +181,7 @@ void __igt_simple_init(void); #define igt_simple_main \ static void igt_tokencat(__real_main, __LINE__)(int argc, char **argv); \ int main(int argc, char **argv) { \ - __igt_simple_init(); \ + __igt_simple_init(argc, argv); \ igt_tokencat(__real_main, __LINE__)(argc, argv); \ igt_exit(); \ } \ diff --git a/tests/gem_gtt_hog.c b/tests/gem_gtt_hog.c index 5d47540..f607ea0 100644 --- a/tests/gem_gtt_hog.c +++ b/tests/gem_gtt_hog.c @@ -150,7 +150,7 @@ static void run(data_t *data, int child) munmap(ptr, size); igt_assert(x == canary); - exit(0); + _exit(0); } igt_simple_main diff --git a/tests/igt_simulation.c b/tests/igt_simulation.c index 6d8a59a..5d38026 100644 --- a/tests/igt_simulation.c +++ b/tests/igt_simulation.c @@ -53,11 +53,11 @@ static int do_fork(void) assert(0); case 0: if (simple) { - __igt_simple_init(); + __igt_simple_init(1, argv_run); igt_skip_on_simulation(); - exit(0); + igt_exit(); } else { if (list_subtests) igt_subtest_init(2, argv_list); -- 1.9.2 _______________________________________________ Intel-gfx mailing list Intel-gfx@xxxxxxxxxxxxxxxxxxxxx http://lists.freedesktop.org/mailman/listinfo/intel-gfx