Index: components/search_engines/search_host_to_urls_map_unittest.cc |
diff --git a/components/search_engines/search_host_to_urls_map_unittest.cc b/components/search_engines/search_host_to_urls_map_unittest.cc |
index 9d29769a7d810d31400100936d4742089313ea62..eb04a98b151d6f358b3bae760e4c3d7576027b52 100644 |
--- a/components/search_engines/search_host_to_urls_map_unittest.cc |
+++ b/components/search_engines/search_host_to_urls_map_unittest.cc |
@@ -9,8 +9,10 @@ |
#include <memory> |
#include "base/macros.h" |
+#include "base/memory/ptr_util.h" |
#include "components/search_engines/search_terms_data.h" |
#include "components/search_engines/template_url.h" |
+#include "components/search_engines/template_url_service.h" |
#include "testing/gtest/include/gtest/gtest.h" |
typedef SearchHostToURLsMap::TemplateURLSet TemplateURLSet; |
@@ -24,7 +26,7 @@ class SearchHostToURLsMapTest : public testing::Test { |
protected: |
std::unique_ptr<SearchHostToURLsMap> provider_map_; |
- std::unique_ptr<TemplateURL> t_urls_[2]; |
+ TemplateURLService::OwnedTemplateURLVector template_urls_; |
std::string host_; |
DISALLOW_COPY_AND_ASSIGN(SearchHostToURLsMapTest); |
@@ -35,15 +37,12 @@ void SearchHostToURLsMapTest::SetUp() { |
host_ = "www.unittest.com"; |
TemplateURLData data; |
data.SetURL("http://" + host_ + "/path1"); |
- t_urls_[0].reset(new TemplateURL(data)); |
+ template_urls_.push_back(base::MakeUnique<TemplateURL>(data)); |
data.SetURL("http://" + host_ + "/path2"); |
- t_urls_[1].reset(new TemplateURL(data)); |
- std::vector<TemplateURL*> template_urls; |
- template_urls.push_back(t_urls_[0].get()); |
- template_urls.push_back(t_urls_[1].get()); |
+ template_urls_.push_back(base::MakeUnique<TemplateURL>(data)); |
provider_map_.reset(new SearchHostToURLsMap); |
- provider_map_->Init(template_urls, SearchTermsData()); |
+ provider_map_->Init(template_urls_, SearchTermsData()); |
} |
TEST_F(SearchHostToURLsMapTest, Add) { |
@@ -57,10 +56,10 @@ TEST_F(SearchHostToURLsMapTest, Add) { |
} |
TEST_F(SearchHostToURLsMapTest, Remove) { |
- provider_map_->Remove(t_urls_[0].get()); |
+ provider_map_->Remove(template_urls_[0].get()); |
const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); |
- ASSERT_EQ(t_urls_[1].get(), found_url); |
+ ASSERT_EQ(template_urls_[1].get(), found_url); |
const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); |
ASSERT_TRUE(urls != NULL); |
@@ -68,14 +67,15 @@ TEST_F(SearchHostToURLsMapTest, Remove) { |
int url_count = 0; |
for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { |
url_count++; |
- ASSERT_EQ(t_urls_[1].get(), *i); |
+ ASSERT_EQ(template_urls_[1].get(), *i); |
} |
ASSERT_EQ(1, url_count); |
} |
TEST_F(SearchHostToURLsMapTest, GetTemplateURLForKnownHost) { |
const TemplateURL* found_url = provider_map_->GetTemplateURLForHost(host_); |
- ASSERT_TRUE(found_url == t_urls_[0].get() || found_url == t_urls_[1].get()); |
+ ASSERT_TRUE(found_url == template_urls_[0].get() || |
+ found_url == template_urls_[1].get()); |
} |
TEST_F(SearchHostToURLsMapTest, GetTemplateURLForUnknownHost) { |
@@ -88,19 +88,19 @@ TEST_F(SearchHostToURLsMapTest, GetURLsForKnownHost) { |
const TemplateURLSet* urls = provider_map_->GetURLsForHost(host_); |
ASSERT_TRUE(urls != NULL); |
- bool found_urls[arraysize(t_urls_)] = { false }; |
+ std::map<size_t, bool> found_urls; |
Peter Kasting
2016/08/31 04:12:56
Wow, this function is a mess.
Seems like everythi
Avi (use Gerrit)
2016/09/01 00:34:27
Done.
|
- for (TemplateURLSet::const_iterator i(urls->begin()); i != urls->end(); ++i) { |
+ for (auto i = urls->begin(); i != urls->end(); ++i) { |
const TemplateURL* url = *i; |
- for (size_t i = 0; i < arraysize(found_urls); ++i) { |
- if (url == t_urls_[i].get()) { |
+ for (size_t i = 0; i < template_urls_.size(); ++i) { |
+ if (url == template_urls_[i].get()) { |
found_urls[i] = true; |
break; |
} |
} |
} |
- for (size_t i = 0; i < arraysize(found_urls); ++i) |
+ for (size_t i = 0; i < template_urls_.size(); ++i) |
ASSERT_TRUE(found_urls[i]); |
} |