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
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/pdf/infobar/pdf_infobar_prefs.h"
#include "base/time/time.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/scoped_testing_local_state.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile.h"
#include "components/prefs/testing_pref_service.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace pdf::infobar {
class PdfInfoBarPrefsTest : public testing::Test {
protected:
PdfInfoBarPrefsTest() : profile_(std::make_unique<TestingProfile>()) {}
TestingPrefServiceSimple* local_state() { return local_state_.Get(); }
TestingProfile* profile() { return profile_.get(); }
void FastForwardBy(base::TimeDelta time) {
task_environment_.FastForwardBy(time);
}
private:
// Must be the first member.
content::BrowserTaskEnvironment task_environment_{
base::test::TaskEnvironment::TimeSource::MOCK_TIME};
// Must be before `profile_`.
ScopedTestingLocalState local_state_{TestingBrowserProcess::GetGlobal()};
const std::unique_ptr<TestingProfile> profile_;
};
TEST_F(PdfInfoBarPrefsTest, SetInfoBarShownRecently) {
SetInfoBarShownRecently();
EXPECT_EQ(1, local_state()->GetInteger(prefs::kPdfInfoBarTimesShown));
EXPECT_EQ(base::Time::Now(),
local_state()->GetTime(prefs::kPdfInfoBarLastShown));
FastForwardBy(base::Hours(1));
SetInfoBarShownRecently();
EXPECT_EQ(2, local_state()->GetInteger(prefs::kPdfInfoBarTimesShown));
EXPECT_EQ(base::Time::Now(),
local_state()->GetTime(prefs::kPdfInfoBarLastShown));
}
TEST_F(PdfInfoBarPrefsTest, InfoBarShownRecentlyOrMaxTimes) {
EXPECT_FALSE(InfoBarShownRecentlyOrMaxTimes());
local_state()->SetInteger(prefs::kPdfInfoBarTimesShown, 1);
local_state()->SetTime(prefs::kPdfInfoBarLastShown, base::Time::Now());
EXPECT_TRUE(InfoBarShownRecentlyOrMaxTimes());
}
TEST_F(PdfInfoBarPrefsTest, InfoBarShownRecentlyOrMaxTimesExponential) {
EXPECT_FALSE(InfoBarShownRecentlyOrMaxTimes());
SetInfoBarShownRecently();
EXPECT_TRUE(InfoBarShownRecentlyOrMaxTimes());
// The infobar was shown once, so it shouldn't be shown again for
// `kPdfInfoBarShowIntervalDays` days.
FastForwardBy(base::Days(kPdfInfoBarShowIntervalDays - 1));
EXPECT_TRUE(InfoBarShownRecentlyOrMaxTimes());
FastForwardBy(base::Days(1));
EXPECT_FALSE(InfoBarShownRecentlyOrMaxTimes());
SetInfoBarShownRecently();
// The infobar was shown twice, so it shouldn't be shown again for
// `kPdfInfoBarShowIntervalDays * 2` days.
FastForwardBy(base::Days((kPdfInfoBarShowIntervalDays * 2) - 1));
EXPECT_TRUE(InfoBarShownRecentlyOrMaxTimes());
FastForwardBy(base::Days(1));
EXPECT_FALSE(InfoBarShownRecentlyOrMaxTimes());
SetInfoBarShownRecently();
// The infobar was shown three times, so it shouldn't be shown again for
// `kPdfInfoBarShowIntervalDays * 4` days.
FastForwardBy(base::Days((kPdfInfoBarShowIntervalDays * 4) - 1));
EXPECT_TRUE(InfoBarShownRecentlyOrMaxTimes());
FastForwardBy(base::Days(1));
EXPECT_FALSE(InfoBarShownRecentlyOrMaxTimes());
}
TEST_F(PdfInfoBarPrefsTest, InfoBarShownRecentlyOrMaxTimesMaxReached) {
local_state()->SetInteger(prefs::kPdfInfoBarTimesShown,
kPdfInfoBarMaxTimesToShow);
local_state()->SetTime(prefs::kPdfInfoBarLastShown,
base::Time::Now() - base::Days(3650));
// The infobar has been shown the max number of times, so it shouldn't be
// shown again regardless of how much time has passed.
EXPECT_TRUE(InfoBarShownRecentlyOrMaxTimes());
}
TEST_F(PdfInfoBarPrefsTest, IsPdfViewerDisabled) {
EXPECT_FALSE(IsPdfViewerDisabled(profile()));
profile()->GetPrefs()->SetBoolean(prefs::kPluginsAlwaysOpenPdfExternally,
false);
EXPECT_FALSE(IsPdfViewerDisabled(profile()));
profile()->GetPrefs()->SetBoolean(prefs::kPluginsAlwaysOpenPdfExternally,
true);
EXPECT_TRUE(IsPdfViewerDisabled(profile()));
}
TEST_F(PdfInfoBarPrefsTest, IsDefaultBrowserPolicyControlled) {
EXPECT_FALSE(IsDefaultBrowserPolicyControlled());
local_state()->SetManagedPref(prefs::kDefaultBrowserSettingEnabled,
base::Value(false));
EXPECT_FALSE(IsDefaultBrowserPolicyControlled());
local_state()->SetManagedPref(prefs::kDefaultBrowserSettingEnabled,
base::Value(true));
EXPECT_TRUE(IsDefaultBrowserPolicyControlled());
}
} // namespace pdf::infobar
|
