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
|
// 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 <string>
#include "chrome/browser/preloading/bookmarkbar_preload/bookmarkbar_preload_pipeline_manager.h"
#include "chrome/browser/preloading/chrome_preloading.h"
#include "chrome/browser/preloading/prerender/prerender_utils.h"
#include "chrome/browser/search_engines/template_url_service_factory_test_util.h"
#include "chrome/test/base/chrome_render_view_host_test_harness.h"
#include "components/search_engines/template_url_service.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/preloading_test_util.h"
#include "content/public/test/prerender_test_util.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
class BookmarkBarPreloadPipelineManagerTest
: public ChromeRenderViewHostTestHarness {
public:
BookmarkBarPreloadPipelineManagerTest(
const BookmarkBarPreloadPipelineManagerTest&) = delete;
BookmarkBarPreloadPipelineManagerTest& operator=(
const BookmarkBarPreloadPipelineManagerTest&) = delete;
BookmarkBarPreloadPipelineManagerTest()
: ChromeRenderViewHostTestHarness(
content::BrowserTaskEnvironment::REAL_IO_THREAD),
prerender_helper_(base::BindRepeating(
&BookmarkBarPreloadPipelineManagerTest::GetActiveWebContents,
base::Unretained(this))) {}
void SetUp() override {
ASSERT_TRUE(test_server_.Start());
ChromeRenderViewHostTestHarness::SetUp();
// Set up TemplateURLService for search prerendering.
TemplateURLServiceFactoryTestUtil factory_util(profile());
factory_util.VerifyLoad();
TemplateURLData template_url_data;
template_url_data.SetURL(GetUrl(search_site() + "?q={searchTerms}").spec());
factory_util.model()->SetUserSelectedDefaultSearchProvider(
factory_util.model()->Add(
std::make_unique<TemplateURL>(template_url_data)));
BookmarkBarPreloadPipelineManager::CreateForWebContents(
GetActiveWebContents());
}
content::WebContents* GetActiveWebContents() { return web_contents(); }
GURL GetSearchSuggestionUrl(const std::string& original_query,
const std::string& search_terms) {
return GetUrl(search_site() + "?q=" + search_terms +
"&oq=" + original_query);
}
protected:
GURL GetUrl(const std::string& path) { return test_server_.GetURL(path); }
BookmarkBarPreloadPipelineManager* bookmarkbar_preload_manager() {
return BookmarkBarPreloadPipelineManager::FromWebContents(web_contents());
}
private:
static std::string search_site() { return "/title1.html"; }
content::test::PrerenderTestHelper prerender_helper_;
net::EmbeddedTestServer test_server_;
};
// Test that a search related url is ignored by the prerender BookmarkBar
// trigger.
TEST_F(BookmarkBarPreloadPipelineManagerTest, DisallowSearchUrlBookmarkBar) {
base::HistogramTester histogram_tester;
GURL prerendering_url = GetSearchSuggestionUrl("prer", "prerender");
bookmarkbar_preload_manager()->StartPrerender(prerendering_url);
EXPECT_FALSE(bookmarkbar_preload_manager()->IsPreloadingStarted());
histogram_tester.ExpectUniqueSample(
"Prerender.IsPrerenderingSRPUrl.Embedder_BookmarkBar", true, 1);
}
} // namespace
|