| 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 "base/basictypes.h" | 5 #include "base/basictypes.h" |
| 6 #include "base/memory/scoped_ptr.h" | 6 #include "base/memory/scoped_ptr.h" |
| 7 #include "chrome/browser/search_engines/search_host_to_urls_map.h" | 7 #include "chrome/browser/search_engines/search_host_to_urls_map.h" |
| 8 #include "chrome/browser/search_engines/search_terms_data.h" | 8 #include "chrome/browser/search_engines/search_terms_data.h" |
| 9 #include "chrome/browser/search_engines/template_url.h" | 9 #include "chrome/browser/search_engines/template_url.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| (...skipping 11 matching lines...) Expand all Loading... |
| 22 scoped_ptr<SearchHostToURLsMap> provider_map_; | 22 scoped_ptr<SearchHostToURLsMap> provider_map_; |
| 23 scoped_ptr<TemplateURL> t_urls_[2]; | 23 scoped_ptr<TemplateURL> t_urls_[2]; |
| 24 std::string host_; | 24 std::string host_; |
| 25 | 25 |
| 26 DISALLOW_COPY_AND_ASSIGN(SearchHostToURLsMapTest); | 26 DISALLOW_COPY_AND_ASSIGN(SearchHostToURLsMapTest); |
| 27 }; | 27 }; |
| 28 | 28 |
| 29 void SearchHostToURLsMapTest::SetUp() { | 29 void SearchHostToURLsMapTest::SetUp() { |
| 30 // Add some entries to the search host map. | 30 // Add some entries to the search host map. |
| 31 host_ = "www.unittest.com"; | 31 host_ = "www.unittest.com"; |
| 32 t_urls_[0].reset(new TemplateURL()); | 32 TemplateURLData data; |
| 33 t_urls_[0]->SetURL("http://" + host_ + "/path1"); | 33 data.SetURL("http://" + host_ + "/path1"); |
| 34 t_urls_[1].reset(new TemplateURL()); | 34 t_urls_[0].reset(new TemplateURL(data)); |
| 35 t_urls_[1]->SetURL("http://" + host_ + "/path2"); | 35 data.SetURL("http://" + host_ + "/path2"); |
| 36 | 36 t_urls_[1].reset(new TemplateURL(data)); |
| 37 std::vector<const TemplateURL*> template_urls; | 37 std::vector<const TemplateURL*> template_urls; |
| 38 template_urls.push_back(t_urls_[0].get()); | 38 template_urls.push_back(t_urls_[0].get()); |
| 39 template_urls.push_back(t_urls_[1].get()); | 39 template_urls.push_back(t_urls_[1].get()); |
| 40 | 40 |
| 41 provider_map_.reset(new SearchHostToURLsMap); | 41 provider_map_.reset(new SearchHostToURLsMap); |
| 42 UIThreadSearchTermsData search_terms_data; | 42 UIThreadSearchTermsData search_terms_data; |
| 43 provider_map_->Init(template_urls, search_terms_data); | 43 provider_map_->Init(template_urls, search_terms_data); |
| 44 } | 44 } |
| 45 | 45 |
| 46 TEST_F(SearchHostToURLsMapTest, Add) { | 46 TEST_F(SearchHostToURLsMapTest, Add) { |
| 47 std::string new_host = "example.com"; | 47 std::string new_host = "example.com"; |
| 48 TemplateURL new_t_url; | 48 TemplateURLData data; |
| 49 new_t_url.SetURL("http://" + new_host + "/"); | 49 data.SetURL("http://" + new_host + "/"); |
| 50 TemplateURL new_t_url(data); |
| 50 UIThreadSearchTermsData search_terms_data; | 51 UIThreadSearchTermsData search_terms_data; |
| 51 provider_map_->Add(&new_t_url, search_terms_data); | 52 provider_map_->Add(&new_t_url, search_terms_data); |
| 52 | 53 |
| 53 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost(new_host)); | 54 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost(new_host)); |
| 54 } | 55 } |
| 55 | 56 |
| 56 TEST_F(SearchHostToURLsMapTest, Remove) { | 57 TEST_F(SearchHostToURLsMapTest, Remove) { |
| 57 provider_map_->Remove(t_urls_[0].get()); | 58 provider_map_->Remove(t_urls_[0].get()); |
| 58 | 59 |
| 59 const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); | 60 const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); |
| 60 ASSERT_EQ(t_urls_[1].get(), found_url); | 61 ASSERT_EQ(t_urls_[1].get(), found_url); |
| 61 | 62 |
| 62 const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); | 63 const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); |
| 63 ASSERT_TRUE(urls != NULL); | 64 ASSERT_TRUE(urls != NULL); |
| 64 | 65 |
| 65 int url_count = 0; | 66 int url_count = 0; |
| 66 for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { | 67 for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { |
| 67 url_count++; | 68 url_count++; |
| 68 ASSERT_EQ(t_urls_[1].get(), *i); | 69 ASSERT_EQ(t_urls_[1].get(), *i); |
| 69 } | 70 } |
| 70 ASSERT_EQ(1, url_count); | 71 ASSERT_EQ(1, url_count); |
| 71 } | 72 } |
| 72 | 73 |
| 73 TEST_F(SearchHostToURLsMapTest, Update) { | |
| 74 std::string new_host = "example.com"; | |
| 75 TemplateURL new_values; | |
| 76 new_values.SetURL("http://" + new_host + "/"); | |
| 77 | |
| 78 UIThreadSearchTermsData search_terms_data; | |
| 79 provider_map_->Update(t_urls_[0].get(), new_values, search_terms_data); | |
| 80 | |
| 81 ASSERT_EQ(t_urls_[0].get(), provider_map_->GetTemplateURLForHost(new_host)); | |
| 82 ASSERT_EQ(t_urls_[1].get(), provider_map_->GetTemplateURLForHost(host_)); | |
| 83 } | |
| 84 | |
| 85 TEST_F(SearchHostToURLsMapTest, UpdateGoogleBaseURLs) { | 74 TEST_F(SearchHostToURLsMapTest, UpdateGoogleBaseURLs) { |
| 86 UIThreadSearchTermsData search_terms_data; | 75 UIThreadSearchTermsData search_terms_data; |
| 87 std::string google_base_url = "google.com"; | 76 std::string google_base_url = "google.com"; |
| 88 search_terms_data.SetGoogleBaseURL( | 77 search_terms_data.SetGoogleBaseURL( |
| 89 new std::string("http://" + google_base_url +"/")); | 78 new std::string("http://" + google_base_url +"/")); |
| 90 | 79 |
| 91 // Add in a url with the templated Google base url. | 80 // Add in a url with the templated Google base url. |
| 92 TemplateURL new_t_url; | 81 TemplateURLData data; |
| 93 new_t_url.SetURL("{google:baseURL}?q={searchTerms}"); | 82 data.SetURL("{google:baseURL}?q={searchTerms}"); |
| 83 TemplateURL new_t_url(data); |
| 94 provider_map_->Add(&new_t_url, search_terms_data); | 84 provider_map_->Add(&new_t_url, search_terms_data); |
| 95 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost(google_base_url)); | 85 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost(google_base_url)); |
| 96 | 86 |
| 97 // Now change the Google base url and verify the result. | 87 // Now change the Google base url and verify the result. |
| 98 std::string new_google_base_url = "google.co.uk"; | 88 std::string new_google_base_url = "google.co.uk"; |
| 99 search_terms_data.SetGoogleBaseURL( | 89 search_terms_data.SetGoogleBaseURL( |
| 100 new std::string("http://" + new_google_base_url +"/")); | 90 new std::string("http://" + new_google_base_url +"/")); |
| 101 provider_map_->UpdateGoogleBaseURLs(search_terms_data); | 91 provider_map_->UpdateGoogleBaseURLs(search_terms_data); |
| 102 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost( | 92 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost( |
| 103 new_google_base_url)); | 93 new_google_base_url)); |
| (...skipping 29 matching lines...) Expand all Loading... |
| 133 | 123 |
| 134 for (size_t i = 0; i < arraysize(found_urls); ++i) | 124 for (size_t i = 0; i < arraysize(found_urls); ++i) |
| 135 ASSERT_TRUE(found_urls[i]); | 125 ASSERT_TRUE(found_urls[i]); |
| 136 } | 126 } |
| 137 | 127 |
| 138 TEST_F(SearchHostToURLsMapTest, GetURLsForUnknownHost) { | 128 TEST_F(SearchHostToURLsMapTest, GetURLsForUnknownHost) { |
| 139 const SearchHostToURLsMap::TemplateURLSet* urls = | 129 const SearchHostToURLsMap::TemplateURLSet* urls = |
| 140 provider_map_->GetURLsForHost("a" + host_); | 130 provider_map_->GetURLsForHost("a" + host_); |
| 141 ASSERT_TRUE(urls == NULL); | 131 ASSERT_TRUE(urls == NULL); |
| 142 } | 132 } |
| OLD | NEW |