| 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 |