lib: igt_simple_main/init for subtest-less tests

Atm only used to print the version information.

Signed-off-by: Daniel Vetter <daniel.vetter@ffwll.ch>

2 files changed, 16 insertions, 0 deletions

diff --git a/lib/drmtest.c b/lib/drmtest.c
index 943a6cfc..3d79a4d3 100644
--- a/lib/drmtest.c
+++ b/lib/drmtest.c
@@ -932,6 +932,11 @@ void igt_subtest_init(int argc, char **argv)
 	optind = 1;
 }
 
+void igt_simple_init(void)
+{
+	print_version();
+}
+
 /*
  * Note: Testcases which use these helpers MUST NOT output anything to stdout
  * outside of places protected by igt_run_subtest checks - the piglit
diff --git a/lib/drmtest.h b/lib/drmtest.h
index 20e26205..a8de2b49 100644
--- a/lib/drmtest.h
+++ b/lib/drmtest.h
@@ -109,6 +109,17 @@ void igt_permute_array(void *array, unsigned size,
 						 unsigned j));
 void igt_progress(const char *header, uint64_t i, uint64_t total);
 
+/* init for simple tests without subtests */
+void igt_simple_init(void);
+#define igt_simple_main \
+	static void igt_tokencat(__real_main, __LINE__)(void); \
+	int main(int argc, char **argv) { \
+		igt_simple_init(); \
+		igt_tokencat(__real_main, __LINE__)(); \
+		exit(0); \
+	} \
+	static void igt_tokencat(__real_main, __LINE__)(void) \
+
 /* subtest infrastructure */
 jmp_buf igt_subtest_jmpbuf;
 void igt_subtest_init(int argc, char **argv);