1 // Copyright 2020 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/test/scoped_feature_list.h"
6 #include "build/build_config.h"
7 #include "chrome/browser/navigation_predictor/navigation_predictor_keyed_service.h"
8 #include "chrome/browser/navigation_predictor/navigation_predictor_keyed_service_factory.h"
9 #include "chrome/browser/profiles/profile.h"
10 #include "chrome/browser/ui/browser.h"
11 #include "chrome/browser/ui/browser_tabstrip.h"
12 #include "chrome/browser/ui/tabs/tab_strip_model.h"
13 #include "chrome/test/base/in_process_browser_test.h"
14 #include "chrome/test/base/ui_test_utils.h"
15 #include "components/ukm/test_ukm_recorder.h"
16 #include "content/public/browser/render_process_host.h"
17 #include "content/public/test/browser_test.h"
18 #include "content/public/test/browser_test_base.h"
19 #include "content/public/test/browser_test_utils.h"
20 #include "content/public/test/test_utils.h"
21 #include "services/metrics/public/cpp/metrics_utils.h"
22 #include "services/metrics/public/cpp/ukm_builders.h"
23 #include "services/metrics/public/cpp/ukm_recorder.h"
24 #include "services/metrics/public/cpp/ukm_source.h"
25 #include "testing/gtest/include/gtest/gtest.h"
26 #include "url/gurl.h"
27
28 namespace {
29 const base::Feature kNavigationPredictorRendererWarmup{
30 "NavigationPredictorRendererWarmup", base::FEATURE_DISABLED_BY_DEFAULT};
31 }
32
33 // Occasional flakes on Windows (https://crbug.com/1045971).
34 #if defined(OS_WIN) || defined(OS_MAC) || defined(OS_CHROMEOS)
35 #define DISABLE_ON_WIN_MAC_CHROMEOS(x) DISABLED_##x
36 #else
37 #define DISABLE_ON_WIN_MAC_CHROMEOS(x) x
38 #endif
39
40 class NavigationPredictorRendererWarmupClientBrowserTest
41 : public InProcessBrowserTest {
42 public:
43 NavigationPredictorRendererWarmupClientBrowserTest() = default;
44 ~NavigationPredictorRendererWarmupClientBrowserTest() override = default;
45
SetUp()46 void SetUp() override {
47 scoped_feature_list_.InitAndEnableFeature(
48 kNavigationPredictorRendererWarmup);
49 InProcessBrowserTest::SetUp();
50 }
51
SetUpOnMainThread()52 void SetUpOnMainThread() override {
53 InProcessBrowserTest::SetUpOnMainThread();
54 ukm_recorder_ = std::make_unique<ukm::TestAutoSetUkmRecorder>();
55 }
56
SpareRendererCount() const57 size_t SpareRendererCount() const {
58 size_t count = 0;
59 for (content::RenderProcessHost::iterator iter(
60 content::RenderProcessHost::AllHostsIterator());
61 !iter.IsAtEnd(); iter.Advance()) {
62 if (iter.GetCurrentValue()->IsUnused()) {
63 count++;
64 }
65 }
66 return count;
67 }
68
MakeSpareRenderer()69 void MakeSpareRenderer() {
70 content::RenderProcessHost::WarmupSpareRenderProcessHost(
71 browser()->profile());
72 }
73
GetWebContents() const74 content::WebContents* GetWebContents() const {
75 return browser()->tab_strip_model()->GetActiveWebContents();
76 }
77
MakeEligibleNavigationPrediction()78 void MakeEligibleNavigationPrediction() {
79 NavigationPredictorKeyedServiceFactory::GetForProfile(browser()->profile())
80 ->OnPredictionUpdated(
81 GetWebContents(), GURL("https://www.google.com/search?q=test"),
82 NavigationPredictorKeyedService::PredictionSource::
83 kAnchorElementsParsedFromWebPage,
84 {});
85 }
86
VerifyHasUKMEntry(const GURL & url)87 void VerifyHasUKMEntry(const GURL& url) {
88 auto entries = ukm_recorder_->GetEntriesByName(
89 ukm::builders::NavigationPredictorRendererWarmup::kEntryName);
90 ASSERT_EQ(1U, entries.size());
91
92 const auto* entry = entries.front();
93 ukm_recorder_->ExpectEntrySourceHasUrl(entry, url);
94 }
95
96 private:
97 base::test::ScopedFeatureList scoped_feature_list_;
98 std::unique_ptr<ukm::TestAutoSetUkmRecorder> ukm_recorder_;
99 };
100
IN_PROC_BROWSER_TEST_F(NavigationPredictorRendererWarmupClientBrowserTest,DISABLE_ON_WIN_MAC_CHROMEOS (StartsSpareRenderer))101 IN_PROC_BROWSER_TEST_F(NavigationPredictorRendererWarmupClientBrowserTest,
102 DISABLE_ON_WIN_MAC_CHROMEOS(StartsSpareRenderer)) {
103 // Navigate to a site so that the default renderer is used.
104 embedded_test_server()->ServeFilesFromSourceDirectory("chrome/test/data");
105 ASSERT_TRUE(embedded_test_server()->Start());
106
107 GURL url = embedded_test_server()->GetURL("/simple.html");
108
109 ui_test_utils::NavigateToURL(browser(), url);
110 MakeEligibleNavigationPrediction();
111 base::RunLoop().RunUntilIdle();
112
113 EXPECT_EQ(SpareRendererCount(), 1U);
114 VerifyHasUKMEntry(url);
115 }
116
IN_PROC_BROWSER_TEST_F(NavigationPredictorRendererWarmupClientBrowserTest,DISABLE_ON_WIN_MAC_CHROMEOS (PreexistingSpareRenderer))117 IN_PROC_BROWSER_TEST_F(NavigationPredictorRendererWarmupClientBrowserTest,
118 DISABLE_ON_WIN_MAC_CHROMEOS(PreexistingSpareRenderer)) {
119 // Navigate to a site so that the default renderer is used.
120 embedded_test_server()->ServeFilesFromSourceDirectory("chrome/test/data");
121 ASSERT_TRUE(embedded_test_server()->Start());
122
123 GURL url = embedded_test_server()->GetURL("/simple.html");
124
125 ui_test_utils::NavigateToURL(browser(), url);
126
127 MakeSpareRenderer();
128 base::RunLoop().RunUntilIdle();
129 EXPECT_EQ(SpareRendererCount(), 1U);
130
131 MakeEligibleNavigationPrediction();
132 base::RunLoop().RunUntilIdle();
133
134 EXPECT_EQ(SpareRendererCount(), 1U);
135 VerifyHasUKMEntry(url);
136 }
137