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
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_USER_EDUCATION_TEST_USER_EDUCATION_SESSION_MOCKS_H_
#define COMPONENTS_USER_EDUCATION_TEST_USER_EDUCATION_SESSION_MOCKS_H_
#include "base/callback_list.h"
#include "components/user_education/common/feature_promo/feature_promo_session_policy.h"
#include "components/user_education/common/session/user_education_idle_observer.h"
#include "components/user_education/common/session/user_education_idle_policy.h"
#include "components/user_education/common/session/user_education_session_manager.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace user_education::test {
// Version of `IdleObserver` that returns a provided state and which sends state
// updates only when `UpdateState()` is called.
class TestIdleObserver : public UserEducationIdleObserver {
public:
explicit TestIdleObserver(std::optional<base::Time> initial_active_time);
~TestIdleObserver() override;
// Call to modify the current state and optionally send an update.
void SetLastActiveTime(std::optional<base::Time> new_last_active_time,
bool send_update);
private:
// UserEducationIdleObserver:
void StartObserving() final;
std::optional<base::Time> MaybeGetNewLastActiveTime() const final;
std::optional<base::Time> last_active_time_;
};
// Mock version of `IdlePolicy` that allows specific queries to be intercepted.
class MockIdlePolicy : public UserEducationIdlePolicy {
public:
MockIdlePolicy();
~MockIdlePolicy() override;
MOCK_METHOD(bool,
IsNewSession,
(base::Time, base::Time, base::Time),
(const, override));
};
// Fakes the behavior of a session provider.
class TestUserEducationSessionProvider : public UserEducationSessionProvider {
public:
explicit TestUserEducationSessionProvider(bool has_new_session);
~TestUserEducationSessionProvider() override;
// UserEducationSessionProvider:
base::CallbackListSubscription AddNewSessionCallback(
base::RepeatingClosure new_session_callback) override;
bool GetNewSessionSinceStartup() const override;
// Starts a new session.
void StartNewSession();
private:
base::RepeatingClosureList callbacks_;
bool has_new_session_;
};
// Mock version of `UserEducationSessionManager` that can monitor when updates
// or new sessions happen.
class MockUserEducationSessionManager : public UserEducationSessionManager {
public:
MockUserEducationSessionManager();
~MockUserEducationSessionManager() override;
MOCK_METHOD(void, OnLastActiveTimeUpdating, (base::Time), (override));
MOCK_METHOD(void,
OnNewSession,
(const base::Time, const base::Time, const base::Time),
(override));
};
// Allows for overriding of `CanShowPromo()`. Other methods work as they
// currently do in order to ensure that data is written to storage. Note that
// a session manager is not necessary and may be passed to `Init()` as
// `nullptr`.
class MockFeaturePromoSessionPolicy : public FeaturePromoSessionPolicy {
public:
MockFeaturePromoSessionPolicy();
~MockFeaturePromoSessionPolicy() override;
MOCK_METHOD(FeaturePromoResult,
CanShowPromo,
(PromoPriorityInfo to_show,
std::optional<PromoPriorityInfo> currently_showing),
(const override));
};
} // namespace user_education::test
#endif // COMPONENTS_USER_EDUCATION_TEST_USER_EDUCATION_SESSION_MOCKS_H_
|
