utils/benchmark/test/diagnostics_test.cc

   1 // Testing:
   2 //   State::PauseTiming()
   3 //   State::ResumeTiming()
   4 // Test that CHECK's within these function diagnose when they are called
   5 // outside of the KeepRunning() loop.
   6 //
   7 // NOTE: Users should NOT include or use src/check.h. This is only done in
   8 // order to test library internals.
   9
  10 #include <cstdlib>
  11 #include <stdexcept>
  12
  13 #include "../src/check.h"
  14 #include "benchmark/benchmark.h"
  15
  16 #if defined(__GNUC__) && !defined(__EXCEPTIONS)
  17 #define TEST_HAS_NO_EXCEPTIONS
  18 #endif
  19
  20 void TestHandler() {
  21 #ifndef TEST_HAS_NO_EXCEPTIONS
  22   throw std::logic_error("");
  23 #else
  24   std::abort();
  25 #endif
  26 }
  27
  28 void try_invalid_pause_resume(benchmark::State& state) {
  29 #if !defined(TEST_BENCHMARK_LIBRARY_HAS_NO_ASSERTIONS) && !defined(TEST_HAS_NO_EXCEPTIONS)
  30   try {
  31     state.PauseTiming();
  32     std::abort();
  33   } catch (std::logic_error const&) {
  34   }
  35   try {
  36     state.ResumeTiming();
  37     std::abort();
  38   } catch (std::logic_error const&) {
  39   }
  40 #else
  41   (void)state;  // avoid unused warning
  42 #endif
  43 }
  44
  45 void BM_diagnostic_test(benchmark::State& state) {
  46   static bool called_once = false;
  47
  48   if (called_once == false) try_invalid_pause_resume(state);
  49
  50   for (auto _ : state) {
  51     benchmark::DoNotOptimize(state.iterations());
  52   }
  53
  54   if (called_once == false) try_invalid_pause_resume(state);
  55
  56   called_once = true;
  57 }
  58 BENCHMARK(BM_diagnostic_test);
  59
  60
  61 void BM_diagnostic_test_keep_running(benchmark::State& state) {
  62   static bool called_once = false;
  63
  64   if (called_once == false) try_invalid_pause_resume(state);
  65
  66   while(state.KeepRunning()) {
  67     benchmark::DoNotOptimize(state.iterations());
  68   }
  69
  70   if (called_once == false) try_invalid_pause_resume(state);
  71
  72   called_once = true;
  73 }
  74 BENCHMARK(BM_diagnostic_test_keep_running);
  75
  76 int main(int argc, char* argv[]) {
  77   benchmark::internal::GetAbortHandler() = &TestHandler;
  78   benchmark::Initialize(&argc, argv);
  79   benchmark::RunSpecifiedBenchmarks();
  80 }