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 2023 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/tabs/organization/trigger.h"
#include <memory>
#include "chrome/browser/ui/browser_window/test/mock_browser_window_interface.h"
#include "chrome/browser/ui/tabs/organization/trigger_policies.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/browser/ui/tabs/test_tab_strip_model_delegate.h"
#include "chrome/browser/ui/tabs/test_util.h"
#include "chrome/browser/ui/ui_features.h"
#include "chrome/test/base/testing_profile.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/test_renderer_host.h"
#include "content/public/test/web_contents_tester.h"
#include "testing/gtest/include/gtest/gtest.h"
class TriggerTest : public testing::Test {
public:
TriggerTest()
: profile_(new TestingProfile),
delegate_(new TestTabStripModelDelegate),
tab_strip_model_(new TabStripModel(delegate(), profile())),
browser_window_interface_(new MockBrowserWindowInterface()) {
ON_CALL(*browser_window_interface_, GetTabStripModel)
.WillByDefault(::testing::Return(tab_strip_model_.get()));
delegate_->SetBrowserWindowInterface(browser_window_interface_.get());
}
~TriggerTest() override {
// Break loop so we can deconstruct without dangling pointers.
delegate_->SetBrowserWindowInterface(nullptr);
}
TestingProfile* profile() { return profile_.get(); }
TestTabStripModelDelegate* delegate() { return delegate_.get(); }
TabStripModel* tab_strip_model() { return tab_strip_model_.get(); }
std::unique_ptr<content::WebContents> CreateWebContents() {
return content::WebContentsTester::CreateTestWebContents(profile(),
nullptr);
}
base::test::ScopedFeatureList& scoped_feature_list() {
return scoped_feature_list_;
}
content::WebContents* AddTab(GURL url) {
std::unique_ptr<content::WebContents> contents_unique_ptr =
CreateWebContents();
content::WebContents* content_ptr = contents_unique_ptr.get();
content::WebContentsTester::For(contents_unique_ptr.get())
->NavigateAndCommit(url);
tab_strip_model()->AppendWebContents(std::move(contents_unique_ptr), true);
return content_ptr;
}
private:
content::BrowserTaskEnvironment task_environment_;
content::RenderViewHostTestEnabler rvh_test_enabler_;
const std::unique_ptr<TestingProfile> profile_;
const std::unique_ptr<TestTabStripModelDelegate> delegate_;
const std::unique_ptr<TabStripModel> tab_strip_model_;
const std::unique_ptr<MockBrowserWindowInterface> browser_window_interface_;
tabs::PreventTabFeatureInitialization prevent_;
base::test::ScopedFeatureList scoped_feature_list_;
};
TEST_F(TriggerTest, TriggerHappyPath) {
auto trigger = std::make_unique<TabOrganizationTrigger>(
GetTriggerScoringFunction(), GetTriggerScoreThreshold(),
std::make_unique<DemoTriggerPolicy>());
// Should not trigger under the score threshold.
EXPECT_FALSE(trigger->ShouldTrigger(tab_strip_model()));
// Should trigger the first time over the score threshold.
for (int i = 0; i < 10; i++) {
AddTab(GURL("https://www.example.com"));
}
EXPECT_TRUE(trigger->ShouldTrigger(tab_strip_model()));
// Should trigger every time (because DemoTriggerPolicy).
EXPECT_TRUE(trigger->ShouldTrigger(tab_strip_model()));
}
TEST_F(TriggerTest, DoesntTriggerForEnterprise) {
scoped_feature_list().InitAndDisableFeature(
features::kTabOrganizationEnableNudgeForEnterprise);
auto trigger = std::make_unique<TabOrganizationTrigger>(
GetTriggerScoringFunction(), GetTriggerScoreThreshold(),
std::make_unique<DemoTriggerPolicy>());
// Should trigger when over the score threshold for non-enterprise.
for (int i = 0; i < 10; i++) {
AddTab(GURL("https://www.example.com"));
}
EXPECT_TRUE(trigger->ShouldTrigger(tab_strip_model()));
// Shouldn't trigger under the same conditions for enterprise
TestingProfile::Builder enterprise_profile_builder;
enterprise_profile_builder.SetIsSupervisedProfile();
std::unique_ptr<TestingProfile> enterprise_profile =
enterprise_profile_builder.Build();
std::unique_ptr<TabStripModel> enterprise_tab_strip_model(
std::make_unique<TabStripModel>(delegate(), enterprise_profile.get()));
auto enterprise_trigger = std::make_unique<TabOrganizationTrigger>(
GetTriggerScoringFunction(), GetTriggerScoreThreshold(),
GetTriggerPolicy(nullptr, enterprise_profile.get()));
EXPECT_FALSE(
enterprise_trigger->ShouldTrigger(enterprise_tab_strip_model.get()));
}
|
