lldb/unittests/Utility/TimerTest.cpp

   1 //===-- TimerTest.cpp -----------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "lldb/Utility/StreamString.h"
  10 #include "lldb/Utility/Timer.h"
  11 #include "gtest/gtest.h"
  12 #include <thread>
  13
  14 using namespace lldb_private;
  15
  16 TEST(TimerTest, CategoryTimes) {
  17   Timer::ResetCategoryTimes();
  18   {
  19     static Timer::Category tcat("CAT1");
  20     Timer t(tcat, ".");
  21     std::this_thread::sleep_for(std::chrono::milliseconds(10));
  22   }
  23   StreamString ss;
  24   Timer::DumpCategoryTimes(ss);
  25   double seconds;
  26   ASSERT_EQ(1, sscanf(ss.GetData(), "%lf sec for CAT1", &seconds));
  27   EXPECT_LT(0.001, seconds);
  28   EXPECT_GT(0.1, seconds);
  29 }
  30
  31 TEST(TimerTest, CategoryTimesNested) {
  32   Timer::ResetCategoryTimes();
  33   {
  34     static Timer::Category tcat1("CAT1");
  35     Timer t1(tcat1, ".");
  36     std::this_thread::sleep_for(std::chrono::milliseconds(10));
  37     // Explicitly testing the same category as above.
  38     Timer t2(tcat1, ".");
  39     std::this_thread::sleep_for(std::chrono::milliseconds(10));
  40   }
  41   StreamString ss;
  42   Timer::DumpCategoryTimes(ss);
  43   double seconds;
  44   // It should only appear once.
  45   ASSERT_EQ(ss.GetString().count("CAT1"), 1U);
  46   ASSERT_EQ(1, sscanf(ss.GetData(), "%lf sec for CAT1", &seconds));
  47   EXPECT_LT(0.002, seconds);
  48   EXPECT_GT(0.2, seconds);
  49 }
  50
  51 TEST(TimerTest, CategoryTimes2) {
  52   Timer::ResetCategoryTimes();
  53   {
  54     static Timer::Category tcat1("CAT1");
  55     Timer t1(tcat1, ".");
  56     std::this_thread::sleep_for(std::chrono::milliseconds(100));
  57     static Timer::Category tcat2("CAT2");
  58     Timer t2(tcat2, ".");
  59     std::this_thread::sleep_for(std::chrono::milliseconds(10));
  60   }
  61   StreamString ss;
  62   Timer::DumpCategoryTimes(ss);
  63   double seconds1, seconds2;
  64   ASSERT_EQ(2, sscanf(ss.GetData(),
  65                       "%lf sec (total: %*fs; child: %*fs; count: %*d) for "
  66                       "CAT1%*[\n ]%lf sec for CAT2",
  67                       &seconds1, &seconds2))
  68       << "String: " << ss.GetData();
  69   EXPECT_LT(0.01, seconds1);
  70   EXPECT_GT(1, seconds1);
  71   EXPECT_LT(0.001, seconds2);
  72   EXPECT_GT(0.1, seconds2);
  73 }
  74
  75 TEST(TimerTest, CategoryTimesStats) {
  76   Timer::ResetCategoryTimes();
  77   {
  78     static Timer::Category tcat1("CAT1");
  79     Timer t1(tcat1, ".");
  80     std::this_thread::sleep_for(std::chrono::milliseconds(100));
  81     static Timer::Category tcat2("CAT2");
  82     {
  83       Timer t2(tcat2, ".");
  84       std::this_thread::sleep_for(std::chrono::milliseconds(10));
  85     }
  86     {
  87       Timer t3(tcat2, ".");
  88       std::this_thread::sleep_for(std::chrono::milliseconds(10));
  89     }
  90   }
  91   // Example output:
  92   // 0.105202764 sec (total: 0.132s; child: 0.027s; count: 1) for CAT1
  93   // 0.026772798 sec (total: 0.027s; child: 0.000s; count: 2) for CAT2
  94   StreamString ss;
  95   Timer::DumpCategoryTimes(ss);
  96   double seconds1, total1, child1, seconds2;
  97   int count1, count2;
  98   ASSERT_EQ(
  99       6, sscanf(ss.GetData(),
 100                 "%lf sec (total: %lfs; child: %lfs; count: %d) for CAT1%*[\n\r ]"
 101                 "%lf sec (total: %*fs; child: %*fs; count: %d) for CAT2",
 102                 &seconds1, &total1, &child1, &count1, &seconds2, &count2))
 103       << "String: " << ss.GetData();
 104   EXPECT_NEAR(total1 - child1, seconds1, 0.002);
 105   EXPECT_EQ(1, count1);
 106   EXPECT_NEAR(child1, seconds2, 0.002);
 107   EXPECT_EQ(2, count2);
 108 }