xref: /aosp_15_r20/external/google-benchmark/test/memory_manager_test.cc (revision dbb99499c3810fa1611fa2242a2fc446be01a57c)
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 {
Start()8   void Start() override {}
Stop(Result & result)9   void Stop(Result& result) override {
10     result.num_allocs = 42;
11     result.max_bytes_used = 42000;
12   }
13 };
14 
BM_empty(benchmark::State & state)15 void BM_empty(benchmark::State& state) {
16   for (auto _ : state) {
17     auto iterations = double(state.iterations()) * double(state.iterations());
18     benchmark::DoNotOptimize(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 
main(int argc,char * argv[])41 int main(int argc, char* argv[]) {
42   std::unique_ptr<benchmark::MemoryManager> mm(new TestMemoryManager());
43 
44   benchmark::RegisterMemoryManager(mm.get());
45   RunOutputTests(argc, argv);
46   benchmark::RegisterMemoryManager(nullptr);
47 }
48