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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
// Copyright 2021 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/hats/hats_helper.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/page_info/merchant_trust_service_factory.h"
#include "chrome/browser/ui/hats/hats_service_factory.h"
#include "chrome/browser/ui/hats/mock_hats_service.h"
#include "chrome/browser/ui/hats/mock_trust_safety_sentiment_service.h"
#include "chrome/browser/ui/hats/trust_safety_sentiment_service_factory.h"
#include "chrome/common/webui_url_constants.h"
#include "chrome/test/base/testing_profile.h"
#include "components/page_info/core/merchant_trust_service.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/navigation_simulator.h"
#include "content/public/test/test_renderer_host.h"
#include "content/public/test/web_contents_tester.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/features.h"
namespace {} // namespace
class MockMerchantTrustService : public page_info::MerchantTrustService {
public:
MockMerchantTrustService()
: MerchantTrustService(nullptr, nullptr, false, nullptr) {}
MOCK_METHOD(void, MaybeShowEvaluationSurvey, (), (override));
};
std::unique_ptr<KeyedService> BuildMockMerchantTrustService(
content::BrowserContext* context) {
return std::make_unique<::testing::NiceMock<MockMerchantTrustService>>();
}
class HatsHelperTest : public testing::Test {
public:
HatsHelperTest() {
mock_hats_service_ = static_cast<MockHatsService*>(
HatsServiceFactory::GetInstance()->SetTestingFactoryAndUse(
profile(), base::BindRepeating(&BuildMockHatsService)));
EXPECT_CALL(*mock_hats_service(), CanShowAnySurvey(testing::_))
.WillRepeatedly(testing::Return(true));
mock_sentiment_service_ = static_cast<MockTrustSafetySentimentService*>(
TrustSafetySentimentServiceFactory::GetInstance()
->SetTestingFactoryAndUse(
profile(),
base::BindRepeating(&BuildMockTrustSafetySentimentService)));
mock_merchant_trust_service_ = static_cast<MockMerchantTrustService*>(
MerchantTrustServiceFactory::GetInstance()->SetTestingFactoryAndUse(
profile(), base::BindRepeating(&BuildMockMerchantTrustService)));
web_contents_ =
content::WebContentsTester::CreateTestWebContents(profile(), nullptr);
HatsHelper::CreateForWebContents(web_contents_.get());
}
protected:
TestingProfile* profile() { return &profile_; }
MockHatsService* mock_hats_service() { return mock_hats_service_; }
MockTrustSafetySentimentService* mock_sentiment_service() {
return mock_sentiment_service_;
}
MockMerchantTrustService* mock_merchant_trust_service() {
return mock_merchant_trust_service_;
}
content::WebContentsTester* test_web_contents() {
return content::WebContentsTester::For(web_contents_.get());
}
content::RenderFrameHost* main_rfh() {
return web_contents_->GetPrimaryMainFrame();
}
private:
content::BrowserTaskEnvironment task_environment_;
TestingProfile profile_;
content::RenderViewHostTestEnabler rvh_test_enabler_;
std::unique_ptr<content::WebContents> web_contents_;
raw_ptr<MockHatsService> mock_hats_service_;
raw_ptr<MockTrustSafetySentimentService> mock_sentiment_service_;
raw_ptr<MockMerchantTrustService> mock_merchant_trust_service_;
};
TEST_F(HatsHelperTest, SentimentServiceInformed) {
// Check that the helper correctly informs the sentiment service that the
// user has visited the NTP.
EXPECT_CALL(*mock_sentiment_service(), OpenedNewTabPage());
test_web_contents()->NavigateAndCommit(GURL(chrome::kChromeUINewTabURL));
testing::Mock::VerifyAndClearExpectations(mock_hats_service());
// Navigations to non-NTP pages should not inform the service.
EXPECT_CALL(*mock_sentiment_service(), OpenedNewTabPage()).Times(0);
test_web_contents()->NavigateAndCommit(GURL("https://unrelated.com"));
}
TEST_F(HatsHelperTest, MerchantTrustSurveyAttempted) {
// Check that the helper attempts to show the survey.
EXPECT_CALL(*mock_merchant_trust_service(), MaybeShowEvaluationSurvey());
test_web_contents()->NavigateAndCommit(GURL(chrome::kChromeUINewTabURL));
testing::Mock::VerifyAndClearExpectations(mock_merchant_trust_service());
// Navigations to non-NTP pages should not inform the service.
EXPECT_CALL(*mock_merchant_trust_service(), MaybeShowEvaluationSurvey())
.Times(0);
test_web_contents()->NavigateAndCommit(GURL("https://unrelated.com"));
}
class HatsHelperFencedFrameTest : public HatsHelperTest {
public:
HatsHelperFencedFrameTest() {
scoped_feature_list_.InitAndEnableFeatureWithParameters(
blink::features::kFencedFrames, {{"implementation_type", "mparch"}});
}
~HatsHelperFencedFrameTest() override = default;
content::RenderFrameHost* CreateFencedFrame(
content::RenderFrameHost* parent) {
content::RenderFrameHost* fenced_frame =
content::RenderFrameHostTester::For(parent)->AppendFencedFrame();
return fenced_frame;
}
private:
base::test::ScopedFeatureList scoped_feature_list_;
};
TEST_F(HatsHelperFencedFrameTest,
SentimentServiceIsNotInformedAfterNavigationInFencedFrame) {
// Check that the helper correctly informs the sentiment service that the
// user has visited the NTP on the main frame.
EXPECT_CALL(*mock_sentiment_service(), OpenedNewTabPage());
test_web_contents()->NavigateAndCommit(GURL(chrome::kChromeUINewTabURL));
testing::Mock::VerifyAndClearExpectations(mock_hats_service());
// Navigations in fenced frames should not inform the service.
EXPECT_CALL(*mock_sentiment_service(), OpenedNewTabPage()).Times(0);
content::RenderFrameHostTester::For(main_rfh())
->InitializeRenderFrameIfNeeded();
content::RenderFrameHost* fenced_frame_rfh = CreateFencedFrame(main_rfh());
GURL fenced_frame_url = GURL("https://unrelated.com");
std::unique_ptr<content::NavigationSimulator> navigation_simulator =
content::NavigationSimulator::CreateRendererInitiated(fenced_frame_url,
fenced_frame_rfh);
navigation_simulator->Commit();
fenced_frame_rfh = navigation_simulator->GetFinalRenderFrameHost();
EXPECT_TRUE(fenced_frame_rfh->IsFencedFrameRoot());
}
|
