OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "chrome/browser/sync/test/integration/search_engines_helper.h" | 5 #include "chrome/browser/sync/test/integration/search_engines_helper.h" |
6 | 6 |
7 #include <vector> | 7 #include <vector> |
8 | 8 |
9 #include "base/string_util.h" | 9 #include "base/string_util.h" |
10 #include "base/stringprintf.h" | 10 #include "base/stringprintf.h" |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
54 if (!result) { | 54 if (!result) { |
55 LOG(ERROR) << "TemplateURLs did not match: " << GetTURLInfoString(turl1) | 55 LOG(ERROR) << "TemplateURLs did not match: " << GetTURLInfoString(turl1) |
56 << " vs " << GetTURLInfoString(turl2); | 56 << " vs " << GetTURLInfoString(turl2); |
57 } | 57 } |
58 | 58 |
59 return result; | 59 return result; |
60 } | 60 } |
61 | 61 |
62 bool ServicesMatch(int profile_a, int profile_b) { | 62 bool ServicesMatch(int profile_a, int profile_b) { |
63 TemplateURLService* service_a = | 63 TemplateURLService* service_a = |
64 search_engines_helper::GetServiceForProfile(profile_a); | 64 search_engines_helper::GetServiceForBrowserContext(profile_a); |
65 TemplateURLService* service_b = | 65 TemplateURLService* service_b = |
66 search_engines_helper::GetServiceForProfile(profile_b); | 66 search_engines_helper::GetServiceForBrowserContext(profile_b); |
67 CHECK(service_a); | 67 CHECK(service_a); |
68 CHECK(service_b); | 68 CHECK(service_b); |
69 | 69 |
70 // Services that have synced should have identical TURLs, including the GUIDs. | 70 // Services that have synced should have identical TURLs, including the GUIDs. |
71 // Make sure we compare those fields in addition to the user-visible fields. | 71 // Make sure we compare those fields in addition to the user-visible fields. |
72 GUIDToTURLMap a_turls = CreateGUIDToTURLMap(service_a); | 72 GUIDToTURLMap a_turls = CreateGUIDToTURLMap(service_a); |
73 GUIDToTURLMap b_turls = CreateGUIDToTURLMap(service_b); | 73 GUIDToTURLMap b_turls = CreateGUIDToTURLMap(service_b); |
74 | 74 |
75 if (a_turls.size() != b_turls.size()) { | 75 if (a_turls.size() != b_turls.size()) { |
76 LOG(ERROR) << "Service a and b do not match in size: " << a_turls.size() | 76 LOG(ERROR) << "Service a and b do not match in size: " << a_turls.size() |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
108 // Convenience helper for consistently generating the same keyword for a given | 108 // Convenience helper for consistently generating the same keyword for a given |
109 // seed. | 109 // seed. |
110 string16 CreateKeyword(int seed) { | 110 string16 CreateKeyword(int seed) { |
111 return ASCIIToUTF16(base::StringPrintf("test%d", seed)); | 111 return ASCIIToUTF16(base::StringPrintf("test%d", seed)); |
112 } | 112 } |
113 | 113 |
114 } // namespace | 114 } // namespace |
115 | 115 |
116 namespace search_engines_helper { | 116 namespace search_engines_helper { |
117 | 117 |
118 TemplateURLService* GetServiceForProfile(int profile_index) { | 118 TemplateURLService* GetServiceForBrowserContext(int profile_index) { |
119 return TemplateURLServiceFactory::GetForProfile( | 119 return TemplateURLServiceFactory::GetForProfile( |
120 test()->GetProfile(profile_index)); | 120 test()->GetProfile(profile_index)); |
121 } | 121 } |
122 | 122 |
123 TemplateURLService* GetVerifierService() { | 123 TemplateURLService* GetVerifierService() { |
124 return TemplateURLServiceFactory::GetForProfile(test()->verifier()); | 124 return TemplateURLServiceFactory::GetForProfile(test()->verifier()); |
125 } | 125 } |
126 | 126 |
127 bool ServiceMatchesVerifier(int profile_index) { | 127 bool ServiceMatchesVerifier(int profile_index) { |
128 TemplateURLService* verifier = GetVerifierService(); | 128 TemplateURLService* verifier = GetVerifierService(); |
129 TemplateURLService* other = GetServiceForProfile(profile_index); | 129 TemplateURLService* other = GetServiceForBrowserContext(profile_index); |
130 | 130 |
131 CHECK(verifier); | 131 CHECK(verifier); |
132 CHECK(other); | 132 CHECK(other); |
133 | 133 |
134 TemplateURLService::TemplateURLVector verifier_turls = | 134 TemplateURLService::TemplateURLVector verifier_turls = |
135 verifier->GetTemplateURLs(); | 135 verifier->GetTemplateURLs(); |
136 if (verifier_turls.size() != other->GetTemplateURLs().size()) { | 136 if (verifier_turls.size() != other->GetTemplateURLs().size()) { |
137 LOG(ERROR) << "Verifier and other service have a different count of TURLs: " | 137 LOG(ERROR) << "Verifier and other service have a different count of TURLs: " |
138 << verifier_turls.size() << " vs " | 138 << verifier_turls.size() << " vs " |
139 << other->GetTemplateURLs().size() << " respectively."; | 139 << other->GetTemplateURLs().size() << " respectively."; |
(...skipping 73 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
213 if (test()->use_verifier()) | 213 if (test()->use_verifier()) |
214 GetVerifierService()->Add(CreateTestTemplateURL(profile, seed)); | 214 GetVerifierService()->Add(CreateTestTemplateURL(profile, seed)); |
215 } | 215 } |
216 | 216 |
217 void EditSearchEngine(int profile_index, | 217 void EditSearchEngine(int profile_index, |
218 const string16& keyword, | 218 const string16& keyword, |
219 const string16& short_name, | 219 const string16& short_name, |
220 const string16& new_keyword, | 220 const string16& new_keyword, |
221 const std::string& url) { | 221 const std::string& url) { |
222 DCHECK(!url.empty()); | 222 DCHECK(!url.empty()); |
223 TemplateURLService* service = GetServiceForProfile(profile_index); | 223 TemplateURLService* service = GetServiceForBrowserContext(profile_index); |
224 TemplateURL* turl = service->GetTemplateURLForKeyword(keyword); | 224 TemplateURL* turl = service->GetTemplateURLForKeyword(keyword); |
225 EXPECT_TRUE(turl); | 225 EXPECT_TRUE(turl); |
226 ASSERT_FALSE(new_keyword.empty()); | 226 ASSERT_FALSE(new_keyword.empty()); |
227 service->ResetTemplateURL(turl, short_name, new_keyword, url); | 227 service->ResetTemplateURL(turl, short_name, new_keyword, url); |
228 // Make sure we do the same on the verifier. | 228 // Make sure we do the same on the verifier. |
229 if (test()->use_verifier()) { | 229 if (test()->use_verifier()) { |
230 TemplateURL* verifier_turl = | 230 TemplateURL* verifier_turl = |
231 GetVerifierService()->GetTemplateURLForKeyword(keyword); | 231 GetVerifierService()->GetTemplateURLForKeyword(keyword); |
232 EXPECT_TRUE(verifier_turl); | 232 EXPECT_TRUE(verifier_turl); |
233 GetVerifierService()->ResetTemplateURL(verifier_turl, short_name, | 233 GetVerifierService()->ResetTemplateURL(verifier_turl, short_name, |
234 new_keyword, url); | 234 new_keyword, url); |
235 } | 235 } |
236 } | 236 } |
237 | 237 |
238 void DeleteSearchEngineBySeed(int profile_index, int seed) { | 238 void DeleteSearchEngineBySeed(int profile_index, int seed) { |
239 TemplateURLService* service = GetServiceForProfile(profile_index); | 239 TemplateURLService* service = GetServiceForBrowserContext(profile_index); |
240 string16 keyword(CreateKeyword(seed)); | 240 string16 keyword(CreateKeyword(seed)); |
241 TemplateURL* turl = service->GetTemplateURLForKeyword(keyword); | 241 TemplateURL* turl = service->GetTemplateURLForKeyword(keyword); |
242 EXPECT_TRUE(turl); | 242 EXPECT_TRUE(turl); |
243 service->Remove(turl); | 243 service->Remove(turl); |
244 // Make sure we do the same on the verifier. | 244 // Make sure we do the same on the verifier. |
245 if (test()->use_verifier()) { | 245 if (test()->use_verifier()) { |
246 TemplateURL* verifier_turl = | 246 TemplateURL* verifier_turl = |
247 GetVerifierService()->GetTemplateURLForKeyword(keyword); | 247 GetVerifierService()->GetTemplateURLForKeyword(keyword); |
248 EXPECT_TRUE(verifier_turl); | 248 EXPECT_TRUE(verifier_turl); |
249 GetVerifierService()->Remove(verifier_turl); | 249 GetVerifierService()->Remove(verifier_turl); |
250 } | 250 } |
251 } | 251 } |
252 | 252 |
253 void ChangeDefaultSearchProvider(int profile_index, int seed) { | 253 void ChangeDefaultSearchProvider(int profile_index, int seed) { |
254 TemplateURLService* service = GetServiceForProfile(profile_index); | 254 TemplateURLService* service = GetServiceForBrowserContext(profile_index); |
255 ASSERT_TRUE(service); | 255 ASSERT_TRUE(service); |
256 TemplateURL* turl = service->GetTemplateURLForKeyword(CreateKeyword(seed)); | 256 TemplateURL* turl = service->GetTemplateURLForKeyword(CreateKeyword(seed)); |
257 ASSERT_TRUE(turl); | 257 ASSERT_TRUE(turl); |
258 service->SetDefaultSearchProvider(turl); | 258 service->SetDefaultSearchProvider(turl); |
259 if (test()->use_verifier()) { | 259 if (test()->use_verifier()) { |
260 TemplateURL* verifier_turl = | 260 TemplateURL* verifier_turl = |
261 GetVerifierService()->GetTemplateURLForKeyword(CreateKeyword(seed)); | 261 GetVerifierService()->GetTemplateURLForKeyword(CreateKeyword(seed)); |
262 ASSERT_TRUE(verifier_turl); | 262 ASSERT_TRUE(verifier_turl); |
263 GetVerifierService()->SetDefaultSearchProvider(verifier_turl); | 263 GetVerifierService()->SetDefaultSearchProvider(verifier_turl); |
264 } | 264 } |
265 } | 265 } |
266 | 266 |
267 } // namespace search_engines_helper | 267 } // namespace search_engines_helper |
OLD | NEW |