Imported Upstream version 1.41.0

[platform/upstream/grpc.git] / test / cpp / microbenchmarks / bm_alarm.cc

diff --git a/test/cpp/microbenchmarks/bm_alarm.cc b/test/cpp/microbenchmarks/bm_alarm.cc
index d95771a..83899d1 100644 (file)
--- a/test/cpp/microbenchmarks/bm_alarm.cc
+++ b/test/cpp/microbenchmarks/bm_alarm.cc
@@ -19,10 +19,12 @@
 /* This benchmark exists to ensure that immediately-firing alarms are fast */
 
 #include <benchmark/benchmark.h>
+
 #include <grpc/grpc.h>
 #include <grpcpp/alarm.h>
 #include <grpcpp/completion_queue.h>
 #include <grpcpp/impl/grpc_library.h>
+
 #include "test/core/util/test_config.h"
 #include "test/cpp/microbenchmarks/helpers.h"
 #include "test/cpp/util/test_config.h"
@@ -37,7 +39,7 @@ static void BM_Alarm_Tag_Immediate(benchmark::State& state) {
   void* output_tag;
   bool ok;
   auto deadline = grpc_timeout_seconds_to_deadline(0);
-  while (state.KeepRunning()) {
+  for (auto _ : state) {
     alarm.Set(&cq, deadline, nullptr);
     cq.Next(&output_tag, &ok);
   }
@@ -55,6 +57,7 @@ void RunTheBenchmarksNamespaced() { RunSpecifiedBenchmarks(); }
 }  // namespace benchmark
 
 int main(int argc, char** argv) {
+  grpc::testing::TestEnvironment env(argc, argv);
   LibraryInitializer libInit;
   ::benchmark::Initialize(&argc, argv);
   ::grpc::testing::InitTest(&argc, &argv, false);