1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/metrics/daily_event.h"
#include <optional>
#include "base/memory/ptr_util.h"
#include "base/memory/raw_ptr.h"
#include "base/test/task_environment.h"
#include "components/prefs/testing_pref_service.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace metrics {
namespace {
const char kTestPrefName[] = "TestPref";
const char kTestMetricName[] = "TestMetric";
class TestDailyObserver : public DailyEvent::Observer {
public:
TestDailyObserver() = default;
TestDailyObserver(const TestDailyObserver&) = delete;
TestDailyObserver& operator=(const TestDailyObserver&) = delete;
bool fired() const { return type_.has_value(); }
DailyEvent::IntervalType type() const { return type_.value(); }
void OnDailyEvent(DailyEvent::IntervalType type) override { type_ = type; }
void Reset() { type_ = {}; }
private:
// Last-received type, or unset if OnDailyEvent() hasn't been called.
std::optional<DailyEvent::IntervalType> type_;
};
class DailyEventTest : public testing::Test {
public:
DailyEventTest() : event_(&prefs_, kTestPrefName, kTestMetricName) {
DailyEvent::RegisterPref(prefs_.registry(), kTestPrefName);
auto observer = std::make_unique<TestDailyObserver>();
observer_ = observer.get();
event_.AddObserver(std::move(observer));
}
DailyEventTest(const DailyEventTest&) = delete;
DailyEventTest& operator=(const DailyEventTest&) = delete;
protected:
TestingPrefServiceSimple prefs_;
DailyEvent event_; // Owns and outlives `observer_`
raw_ptr<TestDailyObserver> observer_;
base::test::TaskEnvironment env_{
base::test::TaskEnvironment::TimeSource::MOCK_TIME};
};
} // namespace
// The event should fire if the preference is not available.
TEST_F(DailyEventTest, TestNewFires) {
event_.CheckInterval();
ASSERT_TRUE(observer_->fired());
EXPECT_EQ(DailyEvent::IntervalType::FIRST_RUN, observer_->type());
}
// The event should fire if the preference is more than a day old.
TEST_F(DailyEventTest, TestOldFires) {
base::Time last_time = base::Time::Now() - base::Hours(25);
prefs_.SetInt64(kTestPrefName, last_time.since_origin().InMicroseconds());
event_.CheckInterval();
ASSERT_TRUE(observer_->fired());
EXPECT_EQ(DailyEvent::IntervalType::DAY_ELAPSED, observer_->type());
}
// The event should fire if the preference is more than a day in the future.
TEST_F(DailyEventTest, TestFutureFires) {
base::Time last_time = base::Time::Now() + base::Hours(25);
prefs_.SetInt64(kTestPrefName, last_time.since_origin().InMicroseconds());
event_.CheckInterval();
ASSERT_TRUE(observer_->fired());
EXPECT_EQ(DailyEvent::IntervalType::CLOCK_CHANGED, observer_->type());
}
// The event should not fire if the preference is more recent than a day.
TEST_F(DailyEventTest, TestRecentNotFired) {
base::Time last_time = base::Time::Now() - base::Minutes(2);
prefs_.SetInt64(kTestPrefName, last_time.since_origin().InMicroseconds());
event_.CheckInterval();
EXPECT_FALSE(observer_->fired());
}
// The event should not fire if the preference is less than a day in the future.
TEST_F(DailyEventTest, TestSoonNotFired) {
base::Time last_time = base::Time::Now() + base::Minutes(2);
prefs_.SetInt64(kTestPrefName, last_time.since_origin().InMicroseconds());
event_.CheckInterval();
EXPECT_FALSE(observer_->fired());
}
void TestCallback(bool* fired) {
*fired = true;
}
TEST_F(DailyEventTest, TestClosureFired) {
// Verify the event fires on the first check.
bool fired = false;
event_.AddObserverClosure(base::BindRepeating(&TestCallback, &fired));
event_.CheckInterval();
EXPECT_TRUE(fired);
EXPECT_EQ(DailyEvent::IntervalType::FIRST_RUN, observer_->type());
// Reset the flag. A check on the same day should not fire the event.
fired = false;
event_.CheckInterval();
EXPECT_FALSE(fired);
// Advance time by 25 hours to verify the event fires again.
env_.FastForwardBy(base::Hours(25));
event_.CheckInterval();
EXPECT_TRUE(fired);
EXPECT_EQ(DailyEvent::IntervalType::DAY_ELAPSED, observer_->type());
}
} // namespace metrics
|
