add test

Author: EfesX

CMakeLists.txt

@@ -3,6 +3,8 @@ cmake_minimum_required (VERSION 3.13...3.22)
 
 project (benchmark VERSION 1.9.1 LANGUAGES CXX)
 
+set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fsanitize=address")
+
 option(BENCHMARK_ENABLE_TESTING "Enable testing of the benchmark library." ON)
 option(BENCHMARK_ENABLE_EXCEPTIONS "Enable the use of exceptions in the benchmark library." ON)
 option(BENCHMARK_ENABLE_LTO "Enable link time optimisation of the benchmark library." OFF)

test/CMakeLists.txt

@@ -85,6 +85,9 @@ benchmark_add_test(NAME spec_arg_verbosity COMMAND spec_arg_verbosity_test --v=4
 compile_benchmark_test(benchmark_setup_teardown_test)
 benchmark_add_test(NAME benchmark_setup_teardown COMMAND benchmark_setup_teardown_test)
 
+compile_benchmark_test(benchmark_setup_teardown_cb_types_test)
+benchmark_add_test(NAME benchmark_setup_teardown_cb_types COMMAND benchmark_setup_teardown_cb_types_test)
+
 compile_benchmark_test(filter_test)
 macro(add_filter_test name filter expect)
   benchmark_add_test(NAME ${name} COMMAND filter_test --benchmark_min_time=0.01s --benchmark_filter=${filter} ${expect})

test/benchmark_setup_teardown_cb_types_test.cc

@@ -0,0 +1,101 @@
+#include <iostream>
+#include <memory>
+
+#include "benchmark/benchmark.h"
+
+using benchmark::callback_function;
+
+namespace counters {
+static int functor_calls = 0;
+static int lambda_calls = 0;
+static int ctr_copy_calls = 0;
+static int ctr_move_calls = 0;
+};
+
+struct Functor {
+  Functor() {}
+  Functor(const Functor& /*unused*/){
+    counters::ctr_copy_calls++;
+  }
+  Functor(Functor&& /*unused*/){
+    counters::ctr_move_calls++;
+  }
+  //void operator()(const benchmark::State& /*unused*/) { counters::functor_calls++; }
+  void operator()(const benchmark::State& /*unused*/) { }
+};
+
+void BM_DoSomething(benchmark::State& state){
+  for(auto _ : state) {
+  }
+}
+
+
+int main(int argc, char** argv) {
+  benchmark::Initialize(&argc, argv);
+  auto bm = benchmark::RegisterBenchmark("BM_CtrCopy", BM_DoSomething);
+
+  {
+    Functor func;
+    bm->Setup(func);
+    bm->Teardown(func);
+    assert(counters::ctr_copy_calls == 2);
+  }
+  {
+    Functor func;
+    Functor& r_func{func};
+    bm->Setup(r_func);
+    bm->Teardown(r_func);
+    assert(counters::ctr_copy_calls == 4);
+  }
+  {
+    Functor func;
+    const Functor& cr_func{func};
+    bm->Setup(cr_func);
+    bm->Teardown(cr_func);
+    assert(counters::ctr_copy_calls == 6);
+  }
+  {
+    Functor func1;
+    Functor func2;
+    bm->Setup(std::move(func1));
+    bm->Teardown(std::move(func2));
+    assert(counters::ctr_move_calls == 2);
+  }
+  {
+    bm->Setup(Functor{});
+    bm->Teardown(Functor{});
+    assert(counters::ctr_move_calls == 4);
+  }
+  {
+    bm->Setup([](const benchmark::State& /*unused*/){});
+    bm->Teardown([](const benchmark::State& /*unused*/){});
+  }
+  {
+    callback_function func1 = [](const benchmark::State& /*unused*/){};
+    callback_function func2 = func1;
+    bm->Setup(std::move(func1));
+    bm->Teardown(std::move(func2));
+    assert(func1 == nullptr);
+    assert(func2 == nullptr);
+  }
+  {
+    callback_function func1 = [](const benchmark::State& /*unused*/){};
+    callback_function func2 = func1;
+    bm->Setup(func1);
+    bm->Teardown(func2);
+    assert(func1 != nullptr);
+    assert(func2 != nullptr);
+  }
+  {
+    auto func = [](const benchmark::State& /*unused*/){};
+    bm->Setup(func);
+    bm->Teardown(func);
+    assert(func != nullptr);
+  }
+  {
+    auto func1 = [](const benchmark::State& /*unused*/){};
+    auto func2 = func1;
+    bm->Setup(std::move(func1));
+    bm->Teardown(std::move(func2));
+  }
+}