OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/search_engines/default_search_manager.h" | 5 #include "components/search_engines/default_search_manager.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 | 8 |
9 #include <memory> | 9 #include <memory> |
10 | 10 |
11 #include "base/files/scoped_temp_dir.h" | 11 #include "base/files/scoped_temp_dir.h" |
12 #include "base/macros.h" | 12 #include "base/macros.h" |
13 #include "base/strings/string_split.h" | 13 #include "base/strings/string_split.h" |
14 #include "base/strings/string_util.h" | 14 #include "base/strings/string_util.h" |
15 #include "base/strings/utf_string_conversions.h" | 15 #include "base/strings/utf_string_conversions.h" |
16 #include "base/time/time.h" | 16 #include "base/time/time.h" |
17 #include "components/pref_registry/pref_registry_syncable.h" | 17 #include "components/pref_registry/pref_registry_syncable.h" |
18 #include "components/pref_registry/testing_pref_service_syncable.h" | 18 #include "components/pref_registry/testing_pref_service_syncable.h" |
19 #include "components/search_engines/search_engines_pref_names.h" | 19 #include "components/search_engines/search_engines_pref_names.h" |
20 #include "components/search_engines/template_url_data.h" | 20 #include "components/search_engines/template_url_data.h" |
21 #include "components/search_engines/template_url_prepopulate_data.h" | 21 #include "components/search_engines/template_url_prepopulate_data.h" |
| 22 #include "components/search_engines/util.h" |
22 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
23 | 24 |
24 namespace { | 25 namespace { |
25 // A dictionary to hold all data related to the Default Search Engine. | 26 // A dictionary to hold all data related to the Default Search Engine. |
26 // Eventually, this should replace all the data stored in the | 27 // Eventually, this should replace all the data stored in the |
27 // default_search_provider.* prefs. | 28 // default_search_provider.* prefs. |
28 const char kDefaultSearchProviderData[] = | 29 const char kDefaultSearchProviderData[] = |
29 "default_search_provider_data.template_url_data"; | 30 "default_search_provider_data.template_url_data"; |
30 | 31 |
31 // Checks that the two TemplateURLs are similar. Does not check the id, the | 32 // Checks that the two TemplateURLs are similar. Does not check the id, the |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
83 prefs->SetUserPref(prefs::kSearchProviderOverrides, overrides); | 84 prefs->SetUserPref(prefs::kSearchProviderOverrides, overrides); |
84 } | 85 } |
85 | 86 |
86 void SetPolicy(user_prefs::TestingPrefServiceSyncable* prefs, | 87 void SetPolicy(user_prefs::TestingPrefServiceSyncable* prefs, |
87 bool enabled, | 88 bool enabled, |
88 TemplateURLData* data) { | 89 TemplateURLData* data) { |
89 if (enabled) { | 90 if (enabled) { |
90 EXPECT_FALSE(data->keyword().empty()); | 91 EXPECT_FALSE(data->keyword().empty()); |
91 EXPECT_FALSE(data->url().empty()); | 92 EXPECT_FALSE(data->url().empty()); |
92 } | 93 } |
93 std::unique_ptr<base::DictionaryValue> entry(new base::DictionaryValue); | 94 std::unique_ptr<base::DictionaryValue> entry( |
94 entry->SetString(DefaultSearchManager::kShortName, data->short_name()); | 95 TemplateUrlDataToDictionary(*data)); |
95 entry->SetString(DefaultSearchManager::kKeyword, data->keyword()); | |
96 entry->SetString(DefaultSearchManager::kURL, data->url()); | |
97 entry->SetString(DefaultSearchManager::kFaviconURL, data->favicon_url.spec()); | |
98 entry->SetString(DefaultSearchManager::kSuggestionsURL, | |
99 data->suggestions_url); | |
100 entry->SetBoolean(DefaultSearchManager::kSafeForAutoReplace, | |
101 data->safe_for_autoreplace); | |
102 std::unique_ptr<base::ListValue> alternate_urls(new base::ListValue); | |
103 for (std::vector<std::string>::const_iterator it = | |
104 data->alternate_urls.begin(); | |
105 it != data->alternate_urls.end(); | |
106 ++it) { | |
107 alternate_urls->AppendString(*it); | |
108 } | |
109 entry->Set(DefaultSearchManager::kAlternateURLs, alternate_urls.release()); | |
110 | |
111 std::unique_ptr<base::ListValue> encodings(new base::ListValue); | |
112 for (std::vector<std::string>::const_iterator it = | |
113 data->input_encodings.begin(); | |
114 it != data->input_encodings.end(); | |
115 ++it) { | |
116 encodings->AppendString(*it); | |
117 } | |
118 entry->Set(DefaultSearchManager::kInputEncodings, encodings.release()); | |
119 | |
120 entry->SetString(DefaultSearchManager::kSearchTermsReplacementKey, | |
121 data->search_terms_replacement_key); | |
122 entry->SetBoolean(DefaultSearchManager::kDisabledByPolicy, !enabled); | 96 entry->SetBoolean(DefaultSearchManager::kDisabledByPolicy, !enabled); |
123 prefs->SetManagedPref(kDefaultSearchProviderData, entry.release()); | 97 prefs->SetManagedPref(kDefaultSearchProviderData, entry.release()); |
124 } | 98 } |
125 | 99 |
| 100 void SetExtensionControlledDSE(const std::string& extension_id, |
| 101 user_prefs::TestingPrefServiceSyncable* prefs, |
| 102 TemplateURLData* data) { |
| 103 |
| 104 } |
| 105 |
| 106 void ClearExtensionControlledDSE( |
| 107 user_prefs::TestingPrefServiceSyncable* prefs) { |
| 108 |
| 109 } |
| 110 |
126 std::unique_ptr<TemplateURLData> GenerateDummyTemplateURLData( | 111 std::unique_ptr<TemplateURLData> GenerateDummyTemplateURLData( |
127 const std::string& type) { | 112 const std::string& type) { |
128 std::unique_ptr<TemplateURLData> data(new TemplateURLData()); | 113 std::unique_ptr<TemplateURLData> data(new TemplateURLData()); |
129 data->SetShortName(base::UTF8ToUTF16(std::string(type).append("name"))); | 114 data->SetShortName(base::UTF8ToUTF16(std::string(type).append("name"))); |
130 data->SetKeyword(base::UTF8ToUTF16(std::string(type).append("key"))); | 115 data->SetKeyword(base::UTF8ToUTF16(std::string(type).append("key"))); |
131 data->SetURL(std::string("http://").append(type).append("foo/{searchTerms}")); | 116 data->SetURL(std::string("http://").append(type).append("foo/{searchTerms}")); |
132 data->suggestions_url = std::string("http://").append(type).append("sugg"); | 117 data->suggestions_url = std::string("http://").append(type).append("sugg"); |
133 data->alternate_urls.push_back( | 118 data->alternate_urls.push_back( |
134 std::string("http://").append(type).append("foo/alt")); | 119 std::string("http://").append(type).append("foo/alt")); |
135 data->favicon_url = GURL("http://icon1"); | 120 data->favicon_url = GURL("http://icon1"); |
(...skipping 156 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
292 std::unique_ptr<TemplateURLData> data = GenerateDummyTemplateURLData("user"); | 277 std::unique_ptr<TemplateURLData> data = GenerateDummyTemplateURLData("user"); |
293 manager.SetUserSelectedDefaultSearchEngine(*data); | 278 manager.SetUserSelectedDefaultSearchEngine(*data); |
294 | 279 |
295 DefaultSearchManager::Source source = DefaultSearchManager::FROM_FALLBACK; | 280 DefaultSearchManager::Source source = DefaultSearchManager::FROM_FALLBACK; |
296 ExpectSimilar(data.get(), manager.GetDefaultSearchEngine(&source)); | 281 ExpectSimilar(data.get(), manager.GetDefaultSearchEngine(&source)); |
297 EXPECT_EQ(DefaultSearchManager::FROM_USER, source); | 282 EXPECT_EQ(DefaultSearchManager::FROM_USER, source); |
298 | 283 |
299 // Extension trumps prefs: | 284 // Extension trumps prefs: |
300 std::unique_ptr<TemplateURLData> extension_data_1 = | 285 std::unique_ptr<TemplateURLData> extension_data_1 = |
301 GenerateDummyTemplateURLData("ext1"); | 286 GenerateDummyTemplateURLData("ext1"); |
302 manager.SetExtensionControlledDefaultSearchEngine(*extension_data_1); | 287 // manager.SetExtensionControlledDefaultSearchEngine(*extension_data_1); |
| 288 SetExtensionControlledDSE("ext_id1", pref_service(), extension_data_1.get()); |
303 | 289 |
304 ExpectSimilar(extension_data_1.get(), | 290 ExpectSimilar(extension_data_1.get(), |
305 manager.GetDefaultSearchEngine(&source)); | 291 manager.GetDefaultSearchEngine(&source)); |
306 EXPECT_EQ(DefaultSearchManager::FROM_EXTENSION, source); | 292 EXPECT_EQ(DefaultSearchManager::FROM_EXTENSION, source); |
307 | 293 |
308 // Policy trumps extension: | 294 // Policy trumps extension: |
309 std::unique_ptr<TemplateURLData> policy_data = | 295 std::unique_ptr<TemplateURLData> policy_data = |
310 GenerateDummyTemplateURLData("policy"); | 296 GenerateDummyTemplateURLData("policy"); |
311 SetPolicy(pref_service(), true, policy_data.get()); | 297 SetPolicy(pref_service(), true, policy_data.get()); |
312 | 298 |
313 ExpectSimilar(policy_data.get(), manager.GetDefaultSearchEngine(&source)); | 299 ExpectSimilar(policy_data.get(), manager.GetDefaultSearchEngine(&source)); |
314 EXPECT_EQ(DefaultSearchManager::FROM_POLICY, source); | 300 EXPECT_EQ(DefaultSearchManager::FROM_POLICY, source); |
315 pref_service()->RemoveManagedPref(kDefaultSearchProviderData); | 301 pref_service()->RemoveManagedPref(kDefaultSearchProviderData); |
316 | 302 |
317 // Extensions trump each other: | 303 // Extensions trump each other: |
318 std::unique_ptr<TemplateURLData> extension_data_2 = | 304 std::unique_ptr<TemplateURLData> extension_data_2 = |
319 GenerateDummyTemplateURLData("ext2"); | 305 GenerateDummyTemplateURLData("ext2"); |
320 std::unique_ptr<TemplateURLData> extension_data_3 = | 306 std::unique_ptr<TemplateURLData> extension_data_3 = |
321 GenerateDummyTemplateURLData("ext3"); | 307 GenerateDummyTemplateURLData("ext3"); |
322 manager.SetExtensionControlledDefaultSearchEngine(*extension_data_2); | 308 //manager.SetExtensionControlledDefaultSearchEngine(*extension_data_2); |
323 manager.SetExtensionControlledDefaultSearchEngine(*extension_data_3); | 309 SetExtensionControlledDSE("ext_id2", pref_service(), extension_data_2.get()); |
| 310 //manager.SetExtensionControlledDefaultSearchEngine(*extension_data_3); |
| 311 SetExtensionControlledDSE("ext_id3", pref_service(), extension_data_3.get()); |
324 | 312 |
325 ExpectSimilar(extension_data_3.get(), | 313 ExpectSimilar(extension_data_3.get(), |
326 manager.GetDefaultSearchEngine(&source)); | 314 manager.GetDefaultSearchEngine(&source)); |
327 EXPECT_EQ(DefaultSearchManager::FROM_EXTENSION, source); | 315 EXPECT_EQ(DefaultSearchManager::FROM_EXTENSION, source); |
328 | 316 |
329 manager.ClearExtensionControlledDefaultSearchEngine(); | 317 // manager.ClearExtensionControlledDefaultSearchEngine(); |
| 318 ClearExtensionControlledDSE(pref_service()); |
330 | 319 |
331 ExpectSimilar(data.get(), manager.GetDefaultSearchEngine(&source)); | 320 ExpectSimilar(data.get(), manager.GetDefaultSearchEngine(&source)); |
332 EXPECT_EQ(DefaultSearchManager::FROM_USER, source); | 321 EXPECT_EQ(DefaultSearchManager::FROM_USER, source); |
333 } | 322 } |
OLD | NEW |