blob: 9a2e2a1743580d3bfd567f78f2142bcd7c614571 [file] [log] [blame]
stevenjb@chromium.org3199bd12012-11-15 06:03:29 +09001// 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
11namespace dbus {
12
13class 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
63TEST_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
92TEST_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
129TEST_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