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
|
// 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/preloading/prefetch/prefetch_service/chrome_prefetch_service_delegate.h"
#if BUILDFLAG(IS_ANDROID)
#include "chrome/browser/flags/android/chrome_feature_list.h"
#endif // BUILDFLAG(IS_ANDROID)
#include "chrome/browser/search_engines/template_url_service_factory.h"
#include "chrome/browser/search_engines/template_url_service_factory_test_util.h"
#include "chrome/test/base/testing_profile.h"
#include "components/search_engines/template_url_service.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
class ChromePrefetchServiceDelegateTest : public ::testing::Test {
protected:
void SetUp() override { factory_util_.VerifyLoad(); }
TestingProfile& profile() { return profile_; }
TemplateURLService* template_url_service() { return factory_util_.model(); }
private:
content::BrowserTaskEnvironment task_environment_;
TestingProfile profile_;
TemplateURLServiceFactoryTestUtil factory_util_{&profile_};
};
TEST_F(ChromePrefetchServiceDelegateTest,
DefaultSearchEngineIsContaminationExempt) {
#if BUILDFLAG(IS_ANDROID)
base::test::ScopedFeatureList scoped_feature_list;
scoped_feature_list.InitWithFeatures(
{chrome::android::kCCTNavigationalPrefetch}, {});
#endif // BUILDFLAG(IS_ANDROID)
TemplateURLData data;
data.SetShortName(u"Sherlock");
data.SetKeyword(u"sherlock");
data.SetURL("https://sherlock.example/?q={searchTerms}");
TemplateURL* search_engine =
template_url_service()->Add(std::make_unique<TemplateURL>(data));
template_url_service()->SetUserSelectedDefaultSearchProvider(search_engine);
ChromePrefetchServiceDelegate delegate(&profile());
EXPECT_TRUE(delegate.IsContaminationExempt(
GURL("https://sherlock.example/?q=professor+moriarty")));
EXPECT_FALSE(
delegate.IsContaminationExempt(GURL("https://another.example/")));
#if BUILDFLAG(IS_ANDROID)
EXPECT_TRUE(delegate.IsContaminationExemptPerOrigin(
url::Origin::Create(GURL("https://sherlock.example/"))));
EXPECT_FALSE(delegate.IsContaminationExemptPerOrigin(
url::Origin::Create(GURL("https://another.example/"))));
#endif // BUILDFLAG(IS_ANDROID)
}
|
