92 lines · plain
1// Testing:2// State::PauseTiming()3// State::ResumeTiming()4// Test that CHECK's within these function diagnose when they are called5// outside of the KeepRunning() loop.6//7// NOTE: Users should NOT include or use src/check.h. This is only done in8// 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_EXCEPTIONS18#endif19 20void TestHandler() {21#ifndef TEST_HAS_NO_EXCEPTIONS22 throw std::logic_error("");23#else24 std::abort();25#endif26}27 28void try_invalid_pause_resume(benchmark::State& state) {29#if !defined(TEST_BENCHMARK_LIBRARY_HAS_NO_ASSERTIONS) && \30 !defined(TEST_HAS_NO_EXCEPTIONS)31 try {32 state.PauseTiming();33 std::abort();34 } catch (std::logic_error const&) {35 }36 try {37 state.ResumeTiming();38 std::abort();39 } catch (std::logic_error const&) {40 }41#else42 (void)state; // avoid unused warning43#endif44}45 46void BM_diagnostic_test(benchmark::State& state) {47 static bool called_once = false;48 49 if (called_once == false) try_invalid_pause_resume(state);50 51 for (auto _ : state) {52 auto iterations = double(state.iterations()) * double(state.iterations());53 benchmark::DoNotOptimize(iterations);54 }55 56 if (called_once == false) try_invalid_pause_resume(state);57 58 called_once = true;59}60BENCHMARK(BM_diagnostic_test);61 62void BM_diagnostic_test_keep_running(benchmark::State& state) {63 static bool called_once = false;64 65 if (called_once == false) try_invalid_pause_resume(state);66 67 while (state.KeepRunning()) {68 auto iterations = double(state.iterations()) * double(state.iterations());69 benchmark::DoNotOptimize(iterations);70 }71 72 if (called_once == false) try_invalid_pause_resume(state);73 74 called_once = true;75}76BENCHMARK(BM_diagnostic_test_keep_running);77 78int main(int argc, char* argv[]) {79#ifdef NDEBUG80 // This test is exercising functionality for debug builds, which are not81 // available in release builds. Skip the test if we are in that environment82 // to avoid a test failure.83 std::cout << "Diagnostic test disabled in release build" << std::endl;84 (void)argc;85 (void)argv;86#else87 benchmark::internal::GetAbortHandler() = &TestHandler;88 benchmark::Initialize(&argc, argv);89 benchmark::RunSpecifiedBenchmarks();90#endif91}92