From: Tim Gore <[email protected]>

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 <[email protected]>
---
 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
[email protected]
http://lists.freedesktop.org/mailman/listinfo/intel-gfx

Reply via email to