- 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();
+ }