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
|
// Copyright 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/omnibox/browser/query_in_omnibox.h"
#include "base/message_loop/message_loop.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/scoped_feature_list.h"
#include "components/omnibox/browser/omnibox_field_trial.h"
#include "components/omnibox/browser/test_omnibox_client.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
const GURL kValidSearchResultsPage =
GURL("https://www.google.com/search?q=foo+query");
} // namespace
class QueryInOmniboxTest : public testing::Test {
protected:
QueryInOmniboxTest()
: omnibox_client_(new TestOmniboxClient),
model_(new QueryInOmnibox(omnibox_client_->GetAutocompleteClassifier(),
omnibox_client_->GetTemplateURLService())) {}
QueryInOmnibox* model() { return model_.get(); }
private:
base::MessageLoop message_loop_;
std::unique_ptr<TestOmniboxClient> omnibox_client_;
std::unique_ptr<QueryInOmnibox> model_;
};
TEST_F(QueryInOmniboxTest, FeatureFlagWorks) {
EXPECT_FALSE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::SECURE, kValidSearchResultsPage, nullptr));
base::test::ScopedFeatureList scoped_feature_list;
scoped_feature_list.InitAndEnableFeature(omnibox::kQueryInOmnibox);
EXPECT_TRUE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::SECURE, kValidSearchResultsPage, nullptr));
}
TEST_F(QueryInOmniboxTest, SecurityLevel) {
base::test::ScopedFeatureList scoped_feature_list;
scoped_feature_list.InitAndEnableFeature(omnibox::kQueryInOmnibox);
EXPECT_TRUE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::SECURE, kValidSearchResultsPage, nullptr));
EXPECT_TRUE(
model()->GetDisplaySearchTerms(security_state::SecurityLevel::EV_SECURE,
kValidSearchResultsPage, nullptr));
// Insecure levels should not be allowed to display search terms.
EXPECT_FALSE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::NONE, kValidSearchResultsPage, nullptr));
EXPECT_FALSE(
model()->GetDisplaySearchTerms(security_state::SecurityLevel::DANGEROUS,
kValidSearchResultsPage, nullptr));
// But respect the flag on to ignore the security level.
model()->set_ignore_security_level(true);
EXPECT_TRUE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::NONE, kValidSearchResultsPage, nullptr));
EXPECT_TRUE(
model()->GetDisplaySearchTerms(security_state::SecurityLevel::DANGEROUS,
kValidSearchResultsPage, nullptr));
}
TEST_F(QueryInOmniboxTest, DefaultSearchProviderWithAndWithoutQuery) {
base::test::ScopedFeatureList scoped_feature_list;
scoped_feature_list.InitAndEnableFeature(omnibox::kQueryInOmnibox);
base::string16 result;
EXPECT_TRUE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::SECURE, kValidSearchResultsPage, &result));
EXPECT_EQ(base::ASCIIToUTF16("foo query"), result);
const GURL kDefaultSearchProviderURLWithNoQuery(
"https://www.google.com/maps");
result.clear();
EXPECT_FALSE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::SECURE,
kDefaultSearchProviderURLWithNoQuery, &result));
EXPECT_EQ(base::string16(), result);
}
TEST_F(QueryInOmniboxTest, NonDefaultSearchProvider) {
base::test::ScopedFeatureList scoped_feature_list;
scoped_feature_list.InitAndEnableFeature(omnibox::kQueryInOmnibox);
const GURL kNonDefaultSearchProvider(
"https://search.yahoo.com/search?ei=UTF-8&fr=crmas&p=foo+query");
base::string16 result;
EXPECT_FALSE(
model()->GetDisplaySearchTerms(security_state::SecurityLevel::SECURE,
kNonDefaultSearchProvider, &result));
EXPECT_EQ(base::string16(), result);
}
TEST_F(QueryInOmniboxTest, LookalikeURL) {
base::test::ScopedFeatureList scoped_feature_list;
scoped_feature_list.InitAndEnableFeature(omnibox::kQueryInOmnibox);
const GURL kLookalikeURLQuery(
"https://www.google.com/search?q=lookalike.com");
base::string16 result;
EXPECT_FALSE(model()->GetDisplaySearchTerms(
security_state::SecurityLevel::SECURE, kLookalikeURLQuery, &result));
EXPECT_EQ(base::string16(), result);
}
|
