OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "base/file_util.h" | 5 #include "base/file_util.h" |
6 #include "base/memory/scoped_vector.h" | 6 #include "base/memory/scoped_vector.h" |
7 #include "base/scoped_temp_dir.h" | 7 #include "base/scoped_temp_dir.h" |
8 #include "base/utf_string_conversions.h" | 8 #include "base/utf_string_conversions.h" |
9 #include "chrome/browser/search_engines/search_engine_type.h" | 9 #include "chrome/browser/search_engines/search_engine_type.h" |
10 #include "chrome/browser/search_engines/search_terms_data.h" | 10 #include "chrome/browser/search_engines/search_terms_data.h" |
(...skipping 130 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
141 profile.GetPrefs()->SetInteger(prefs::kCountryIDAtInstall, | 141 profile.GetPrefs()->SetInteger(prefs::kCountryIDAtInstall, |
142 kCountryIds[i]); | 142 kCountryIds[i]); |
143 ScopedVector<TemplateURL> urls; | 143 ScopedVector<TemplateURL> urls; |
144 size_t default_index; | 144 size_t default_index; |
145 TemplateURLPrepopulateData::GetPrepopulatedEngines( | 145 TemplateURLPrepopulateData::GetPrepopulatedEngines( |
146 profile.GetPrefs(), &(urls.get()), &default_index); | 146 profile.GetPrefs(), &(urls.get()), &default_index); |
147 for (size_t turl_i = 0; turl_i < urls.size(); ++turl_i) { | 147 for (size_t turl_i = 0; turl_i < urls.size(); ++turl_i) { |
148 GURL engine_url(urls[turl_i]->url()->url()); | 148 GURL engine_url(urls[turl_i]->url()->url()); |
149 if (!engine_url.is_valid()) { | 149 if (!engine_url.is_valid()) { |
150 engine_url = TemplateURLService::GenerateSearchURLUsingTermsData( | 150 engine_url = TemplateURLService::GenerateSearchURLUsingTermsData( |
151 urls[turl_i], search_terms_data); | 151 &profile, urls[turl_i], search_terms_data); |
152 } | 152 } |
153 GURL origin = engine_url.GetOrigin(); | 153 GURL origin = engine_url.GetOrigin(); |
154 unique_engines.insert(origin); | 154 unique_engines.insert(origin); |
155 } | 155 } |
156 } | 156 } |
157 } | 157 } |
158 | 158 |
159 TestingProfile profile; | 159 TestingProfile profile; |
160 for (std::set<GURL>::iterator it = unique_engines.begin(); | 160 for (std::set<GURL>::iterator it = unique_engines.begin(); |
161 it != unique_engines.end(); ++it) { | 161 it != unique_engines.end(); ++it) { |
162 scoped_ptr<TemplateURL> found_url( | 162 scoped_ptr<TemplateURL> found_url( |
163 TemplateURLPrepopulateData::GetEngineForOrigin(profile.GetPrefs(), | 163 TemplateURLPrepopulateData::GetEngineForOrigin(profile.GetPrefs(), |
164 *it)); | 164 *it)); |
165 EXPECT_EQ( | 165 EXPECT_EQ( |
166 TemplateURLService::GenerateSearchURLUsingTermsData( | 166 TemplateURLService::GenerateSearchURLUsingTermsData( |
167 found_url.get(), search_terms_data).GetOrigin(), | 167 &profile, found_url.get(), search_terms_data).GetOrigin(), |
168 it->GetOrigin()); | 168 it->GetOrigin()); |
169 } | 169 } |
170 | 170 |
171 GURL not_a_search_engine("http://example.com/"); | 171 GURL not_a_search_engine("http://example.com/"); |
172 EXPECT_EQ(NULL, TemplateURLPrepopulateData::GetEngineForOrigin( | 172 EXPECT_EQ(NULL, TemplateURLPrepopulateData::GetEngineForOrigin( |
173 profile.GetPrefs(), | 173 profile.GetPrefs(), |
174 not_a_search_engine)); | 174 not_a_search_engine)); |
175 } | 175 } |
176 | 176 |
177 TEST_F(TemplateURLPrepopulateDataTest, GetSearchEngineLogo) { | 177 TEST_F(TemplateURLPrepopulateDataTest, GetSearchEngineLogo) { |
178 GURL bad_engine("http://example.com/"); | 178 GURL bad_engine("http://example.com/"); |
179 EXPECT_EQ(kNoSearchEngineLogo, | 179 EXPECT_EQ(kNoSearchEngineLogo, |
180 TemplateURLPrepopulateData::GetSearchEngineLogo(bad_engine)); | 180 TemplateURLPrepopulateData::GetSearchEngineLogo(bad_engine)); |
181 GURL engine_with_logo("http://www.ask.com/"); | 181 GURL engine_with_logo("http://www.ask.com/"); |
182 EXPECT_EQ(IDR_SEARCH_ENGINE_LOGO_ASK, | 182 EXPECT_EQ(IDR_SEARCH_ENGINE_LOGO_ASK, |
183 TemplateURLPrepopulateData::GetSearchEngineLogo(engine_with_logo)); | 183 TemplateURLPrepopulateData::GetSearchEngineLogo(engine_with_logo)); |
184 GURL engine_no_logo("http://araby.com/"); | 184 GURL engine_no_logo("http://araby.com/"); |
185 EXPECT_EQ(kNoSearchEngineLogo, | 185 EXPECT_EQ(kNoSearchEngineLogo, |
186 TemplateURLPrepopulateData::GetSearchEngineLogo(engine_no_logo)); | 186 TemplateURLPrepopulateData::GetSearchEngineLogo(engine_no_logo)); |
187 | 187 |
188 } | 188 } |
OLD | NEW |