OLD | NEW |
| (Empty) |
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 <algorithm> | |
6 #include <string> | |
7 #include <vector> | |
8 | |
9 #include "base/files/file_path.h" | |
10 #include "base/files/scoped_temp_dir.h" | |
11 #include "base/memory/scoped_ptr.h" | |
12 #include "base/time/time.h" | |
13 #include "chrome/browser/performance_monitor/database.h" | |
14 #include "chrome/browser/performance_monitor/key_builder.h" | |
15 #include "chrome/browser/performance_monitor/metric.h" | |
16 #include "chrome/browser/performance_monitor/performance_monitor_util.h" | |
17 #include "extensions/common/extension.h" | |
18 #include "extensions/common/manifest.h" | |
19 #include "testing/gtest/include/gtest/gtest.h" | |
20 #include "third_party/leveldatabase/src/include/leveldb/db.h" | |
21 #include "third_party/leveldatabase/src/include/leveldb/iterator.h" | |
22 | |
23 using extensions::Extension; | |
24 using extensions::Manifest; | |
25 | |
26 namespace performance_monitor { | |
27 | |
28 // A class which is friended by Database, in order to hold the private methods | |
29 // and avoid friending all the different test classes. | |
30 class DatabaseTestHelper { | |
31 public: | |
32 explicit DatabaseTestHelper(Database* database) : database_(database) { } | |
33 ~DatabaseTestHelper() { } | |
34 | |
35 bool Close() { return database_->Close(); } | |
36 | |
37 // Override the check for a metric's validity and insert it in the database. | |
38 // Note: This does not do extraneous updates, like updating the recent_db or | |
39 // the max_value_map. | |
40 bool AddInvalidMetric(std::string activity, Metric metric) { | |
41 std::string metric_key = | |
42 database_->key_builder_->CreateMetricKey(database_->clock_->GetTime(), | |
43 metric.type, | |
44 activity); | |
45 leveldb::Status status = | |
46 database_->metric_db_->Put(database_->write_options_, | |
47 metric_key, | |
48 metric.ValueAsString()); | |
49 return status.ok(); | |
50 } | |
51 | |
52 // Writes an invalid event to the database; since events are stored as JSON | |
53 // strings, this is equivalent to writing a garbage string. | |
54 bool AddInvalidEvent(base::Time time, EventType type) { | |
55 std::string key = database_->key_builder_->CreateEventKey(time, type); | |
56 leveldb::Status status = | |
57 database_->event_db_->Put(database_->write_options_, key, "fake_event"); | |
58 return status.ok(); | |
59 } | |
60 | |
61 size_t GetNumberOfMetricEntries() { | |
62 return GetNumberOfEntries(database_->metric_db_.get()); | |
63 } | |
64 size_t GetNumberOfEventEntries() { | |
65 return GetNumberOfEntries(database_->event_db_.get()); | |
66 } | |
67 | |
68 private: | |
69 // Returns the number of entries in a given database. | |
70 size_t GetNumberOfEntries(leveldb::DB* level_db) { | |
71 size_t number_of_entries = 0; | |
72 scoped_ptr<leveldb::Iterator> iter( | |
73 level_db->NewIterator(database_->read_options_)); | |
74 for (iter->SeekToFirst(); iter->Valid(); iter->Next()) | |
75 ++number_of_entries; | |
76 return number_of_entries; | |
77 } | |
78 | |
79 Database* database_; | |
80 }; | |
81 | |
82 // A clock that increments every access. Great for testing. | |
83 class TestingClock : public Database::Clock { | |
84 public: | |
85 TestingClock() | |
86 : counter_(0) { | |
87 } | |
88 explicit TestingClock(const TestingClock& other) | |
89 : counter_(other.counter_) { | |
90 } | |
91 virtual ~TestingClock() {} | |
92 virtual base::Time GetTime() OVERRIDE { | |
93 return base::Time::FromInternalValue(++counter_); | |
94 } | |
95 private: | |
96 int64 counter_; | |
97 }; | |
98 | |
99 class PerformanceMonitorDatabaseEventTest : public ::testing::Test { | |
100 protected: | |
101 PerformanceMonitorDatabaseEventTest() { | |
102 clock_ = new TestingClock(); | |
103 CHECK(temp_dir_.CreateUniqueTempDir()); | |
104 db_ = Database::Create(temp_dir_.path()); | |
105 CHECK(db_.get()); | |
106 db_->set_clock(scoped_ptr<Database::Clock>(clock_)); | |
107 } | |
108 | |
109 virtual void SetUp() { | |
110 ASSERT_TRUE(db_.get()); | |
111 PopulateDB(); | |
112 } | |
113 | |
114 void PopulateDB() { | |
115 InitEvents(); | |
116 db_->AddEvent(*install_event_1_.get()); | |
117 db_->AddEvent(*install_event_2_.get()); | |
118 db_->AddEvent(*uninstall_event_1_.get()); | |
119 db_->AddEvent(*uninstall_event_2_.get()); | |
120 } | |
121 | |
122 scoped_ptr<Database> db_; | |
123 Database::Clock* clock_; | |
124 base::ScopedTempDir temp_dir_; | |
125 scoped_ptr<Event> install_event_1_; | |
126 scoped_ptr<Event> install_event_2_; | |
127 scoped_ptr<Event> uninstall_event_1_; | |
128 scoped_ptr<Event> uninstall_event_2_; | |
129 | |
130 private: | |
131 void InitEvents() { | |
132 install_event_1_ = util::CreateExtensionEvent( | |
133 EVENT_EXTENSION_INSTALL, clock_->GetTime(), "a", "extension 1", | |
134 "http://foo.com", static_cast<int>(Manifest::UNPACKED), "0.1", | |
135 "Test Test"); | |
136 install_event_2_ = util::CreateExtensionEvent( | |
137 EVENT_EXTENSION_INSTALL, clock_->GetTime(), "b", "extension 2", | |
138 "http://bar.com", static_cast<int>(Manifest::UNPACKED), "0.1", | |
139 "Test Test"); | |
140 uninstall_event_1_ = util::CreateExtensionEvent( | |
141 EVENT_EXTENSION_UNINSTALL, clock_->GetTime(), "a", "extension 1", | |
142 "http://foo.com", static_cast<int>(Manifest::UNPACKED), "0.1", | |
143 "Test Test"); | |
144 uninstall_event_2_ = util::CreateExtensionEvent( | |
145 EVENT_EXTENSION_UNINSTALL, clock_->GetTime(), "b", "extension 2", | |
146 "http://bar.com", static_cast<int>(Manifest::UNPACKED), "0.1", | |
147 "Test Test"); | |
148 } | |
149 }; | |
150 | |
151 class PerformanceMonitorDatabaseMetricTest : public ::testing::Test { | |
152 protected: | |
153 PerformanceMonitorDatabaseMetricTest() { | |
154 clock_ = new TestingClock(); | |
155 CHECK(temp_dir_.CreateUniqueTempDir()); | |
156 db_ = Database::Create(temp_dir_.path()); | |
157 CHECK(db_.get()); | |
158 db_->set_clock(scoped_ptr<Database::Clock>(clock_)); | |
159 activity_ = std::string("A"); | |
160 } | |
161 | |
162 virtual void SetUp() { | |
163 ASSERT_TRUE(db_.get()); | |
164 PopulateDB(); | |
165 } | |
166 | |
167 void PopulateDB() { | |
168 db_->AddMetric(kProcessChromeAggregate, | |
169 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 50.5)); | |
170 db_->AddMetric(activity_, | |
171 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 13.1)); | |
172 db_->AddMetric(kProcessChromeAggregate, | |
173 Metric(METRIC_PRIVATE_MEMORY_USAGE, | |
174 clock_->GetTime(), | |
175 1000000.0)); | |
176 db_->AddMetric(activity_, | |
177 Metric(METRIC_PRIVATE_MEMORY_USAGE, | |
178 clock_->GetTime(), | |
179 3000000.0)); | |
180 } | |
181 | |
182 scoped_ptr<Database> db_; | |
183 Database::Clock* clock_; | |
184 base::ScopedTempDir temp_dir_; | |
185 std::string activity_; | |
186 }; | |
187 | |
188 ////// PerformanceMonitorDatabaseSetupTests //////////////////////////////////// | |
189 TEST(PerformanceMonitorDatabaseSetupTest, OpenClose) { | |
190 base::ScopedTempDir temp_dir; | |
191 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | |
192 scoped_ptr<Database> db = Database::Create(temp_dir.path()); | |
193 ASSERT_TRUE(db.get()); | |
194 DatabaseTestHelper helper(db.get()); | |
195 ASSERT_TRUE(helper.Close()); | |
196 } | |
197 | |
198 TEST(PerformanceMonitorDatabaseSetupTest, ActiveInterval) { | |
199 base::ScopedTempDir temp_dir; | |
200 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | |
201 | |
202 TestingClock* clock_1 = new TestingClock(); | |
203 base::Time start_time = clock_1->GetTime(); | |
204 | |
205 scoped_ptr<Database> db_1 = Database::Create(temp_dir.path()); | |
206 db_1->set_clock(scoped_ptr<Database::Clock>(clock_1)); | |
207 DatabaseTestHelper helper1(db_1.get()); | |
208 db_1->AddStateValue("test", "test"); | |
209 ASSERT_TRUE(db_1.get()); | |
210 ASSERT_TRUE(helper1.Close()); | |
211 | |
212 TestingClock* clock_2 = new TestingClock(*clock_1); | |
213 base::Time mid_time = clock_2->GetTime(); | |
214 scoped_ptr<Database> db_2 = Database::Create(temp_dir.path()); | |
215 db_2->set_clock(scoped_ptr<Database::Clock>(clock_2)); | |
216 DatabaseTestHelper helper2(db_2.get()); | |
217 db_2->AddStateValue("test", "test"); | |
218 ASSERT_TRUE(db_2.get()); | |
219 ASSERT_TRUE(helper2.Close()); | |
220 | |
221 TestingClock* clock_3 = new TestingClock(*clock_2); | |
222 base::Time end_time = clock_3->GetTime(); | |
223 scoped_ptr<Database> db_3 = Database::Create(temp_dir.path()); | |
224 db_3->set_clock(scoped_ptr<Database::Clock>(clock_3)); | |
225 db_3->AddStateValue("test", "test"); | |
226 ASSERT_TRUE(db_3.get()); | |
227 | |
228 std::vector<TimeRange> active_interval = db_3->GetActiveIntervals(start_time, | |
229 end_time); | |
230 ASSERT_EQ(active_interval.size(), 2u); | |
231 ASSERT_TRUE(active_interval[0].start > start_time && | |
232 active_interval[0].end < mid_time); | |
233 ASSERT_TRUE(active_interval[1].start > mid_time && | |
234 active_interval[1].end < end_time); | |
235 } | |
236 | |
237 TEST(PerformanceMonitorDatabaseSetupTest, | |
238 ActiveIntervalRetrievalDuringActiveInterval) { | |
239 base::ScopedTempDir temp_dir; | |
240 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | |
241 | |
242 TestingClock* clock = new TestingClock(); | |
243 scoped_ptr<Database> db = Database::Create(temp_dir.path()); | |
244 db->set_clock(scoped_ptr<Database::Clock>(clock)); | |
245 db->AddStateValue("test", "test"); | |
246 base::Time start_time = clock->GetTime(); | |
247 db->AddStateValue("test", "test"); | |
248 base::Time end_time = clock->GetTime(); | |
249 db->AddStateValue("test", "test"); | |
250 | |
251 std::vector<TimeRange> active_interval = db->GetActiveIntervals(start_time, | |
252 end_time); | |
253 ASSERT_EQ(1u, active_interval.size()); | |
254 EXPECT_LT(active_interval[0].start, start_time); | |
255 EXPECT_GT(active_interval[0].end, start_time); | |
256 } | |
257 | |
258 TEST(PerformanceMonitorDatabaseSetupTest, | |
259 ActiveIntervalRetrievalAfterActiveInterval) { | |
260 base::ScopedTempDir temp_dir; | |
261 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | |
262 | |
263 TestingClock* clock = new TestingClock(); | |
264 scoped_ptr<Database> db = Database::Create(temp_dir.path()); | |
265 db->set_clock(scoped_ptr<Database::Clock>(clock)); | |
266 db->AddStateValue("test", "test"); | |
267 | |
268 base::Time start_time = clock->GetTime(); | |
269 base::Time end_time = clock->GetTime(); | |
270 std::vector<TimeRange> active_interval = db->GetActiveIntervals(start_time, | |
271 end_time); | |
272 EXPECT_TRUE(active_interval.empty()); | |
273 } | |
274 | |
275 ////// PerformanceMonitorDatabaseEventTests //////////////////////////////////// | |
276 TEST_F(PerformanceMonitorDatabaseEventTest, GetAllEvents) { | |
277 Database::EventVector events = db_->GetEvents(); | |
278 ASSERT_EQ(4u, events.size()); | |
279 EXPECT_TRUE(events[0]->data()->Equals(install_event_1_->data())); | |
280 EXPECT_TRUE(events[1]->data()->Equals(install_event_2_->data())); | |
281 EXPECT_TRUE(events[2]->data()->Equals(uninstall_event_1_->data())); | |
282 EXPECT_TRUE(events[3]->data()->Equals(uninstall_event_2_->data())); | |
283 } | |
284 | |
285 TEST_F(PerformanceMonitorDatabaseMetricTest, GetMaxMetric) { | |
286 Metric stat; | |
287 EXPECT_EQ(0.0, db_->GetMaxStatsForActivityAndMetric(activity_, | |
288 METRIC_PAGE_LOAD_TIME)); | |
289 | |
290 EXPECT_EQ(1000000, | |
291 db_->GetMaxStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE)); | |
292 | |
293 db_->AddMetric(kProcessChromeAggregate, | |
294 Metric(METRIC_PRIVATE_MEMORY_USAGE, clock_->GetTime(), 99.0)); | |
295 EXPECT_EQ(1000000, | |
296 db_->GetMaxStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE)); | |
297 | |
298 db_->AddMetric(kProcessChromeAggregate, | |
299 Metric(METRIC_PRIVATE_MEMORY_USAGE, | |
300 clock_->GetTime(), | |
301 6000000.0)); | |
302 EXPECT_EQ(6000000, | |
303 db_->GetMaxStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE)); | |
304 } | |
305 | |
306 TEST_F(PerformanceMonitorDatabaseEventTest, GetAllEventTypes) { | |
307 Database::EventTypeSet types = db_->GetEventTypes(); | |
308 ASSERT_EQ(2u, types.size()); | |
309 ASSERT_EQ(1u, types.count(EVENT_EXTENSION_INSTALL)); | |
310 ASSERT_EQ(1u, types.count(EVENT_EXTENSION_UNINSTALL)); | |
311 } | |
312 | |
313 TEST_F(PerformanceMonitorDatabaseEventTest, GetEventInTimeRange) { | |
314 base::Time start_time = clock_->GetTime(); | |
315 scoped_ptr<Event> crash_event = util::CreateRendererFailureEvent( | |
316 clock_->GetTime(), EVENT_RENDERER_CRASH, "http://www.google.com/"); | |
317 db_->AddEvent(*crash_event.get()); | |
318 Database::EventVector events = db_->GetEvents(start_time, clock_->GetTime()); | |
319 ASSERT_EQ(1u, events.size()); | |
320 EXPECT_TRUE(events[0]->data()->Equals(crash_event->data())); | |
321 } | |
322 | |
323 TEST_F(PerformanceMonitorDatabaseEventTest, GetInstallEvents) { | |
324 Database::EventVector events = db_->GetEvents(EVENT_EXTENSION_INSTALL); | |
325 ASSERT_EQ(2u, events.size()); | |
326 EXPECT_TRUE(events[0]->data()->Equals(install_event_1_->data())); | |
327 EXPECT_TRUE(events[1]->data()->Equals(install_event_2_->data())); | |
328 } | |
329 | |
330 TEST_F(PerformanceMonitorDatabaseEventTest, InvalidEvents) { | |
331 DatabaseTestHelper helper(db_.get()); | |
332 | |
333 // Insert an invalid event into the database; verify it is inserted. | |
334 size_t original_number_of_entries = helper.GetNumberOfEventEntries(); | |
335 ASSERT_TRUE(helper.AddInvalidEvent( | |
336 clock_->GetTime(), EVENT_EXTENSION_INSTALL)); | |
337 ASSERT_EQ(original_number_of_entries + 1u, helper.GetNumberOfEventEntries()); | |
338 | |
339 // Should not retrieve the invalid event. | |
340 Database::EventVector events = db_->GetEvents(); | |
341 ASSERT_EQ(original_number_of_entries, events.size()); | |
342 | |
343 // Invalid event should have been deleted. | |
344 ASSERT_EQ(original_number_of_entries, helper.GetNumberOfEventEntries()); | |
345 } | |
346 | |
347 TEST_F(PerformanceMonitorDatabaseEventTest, GetUnusedEventType) { | |
348 Database::EventVector events = db_->GetEvents(EVENT_EXTENSION_DISABLE); | |
349 ASSERT_TRUE(events.empty()); | |
350 events = db_->GetEvents(EVENT_EXTENSION_DISABLE, clock_->GetTime(), | |
351 clock_->GetTime()); | |
352 ASSERT_TRUE(events.empty()); | |
353 } | |
354 | |
355 TEST_F(PerformanceMonitorDatabaseEventTest, GetEventsTimeRange) { | |
356 base::Time start_time = clock_->GetTime(); | |
357 scoped_ptr<Event> new_install_event = | |
358 util::CreateExtensionEvent(EVENT_EXTENSION_INSTALL, clock_->GetTime(), | |
359 "c", "test extension", "http://foo.com", | |
360 static_cast<int>(Manifest::UNPACKED), "0.1", | |
361 "Test Test"); | |
362 scoped_ptr<Event> new_uninstall_event = | |
363 util::CreateExtensionEvent(EVENT_EXTENSION_UNINSTALL, clock_->GetTime(), | |
364 "c", "test extension", "http://foo.com", | |
365 static_cast<int>(Manifest::UNPACKED), "0.1", | |
366 "Test Test"); | |
367 base::Time end_time = clock_->GetTime(); | |
368 db_->AddEvent(*new_install_event.get()); | |
369 db_->AddEvent(*new_uninstall_event.get()); | |
370 Database::EventVector events = db_->GetEvents(start_time, end_time); | |
371 ASSERT_EQ(2u, events.size()); | |
372 EXPECT_TRUE(events[0]->data()->Equals(new_install_event->data())); | |
373 EXPECT_TRUE(events[1]->data()->Equals(new_uninstall_event->data())); | |
374 events = db_->GetEvents( | |
375 EVENT_EXTENSION_INSTALL, start_time, end_time); | |
376 ASSERT_EQ(1u, events.size()); | |
377 EXPECT_TRUE(events[0]->data()->Equals(new_install_event->data())); | |
378 } | |
379 | |
380 ////// PerformanceMonitorDatabaseMetricTests /////////////////////////////////// | |
381 TEST_F(PerformanceMonitorDatabaseMetricTest, GetActiveMetrics) { | |
382 Database::MetricTypeSet active_metrics = | |
383 db_->GetActiveMetrics(base::Time(), clock_->GetTime()); | |
384 | |
385 Database::MetricTypeSet expected_metrics; | |
386 expected_metrics.insert(METRIC_CPU_USAGE); | |
387 expected_metrics.insert(METRIC_PRIVATE_MEMORY_USAGE); | |
388 EXPECT_EQ(expected_metrics, active_metrics); | |
389 } | |
390 | |
391 TEST_F(PerformanceMonitorDatabaseMetricTest, GetRecentMetric) { | |
392 Metric stat; | |
393 ASSERT_TRUE(db_->GetRecentStatsForActivityAndMetric(activity_, | |
394 METRIC_PRIVATE_MEMORY_USAGE, &stat)); | |
395 EXPECT_EQ(3000000, stat.value); | |
396 | |
397 ASSERT_TRUE(db_->GetRecentStatsForActivityAndMetric(METRIC_CPU_USAGE, &stat)); | |
398 EXPECT_EQ(50.5, stat.value); | |
399 | |
400 base::ScopedTempDir dir; | |
401 CHECK(dir.CreateUniqueTempDir()); | |
402 scoped_ptr<Database> db = Database::Create(dir.path()); | |
403 CHECK(db.get()); | |
404 db->set_clock(scoped_ptr<Database::Clock>(new TestingClock())); | |
405 EXPECT_FALSE(db->GetRecentStatsForActivityAndMetric(METRIC_CPU_USAGE, &stat)); | |
406 } | |
407 | |
408 TEST_F(PerformanceMonitorDatabaseMetricTest, GetState) { | |
409 std::string key("version"); | |
410 std::string value("1.0.0.0.1"); | |
411 db_->AddStateValue(key, value); | |
412 EXPECT_EQ(db_->GetStateValue(key), value); | |
413 } | |
414 | |
415 TEST_F(PerformanceMonitorDatabaseMetricTest, GetStateOverride) { | |
416 std::string key("version"); | |
417 std::string value_1("1.0.0.0.0"); | |
418 std::string value_2("1.0.0.0.1"); | |
419 db_->AddStateValue(key, value_1); | |
420 db_->AddStateValue(key, value_2); | |
421 EXPECT_EQ(db_->GetStateValue(key), value_2); | |
422 } | |
423 | |
424 TEST_F(PerformanceMonitorDatabaseMetricTest, GetStatsForActivityAndMetric) { | |
425 Database::MetricVector stats = *db_->GetStatsForActivityAndMetric( | |
426 activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime()); | |
427 ASSERT_EQ(1u, stats.size()); | |
428 EXPECT_EQ(13.1, stats[0].value); | |
429 base::Time before = clock_->GetTime(); | |
430 db_->AddMetric(activity_, | |
431 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 18.0)); | |
432 stats = *db_->GetStatsForActivityAndMetric(activity_, METRIC_CPU_USAGE, | |
433 before, clock_->GetTime()); | |
434 ASSERT_EQ(1u, stats.size()); | |
435 EXPECT_EQ(18, stats[0].value); | |
436 stats = *db_->GetStatsForActivityAndMetric(activity_, METRIC_CPU_USAGE); | |
437 ASSERT_EQ(2u, stats.size()); | |
438 EXPECT_EQ(13.1, stats[0].value); | |
439 EXPECT_EQ(18, stats[1].value); | |
440 stats = *db_->GetStatsForActivityAndMetric(METRIC_PRIVATE_MEMORY_USAGE); | |
441 ASSERT_EQ(1u, stats.size()); | |
442 EXPECT_EQ(1000000, stats[0].value); | |
443 stats = *db_->GetStatsForActivityAndMetric(activity_, METRIC_CPU_USAGE, | |
444 clock_->GetTime(), | |
445 clock_->GetTime()); | |
446 EXPECT_TRUE(stats.empty()); | |
447 } | |
448 | |
449 TEST_F(PerformanceMonitorDatabaseMetricTest, GetStatsForMetricByActivity) { | |
450 Database::MetricVectorMap stats_map = db_->GetStatsForMetricByActivity( | |
451 METRIC_CPU_USAGE, base::Time(), clock_->GetTime()); | |
452 ASSERT_EQ(2u, stats_map.size()); | |
453 linked_ptr<Database::MetricVector> stats = stats_map[activity_]; | |
454 ASSERT_EQ(1u, stats->size()); | |
455 EXPECT_EQ(13.1, stats->at(0).value); | |
456 stats = stats_map[kProcessChromeAggregate]; | |
457 ASSERT_EQ(1u, stats->size()); | |
458 EXPECT_EQ(50.5, stats->at(0).value); | |
459 stats_map = db_->GetStatsForMetricByActivity( | |
460 METRIC_CPU_USAGE, clock_->GetTime(), clock_->GetTime()); | |
461 EXPECT_EQ(0u, stats_map.size()); | |
462 base::Time before = clock_->GetTime(); | |
463 db_->AddMetric(activity_, | |
464 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 18.0)); | |
465 stats_map = db_->GetStatsForMetricByActivity(METRIC_CPU_USAGE, before, | |
466 clock_->GetTime()); | |
467 ASSERT_EQ(1u, stats_map.size()); | |
468 stats = stats_map[activity_]; | |
469 ASSERT_EQ(1u, stats->size()); | |
470 EXPECT_EQ(18, stats->at(0).value); | |
471 } | |
472 | |
473 TEST_F(PerformanceMonitorDatabaseMetricTest, InvalidMetrics) { | |
474 DatabaseTestHelper helper(db_.get()); | |
475 Metric invalid_metric(METRIC_CPU_USAGE, clock_->GetTime(), -5.0); | |
476 ASSERT_FALSE(invalid_metric.IsValid()); | |
477 | |
478 // Find the original number of entries in the database. | |
479 size_t original_number_of_entries = helper.GetNumberOfMetricEntries(); | |
480 Database::MetricVector stats = *db_->GetStatsForActivityAndMetric( | |
481 activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime()); | |
482 size_t original_number_of_cpu_entries = stats.size(); | |
483 | |
484 // Check that the database normally refuses to insert bad metrics. | |
485 EXPECT_FALSE(db_->AddMetric(invalid_metric)); | |
486 | |
487 // Verify that it was not inserted into the database. | |
488 stats = *db_->GetStatsForActivityAndMetric( | |
489 activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime()); | |
490 ASSERT_EQ(original_number_of_cpu_entries, stats.size()); | |
491 | |
492 // Forcefully insert it into the database. | |
493 ASSERT_TRUE(helper.AddInvalidMetric(activity_, invalid_metric)); | |
494 ASSERT_EQ(original_number_of_entries + 1u, helper.GetNumberOfMetricEntries()); | |
495 | |
496 // Try to retrieve it; should only get one result. | |
497 stats = *db_->GetStatsForActivityAndMetric( | |
498 activity_, METRIC_CPU_USAGE, base::Time(), clock_->GetTime()); | |
499 ASSERT_EQ(original_number_of_cpu_entries, stats.size()); | |
500 | |
501 // Entry should have been deleted in the database. | |
502 ASSERT_EQ(original_number_of_entries, helper.GetNumberOfMetricEntries()); | |
503 } | |
504 | |
505 TEST_F(PerformanceMonitorDatabaseMetricTest, GetFullRange) { | |
506 db_->AddMetric(kProcessChromeAggregate, | |
507 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 3.4)); | |
508 db_->AddMetric(kProcessChromeAggregate, | |
509 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 21.0)); | |
510 Database::MetricVector stats = | |
511 *db_->GetStatsForActivityAndMetric(METRIC_CPU_USAGE); | |
512 ASSERT_EQ(3u, stats.size()); | |
513 ASSERT_EQ(50.5, stats[0].value); | |
514 ASSERT_EQ(3.4, stats[1].value); | |
515 ASSERT_EQ(21, stats[2].value); | |
516 } | |
517 | |
518 TEST_F(PerformanceMonitorDatabaseMetricTest, GetRange) { | |
519 base::Time start = clock_->GetTime(); | |
520 db_->AddMetric(kProcessChromeAggregate, | |
521 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 3.0)); | |
522 db_->AddMetric(kProcessChromeAggregate, | |
523 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 9.0)); | |
524 base::Time end = clock_->GetTime(); | |
525 db_->AddMetric(kProcessChromeAggregate, | |
526 Metric(METRIC_CPU_USAGE, clock_->GetTime(), 21.0)); | |
527 Database::MetricVector stats = | |
528 *db_->GetStatsForActivityAndMetric(METRIC_CPU_USAGE, start, end); | |
529 ASSERT_EQ(2u, stats.size()); | |
530 ASSERT_EQ(3, stats[0].value); | |
531 ASSERT_EQ(9, stats[1].value); | |
532 } | |
533 | |
534 } // namespace performance_monitor | |
OLD | NEW |