| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "base/basictypes.h" | |
| 6 #include "base/memory/scoped_ptr.h" | |
| 7 #include "components/search_engines/search_host_to_urls_map.h" | |
| 8 #include "components/search_engines/search_terms_data.h" | |
| 9 #include "components/search_engines/template_url.h" | |
| 10 #include "testing/gtest/include/gtest/gtest.h" | |
| 11 | |
| 12 typedef SearchHostToURLsMap::TemplateURLSet TemplateURLSet; | |
| 13 | |
| 14 // Basic functionality for the SearchHostToURLsMap tests. | |
| 15 class SearchHostToURLsMapTest : public testing::Test { | |
| 16 public: | |
| 17 SearchHostToURLsMapTest() {} | |
| 18 | |
| 19 virtual void SetUp(); | |
| 20 | |
| 21 protected: | |
| 22 scoped_ptr<SearchHostToURLsMap> provider_map_; | |
| 23 scoped_ptr<TemplateURL> t_urls_[2]; | |
| 24 std::string host_; | |
| 25 | |
| 26 DISALLOW_COPY_AND_ASSIGN(SearchHostToURLsMapTest); | |
| 27 }; | |
| 28 | |
| 29 void SearchHostToURLsMapTest::SetUp() { | |
| 30 // Add some entries to the search host map. | |
| 31 host_ = "www.unittest.com"; | |
| 32 TemplateURLData data; | |
| 33 data.SetURL("http://" + host_ + "/path1"); | |
| 34 t_urls_[0].reset(new TemplateURL(data)); | |
| 35 data.SetURL("http://" + host_ + "/path2"); | |
| 36 t_urls_[1].reset(new TemplateURL(data)); | |
| 37 std::vector<TemplateURL*> template_urls; | |
| 38 template_urls.push_back(t_urls_[0].get()); | |
| 39 template_urls.push_back(t_urls_[1].get()); | |
| 40 | |
| 41 provider_map_.reset(new SearchHostToURLsMap); | |
| 42 provider_map_->Init(template_urls, SearchTermsData()); | |
| 43 } | |
| 44 | |
| 45 TEST_F(SearchHostToURLsMapTest, Add) { | |
| 46 std::string new_host = "example.com"; | |
| 47 TemplateURLData data; | |
| 48 data.SetURL("http://" + new_host + "/"); | |
| 49 TemplateURL new_t_url(data); | |
| 50 provider_map_->Add(&new_t_url, SearchTermsData()); | |
| 51 | |
| 52 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost(new_host)); | |
| 53 } | |
| 54 | |
| 55 TEST_F(SearchHostToURLsMapTest, Remove) { | |
| 56 provider_map_->Remove(t_urls_[0].get()); | |
| 57 | |
| 58 const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); | |
| 59 ASSERT_EQ(t_urls_[1].get(), found_url); | |
| 60 | |
| 61 const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); | |
| 62 ASSERT_TRUE(urls != NULL); | |
| 63 | |
| 64 int url_count = 0; | |
| 65 for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { | |
| 66 url_count++; | |
| 67 ASSERT_EQ(t_urls_[1].get(), *i); | |
| 68 } | |
| 69 ASSERT_EQ(1, url_count); | |
| 70 } | |
| 71 | |
| 72 TEST_F(SearchHostToURLsMapTest, GetTemplateURLForKnownHost) { | |
| 73 const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); | |
| 74 ASSERT_TRUE(found_url == t_urls_[0].get() || found_url == t_urls_[1].get()); | |
| 75 } | |
| 76 | |
| 77 TEST_F(SearchHostToURLsMapTest, GetTemplateURLForUnknownHost) { | |
| 78 const TemplateURL* found_url = | |
| 79 provider_map_->GetTemplateURLForHost("a" + host_); | |
| 80 ASSERT_TRUE(found_url == NULL); | |
| 81 } | |
| 82 | |
| 83 TEST_F(SearchHostToURLsMapTest, GetURLsForKnownHost) { | |
| 84 const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); | |
| 85 ASSERT_TRUE(urls != NULL); | |
| 86 | |
| 87 bool found_urls[arraysize(t_urls_)] = { false }; | |
| 88 | |
| 89 for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { | |
| 90 const TemplateURL* url = *i; | |
| 91 for (size_t i = 0; i < arraysize(found_urls); ++i) { | |
| 92 if (url == t_urls_[i].get()) { | |
| 93 found_urls[i] = true; | |
| 94 break; | |
| 95 } | |
| 96 } | |
| 97 } | |
| 98 | |
| 99 for (size_t i = 0; i < arraysize(found_urls); ++i) | |
| 100 ASSERT_TRUE(found_urls[i]); | |
| 101 } | |
| 102 | |
| 103 TEST_F(SearchHostToURLsMapTest, GetURLsForUnknownHost) { | |
| 104 const SearchHostToURLsMap::TemplateURLSet* urls = | |
| 105 provider_map_->GetURLsForHost("a" + host_); | |
| 106 ASSERT_TRUE(urls == NULL); | |
| 107 } | |
| OLD | NEW |