Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2017 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 "components/feature_engagement_tracker/internal/test/event_util.h" | |
| 6 | |
| 7 #include <map> | |
|
nyquist
2017/05/11 17:08:03
Nit: Did you end up using this include? Maybe I'm
David Trainor- moved to gerrit
2017/05/11 19:30:28
Done.
| |
| 8 | |
| 9 #include "testing/gtest/include/gtest/gtest.h" | |
| 10 | |
| 11 namespace feature_engagement_tracker { | |
| 12 namespace test { | |
| 13 | |
| 14 void SetEventCountForDay(Event* event, uint32_t day, uint32_t count) { | |
| 15 Event_Count* event_count = event->add_events(); | |
| 16 event_count->set_day(day); | |
| 17 event_count->set_count(count); | |
| 18 } | |
| 19 | |
| 20 void VerifyEventCount(const Event* event, uint32_t day, uint32_t count) { | |
| 21 bool found_day = false; | |
| 22 for (int i = 0; i < event->events_size(); ++i) { | |
|
nyquist
2017/05/11 17:08:03
Nit: I know you just moved this around, but should
David Trainor- moved to gerrit
2017/05/11 19:30:28
Done.
| |
| 23 Event_Count event_count = event->events(i); | |
| 24 if (event_count.day() == day) { | |
| 25 EXPECT_FALSE(found_day); | |
| 26 found_day = true; | |
| 27 EXPECT_EQ(count, event_count.count()); | |
| 28 } | |
| 29 } | |
| 30 EXPECT_TRUE(found_day); | |
| 31 } | |
| 32 | |
| 33 void VerifyEventsEqual(const Event* a, const Event* b) { | |
| 34 if (!a || !b) { | |
| 35 // If one of the events are nullptr, both should be nullptr. | |
| 36 ASSERT_EQ(a, b); | |
| 37 return; | |
| 38 } | |
| 39 | |
| 40 EXPECT_EQ(a->name(), b->name()); | |
| 41 EXPECT_EQ(a->events_size(), b->events_size()); | |
| 42 for (int i = 0; i < a->events_size(); ++i) { | |
| 43 VerifyEventCount(b, a->events(i).day(), a->events(i).count()); | |
| 44 } | |
| 45 } | |
| 46 | |
| 47 } // namespace test | |
| 48 } // namespace feature_engagement_tracker | |
| OLD | NEW |