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/template_url.h" | 8 #include "chrome/browser/search_engines/template_url.h" |
9 #include "chrome/browser/search_engines/ui_thread_search_terms_data.h" | 9 #include "components/search_engines/search_terms_data.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
11 | 11 |
12 typedef SearchHostToURLsMap::TemplateURLSet TemplateURLSet; | 12 typedef SearchHostToURLsMap::TemplateURLSet TemplateURLSet; |
13 | 13 |
14 // Basic functionality for the SearchHostToURLsMap tests. | 14 // Basic functionality for the SearchHostToURLsMap tests. |
15 class SearchHostToURLsMapTest : public testing::Test { | 15 class SearchHostToURLsMapTest : public testing::Test { |
16 public: | 16 public: |
17 SearchHostToURLsMapTest() {} | 17 SearchHostToURLsMapTest() {} |
18 | 18 |
19 virtual void SetUp(); | 19 virtual void SetUp(); |
(...skipping 12 matching lines...) Expand all Loading... |
32 TemplateURLData data; | 32 TemplateURLData data; |
33 data.SetURL("http://" + host_ + "/path1"); | 33 data.SetURL("http://" + host_ + "/path1"); |
34 t_urls_[0].reset(new TemplateURL(data)); | 34 t_urls_[0].reset(new TemplateURL(data)); |
35 data.SetURL("http://" + host_ + "/path2"); | 35 data.SetURL("http://" + host_ + "/path2"); |
36 t_urls_[1].reset(new TemplateURL(data)); | 36 t_urls_[1].reset(new TemplateURL(data)); |
37 std::vector<TemplateURL*> template_urls; | 37 std::vector<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(NULL); | 42 provider_map_->Init(template_urls, SearchTermsData()); |
43 provider_map_->Init(template_urls, search_terms_data); | |
44 } | 43 } |
45 | 44 |
46 TEST_F(SearchHostToURLsMapTest, Add) { | 45 TEST_F(SearchHostToURLsMapTest, Add) { |
47 std::string new_host = "example.com"; | 46 std::string new_host = "example.com"; |
48 TemplateURLData data; | 47 TemplateURLData data; |
49 data.SetURL("http://" + new_host + "/"); | 48 data.SetURL("http://" + new_host + "/"); |
50 TemplateURL new_t_url(data); | 49 TemplateURL new_t_url(data); |
51 UIThreadSearchTermsData search_terms_data(NULL); | 50 provider_map_->Add(&new_t_url, SearchTermsData()); |
52 provider_map_->Add(&new_t_url, search_terms_data); | |
53 | 51 |
54 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost(new_host)); | 52 ASSERT_EQ(&new_t_url, provider_map_->GetTemplateURLForHost(new_host)); |
55 } | 53 } |
56 | 54 |
57 TEST_F(SearchHostToURLsMapTest, Remove) { | 55 TEST_F(SearchHostToURLsMapTest, Remove) { |
58 UIThreadSearchTermsData search_terms_data(NULL); | 56 provider_map_->Remove(t_urls_[0].get(), SearchTermsData()); |
59 provider_map_->Remove(t_urls_[0].get(), search_terms_data); | |
60 | 57 |
61 const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); | 58 const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); |
62 ASSERT_EQ(t_urls_[1].get(), found_url); | 59 ASSERT_EQ(t_urls_[1].get(), found_url); |
63 | 60 |
64 const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); | 61 const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); |
65 ASSERT_TRUE(urls != NULL); | 62 ASSERT_TRUE(urls != NULL); |
66 | 63 |
67 int url_count = 0; | 64 int url_count = 0; |
68 for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { | 65 for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { |
69 url_count++; | 66 url_count++; |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
101 | 98 |
102 for (size_t i = 0; i < arraysize(found_urls); ++i) | 99 for (size_t i = 0; i < arraysize(found_urls); ++i) |
103 ASSERT_TRUE(found_urls[i]); | 100 ASSERT_TRUE(found_urls[i]); |
104 } | 101 } |
105 | 102 |
106 TEST_F(SearchHostToURLsMapTest, GetURLsForUnknownHost) { | 103 TEST_F(SearchHostToURLsMapTest, GetURLsForUnknownHost) { |
107 const SearchHostToURLsMap::TemplateURLSet* urls = | 104 const SearchHostToURLsMap::TemplateURLSet* urls = |
108 provider_map_->GetURLsForHost("a" + host_); | 105 provider_map_->GetURLsForHost("a" + host_); |
109 ASSERT_TRUE(urls == NULL); | 106 ASSERT_TRUE(urls == NULL); |
110 } | 107 } |
OLD | NEW |