X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=gtest_sdl_main.cpp;h=70fed134b2ad4b134787c8d26c8875b2ea95ac89;hp=90a929edd7d330853c44a7d291ce7aeca3b24e7b;hb=0ed2c7fe3876a49d1565e3425e5a491206ffe32d;hpb=65c6584f77bff0af0c8e38d1ac90298bcd55e9ac

diff --git a/gtest_sdl_main.cpp b/gtest_sdl_main.cpp
index 90a929e..70fed13 100644
--- a/gtest_sdl_main.cpp
+++ b/gtest_sdl_main.cpp
@@ -9,6 +9,9 @@
 #include <SDL/SDL_error.h>
 #include <SDL/SDL_video.h>
 #endif
+#ifdef HAVE_BENCHMARK
+#include <benchmark/benchmark.h>
+#endif
 #include <stdio.h>
 #include <stdlib.h>
 
@@ -50,8 +53,22 @@ int main(int argc, char **argv) {
 	SDL_WM_SetCaption("OpenGL window for unit test", NULL);
 #endif
 
-	testing::InitGoogleTest(&argc, argv);
-	int err = RUN_ALL_TESTS();
+	int err;
+	if (argc >= 2 && strcmp(argv[1], "--benchmark") == 0) {
+#ifdef HAVE_BENCHMARK
+		--argc;
+		::benchmark::Initialize(&argc, argv + 1);
+		if (::benchmark::ReportUnrecognizedArguments(argc, argv)) return 1;
+		::benchmark::RunSpecifiedBenchmarks();
+		err = 0;
+#else
+		fprintf(stderr, "No support for microbenchmarks compiled in.\n");
+		err = 1;
+#endif
+	} else {
+		testing::InitGoogleTest(&argc, argv);
+		err = RUN_ALL_TESTS();
+	}
 	SDL_Quit();
-	exit(err);
+	return err;
 }