src/dbus/dbus_statistics_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "dbus/dbus_statistics.h"
   6
   7 #include "base/basictypes.h"
   8 #include "base/compiler_specific.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace dbus {
  12
  13 class DBusStatisticsTest : public testing::Test {
  14  public:
  15   DBusStatisticsTest() {
  16   }
  17
  18   virtual void SetUp() override {
  19     statistics::Initialize();
  20   }
  21
  22   virtual void TearDown() override {
  23     statistics::Shutdown();
  24   }
  25
  26  protected:
  27   void AddTestMethodCalls() {
  28     statistics::AddSentMethodCall(
  29         "service1", "service1.interface1", "method1");
  30     statistics::AddReceivedSignal(
  31         "service1", "service1.interface1", "method1");
  32     statistics::AddBlockingSentMethodCall(
  33         "service1", "service1.interface1", "method1");
  34
  35     statistics::AddSentMethodCall(
  36         "service1", "service1.interface1", "method2");
  37     statistics::AddSentMethodCall(
  38         "service1", "service1.interface1", "method2");
  39     statistics::AddReceivedSignal(
  40         "service1", "service1.interface1", "method2");
  41
  42     statistics::AddSentMethodCall(
  43         "service1", "service1.interface1", "method3");
  44     statistics::AddSentMethodCall(
  45         "service1", "service1.interface1", "method3");
  46     statistics::AddSentMethodCall(
  47         "service1", "service1.interface1", "method3");
  48
  49     statistics::AddSentMethodCall(
  50         "service1", "service1.interface2", "method1");
  51
  52     statistics::AddSentMethodCall(
  53         "service1", "service1.interface2", "method2");
  54
  55     statistics::AddSentMethodCall(
  56         "service2", "service2.interface1", "method1");
  57   }
  58
  59  private:
  60   DISALLOW_COPY_AND_ASSIGN(DBusStatisticsTest);
  61 };
  62
  63 TEST_F(DBusStatisticsTest, TestDBusStatsBasic) {
  64   int sent = 0, received = 0, block = 0;
  65
  66   // Add a sent call
  67   statistics::AddSentMethodCall("service1", "service1.interface1", "method1");
  68   ASSERT_TRUE(statistics::testing::GetCalls(
  69       "service1", "service1.interface1", "method1", &sent, &received, &block));
  70   EXPECT_EQ(1, sent);
  71   EXPECT_EQ(0, received);
  72   EXPECT_EQ(0, block);
  73
  74   // Add a received call
  75   statistics::AddReceivedSignal("service1", "service1.interface1", "method1");
  76   ASSERT_TRUE(statistics::testing::GetCalls(
  77       "service1", "service1.interface1", "method1", &sent, &received, &block));
  78   EXPECT_EQ(1, sent);
  79   EXPECT_EQ(1, received);
  80   EXPECT_EQ(0, block);
  81
  82   // Add a block call
  83   statistics::AddBlockingSentMethodCall(
  84       "service1", "service1.interface1", "method1");
  85   ASSERT_TRUE(statistics::testing::GetCalls(
  86       "service1", "service1.interface1", "method1", &sent, &received, &block));
  87   EXPECT_EQ(1, sent);
  88   EXPECT_EQ(1, received);
  89   EXPECT_EQ(1, block);
  90 }
  91
  92 TEST_F(DBusStatisticsTest, TestDBusStatsMulti) {
  93   int sent = 0, received = 0, block = 0;
  94
  95   // Add some more stats to exercise accessing multiple different stats.
  96   AddTestMethodCalls();
  97
  98   // Make sure all entries can be found in the set and their counts were
  99   // incremented correctly.
 100   ASSERT_TRUE(statistics::testing::GetCalls(
 101       "service1", "service1.interface1", "method1", &sent, &received, &block));
 102   EXPECT_EQ(1, sent);
 103   EXPECT_EQ(1, received);
 104   ASSERT_TRUE(statistics::testing::GetCalls(
 105       "service1", "service1.interface1", "method2", &sent, &received, &block));
 106   EXPECT_EQ(2, sent);
 107   EXPECT_EQ(1, received);
 108   ASSERT_TRUE(statistics::testing::GetCalls(
 109       "service1", "service1.interface1", "method3", &sent, &received, &block));
 110   EXPECT_EQ(3, sent);
 111   EXPECT_EQ(0, received);
 112   ASSERT_TRUE(statistics::testing::GetCalls(
 113       "service1", "service1.interface2", "method1", &sent, &received, &block));
 114   EXPECT_EQ(1, sent);
 115   EXPECT_EQ(0, received);
 116   ASSERT_TRUE(statistics::testing::GetCalls(
 117       "service1", "service1.interface2", "method2", &sent, &received, &block));
 118   EXPECT_EQ(1, sent);
 119   EXPECT_EQ(0, received);
 120   ASSERT_TRUE(statistics::testing::GetCalls(
 121       "service2", "service2.interface1", "method1", &sent, &received, &block));
 122   EXPECT_EQ(1, sent);
 123   EXPECT_EQ(0, received);
 124
 125   ASSERT_FALSE(statistics::testing::GetCalls(
 126       "service1", "service1.interface3", "method2", &sent, &received, &block));
 127 }
 128
 129 TEST_F(DBusStatisticsTest, TestGetAsString) {
 130   std::string output_none = GetAsString(statistics::SHOW_SERVICE,
 131                                         statistics::FORMAT_TOTALS);
 132   EXPECT_EQ("No DBus calls.", output_none);
 133
 134   AddTestMethodCalls();
 135
 136   std::string output_service = GetAsString(statistics::SHOW_SERVICE,
 137                                            statistics::FORMAT_TOTALS);
 138   const std::string expected_output_service(
 139       "service1: Sent (BLOCKING): 1 Sent: 8 Received: 2\n"
 140       "service2: Sent: 1\n");
 141   EXPECT_EQ(expected_output_service, output_service);
 142
 143   std::string output_interface = GetAsString(statistics::SHOW_INTERFACE,
 144                                              statistics::FORMAT_TOTALS);
 145   const std::string expected_output_interface(
 146       "service1.interface1: Sent (BLOCKING): 1 Sent: 6 Received: 2\n"
 147       "service1.interface2: Sent: 2\n"
 148       "service2.interface1: Sent: 1\n");
 149   EXPECT_EQ(expected_output_interface, output_interface);
 150
 151   std::string output_per_minute = GetAsString(statistics::SHOW_INTERFACE,
 152                                               statistics::FORMAT_PER_MINUTE);
 153   const std::string expected_output_per_minute(
 154       "service1.interface1: Sent (BLOCKING): 1/min Sent: 6/min"
 155       " Received: 2/min\n"
 156       "service1.interface2: Sent: 2/min\n"
 157       "service2.interface1: Sent: 1/min\n");
 158   EXPECT_EQ(expected_output_per_minute, output_per_minute);
 159
 160   std::string output_all = GetAsString(statistics::SHOW_INTERFACE,
 161                                        statistics::FORMAT_ALL);
 162   const std::string expected_output_all(
 163       "service1.interface1: Sent (BLOCKING): 1 (1/min) Sent: 6 (6/min)"
 164       " Received: 2 (2/min)\n"
 165       "service1.interface2: Sent: 2 (2/min)\n"
 166       "service2.interface1: Sent: 1 (1/min)\n");
 167   EXPECT_EQ(expected_output_all, output_all);
 168
 169
 170   std::string output_method = GetAsString(statistics::SHOW_METHOD,
 171                                           statistics::FORMAT_TOTALS);
 172   const std::string expected_output_method(
 173       "service1.interface1.method1: Sent (BLOCKING): 1 Sent: 1 Received: 1\n"
 174       "service1.interface1.method2: Sent: 2 Received: 1\n"
 175       "service1.interface1.method3: Sent: 3\n"
 176       "service1.interface2.method1: Sent: 1\n"
 177       "service1.interface2.method2: Sent: 1\n"
 178       "service2.interface1.method1: Sent: 1\n");
 179   EXPECT_EQ(expected_output_method, output_method);
 180
 181 }
 182
 183 }  // namespace dbus