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
|
// Copyright 2024 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/rlz/chrome_rlz_tracker_web_contents_observer.h"
#include "chrome/test/base/chrome_render_view_host_test_harness.h"
#include "components/rlz/mock_rlz_tracker_delegate.h"
#include "components/rlz/rlz_tracker.h"
using ::testing::_;
using ::testing::Return;
class ChromeRLZTrackerWebContentsObserverTest
: public ChromeRenderViewHostTestHarness {
public:
rlz::MockRLZTrackerDelegate* delegate() { return delegate_; }
protected:
void SetUp() override {
ChromeRenderViewHostTestHarness::SetUp();
auto delegate = std::make_unique<rlz::MockRLZTrackerDelegate>();
delegate_ = delegate.get();
rlz::RLZTracker::SetRlzDelegate(std::move(delegate));
}
void TearDown() override {
rlz::RLZTracker::SetRlzChromeHomePageSearchRecordedForTesting(false);
delegate_ = nullptr;
rlz::RLZTracker::ClearRlzDelegateForTesting();
ChromeRenderViewHostTestHarness::TearDown();
}
private:
raw_ptr<rlz::MockRLZTrackerDelegate> delegate_;
};
TEST_F(ChromeRLZTrackerWebContentsObserverTest, PerformHomepageSearch) {
EXPECT_CALL(*delegate(), GetBrand(_)).WillRepeatedly(Return(true));
EXPECT_CALL(*delegate(), IsBrandOrganic(_)).WillRepeatedly(Return(false));
ChromeRLZTrackerWebContentsObserver::CreateForWebContentsIfNeeded(
web_contents());
ChromeRLZTrackerWebContentsObserver* observer =
ChromeRLZTrackerWebContentsObserver::FromWebContents(web_contents());
EXPECT_TRUE(observer);
// Search callback is not run for an invalid navigation.
EXPECT_CALL(*delegate(), RunHomepageSearchCallback()).Times(0);
NavigateAndCommit(
GURL("https://www.google.com"),
static_cast<ui::PageTransition>(ui::PAGE_TRANSITION_LINK |
ui::PAGE_TRANSITION_HOME_PAGE));
task_environment()->RunUntilIdle();
// Search callback is run for a valid navigation.
EXPECT_CALL(*delegate(), RunHomepageSearchCallback());
NavigateAndCommit(GURL("https://www.google.com/search?q=test"));
task_environment()->RunUntilIdle();
// Observer has been removed after a valid search.
observer =
ChromeRLZTrackerWebContentsObserver::FromWebContents(web_contents());
EXPECT_FALSE(observer);
}
TEST_F(ChromeRLZTrackerWebContentsObserverTest,
NotCreateObserverForEmptyBrand) {
EXPECT_CALL(*delegate(), GetBrand(_)).WillRepeatedly(Return(false));
ChromeRLZTrackerWebContentsObserver::CreateForWebContentsIfNeeded(
web_contents());
ChromeRLZTrackerWebContentsObserver* observer =
ChromeRLZTrackerWebContentsObserver::FromWebContents(web_contents());
EXPECT_FALSE(observer);
}
TEST_F(ChromeRLZTrackerWebContentsObserverTest,
NotCreateObserverForOrganicBrand) {
EXPECT_CALL(*delegate(), GetBrand(_)).WillRepeatedly(Return(true));
EXPECT_CALL(*delegate(), IsBrandOrganic(_)).WillRepeatedly(Return(true));
ChromeRLZTrackerWebContentsObserver::CreateForWebContentsIfNeeded(
web_contents());
ChromeRLZTrackerWebContentsObserver* observer =
ChromeRLZTrackerWebContentsObserver::FromWebContents(web_contents());
EXPECT_FALSE(observer);
}
TEST_F(ChromeRLZTrackerWebContentsObserverTest,
NotCreateObserverIfSearchRecorded) {
EXPECT_CALL(*delegate(), GetBrand(_)).WillRepeatedly(Return(true));
EXPECT_CALL(*delegate(), IsBrandOrganic(_)).WillRepeatedly(Return(false));
rlz::RLZTracker::SetRlzChromeHomePageSearchRecordedForTesting(true);
ChromeRLZTrackerWebContentsObserver::CreateForWebContentsIfNeeded(
web_contents());
ChromeRLZTrackerWebContentsObserver* observer =
ChromeRLZTrackerWebContentsObserver::FromWebContents(web_contents());
EXPECT_FALSE(observer);
}
TEST_F(ChromeRLZTrackerWebContentsObserverTest,
RemoveObserverOnNavigationIfSearchRecorded) {
EXPECT_CALL(*delegate(), GetBrand(_)).WillRepeatedly(Return(true));
EXPECT_CALL(*delegate(), IsBrandOrganic(_)).WillRepeatedly(Return(false));
ChromeRLZTrackerWebContentsObserver::CreateForWebContentsIfNeeded(
web_contents());
ChromeRLZTrackerWebContentsObserver* observer =
ChromeRLZTrackerWebContentsObserver::FromWebContents(web_contents());
EXPECT_TRUE(observer);
// Simulate that the search has been performed in other web contents.
rlz::RLZTracker::SetRlzChromeHomePageSearchRecordedForTesting(true);
// Navigating the web contents will remove the observer.
EXPECT_CALL(*delegate(), RunHomepageSearchCallback()).Times(0);
NavigateAndCommit(GURL("https://www.google.com/search?q=test"));
task_environment()->RunUntilIdle();
observer =
ChromeRLZTrackerWebContentsObserver::FromWebContents(web_contents());
EXPECT_FALSE(observer);
}
|
