| 1 | #include <memory> |
| 2 | |
| 3 | #include "../src/check.h" |
| 4 | #include "benchmark/benchmark.h" |
| 5 | #include "output_test.h" |
| 6 | |
| 7 | class TestMemoryManager : public benchmark::MemoryManager { |
| 8 | void Start() override {} |
| 9 | void Stop(Result& result) override { |
| 10 | result.num_allocs = 42; |
| 11 | result.max_bytes_used = 42000; |
| 12 | } |
| 13 | }; |
| 14 | |
| 15 | void BM_empty(benchmark::State& state) { |
| 16 | for (auto _ : state) { |
| 17 | auto iterations = double(state.iterations()) * double(state.iterations()); |
| 18 | benchmark::DoNotOptimize(value&: iterations); |
| 19 | } |
| 20 | } |
| 21 | BENCHMARK(BM_empty); |
| 22 | |
| 23 | ADD_CASES(TC_ConsoleOut, {{"^BM_empty %console_report$" }}); |
| 24 | ADD_CASES(TC_JSONOut, {{"\"name\": \"BM_empty\",$" }, |
| 25 | {"\"family_index\": 0,$" , MR_Next}, |
| 26 | {"\"per_family_instance_index\": 0,$" , MR_Next}, |
| 27 | {"\"run_name\": \"BM_empty\",$" , MR_Next}, |
| 28 | {"\"run_type\": \"iteration\",$" , MR_Next}, |
| 29 | {"\"repetitions\": 1,$" , MR_Next}, |
| 30 | {"\"repetition_index\": 0,$" , MR_Next}, |
| 31 | {"\"threads\": 1,$" , MR_Next}, |
| 32 | {"\"iterations\": %int,$" , MR_Next}, |
| 33 | {"\"real_time\": %float,$" , MR_Next}, |
| 34 | {"\"cpu_time\": %float,$" , MR_Next}, |
| 35 | {"\"time_unit\": \"ns\",$" , MR_Next}, |
| 36 | {"\"allocs_per_iter\": %float,$" , MR_Next}, |
| 37 | {"\"max_bytes_used\": 42000$" , MR_Next}, |
| 38 | {"}" , MR_Next}}); |
| 39 | ADD_CASES(TC_CSVOut, {{"^\"BM_empty\",%csv_report$" }}); |
| 40 | |
| 41 | int main(int argc, char* argv[]) { |
| 42 | std::unique_ptr<benchmark::MemoryManager> mm(new TestMemoryManager()); |
| 43 | |
| 44 | benchmark::RegisterMemoryManager(memory_manager: mm.get()); |
| 45 | RunOutputTests(argc, argv); |
| 46 | benchmark::RegisterMemoryManager(memory_manager: nullptr); |
| 47 | } |
| 48 | |