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/memory/scoped_vector.h" | |
6 #include "base/strings/string_util.h" | |
7 #include "base/strings/utf_string_conversions.h" | |
8 #include "components/search_engines/search_terms_data.h" | |
9 #include "components/search_engines/template_url.h" | |
10 #include "components/search_engines/template_url_service.h" | |
11 #include "components/search_engines/util.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 namespace { | |
15 | |
16 scoped_ptr<TemplateURLData> CreatePrepopulateTemplateURLData( | |
17 int prepopulate_id, | |
18 const std::string& keyword, | |
19 TemplateURLID id) { | |
20 scoped_ptr<TemplateURLData> data(new TemplateURLData); | |
21 data->prepopulate_id = prepopulate_id; | |
22 data->SetKeyword(base::ASCIIToUTF16(keyword)); | |
23 data->id = id; | |
24 return data.Pass(); | |
25 } | |
26 | |
27 // Creates a TemplateURL with default values except for the prepopulate ID, | |
28 // keyword and TemplateURLID. Only use this in tests if your tests do not | |
29 // care about other fields. | |
30 scoped_ptr<TemplateURL> CreatePrepopulateTemplateURL(int prepopulate_id, | |
31 const std::string& keyword, | |
32 TemplateURLID id) { | |
33 return make_scoped_ptr(new TemplateURL( | |
34 *CreatePrepopulateTemplateURLData(prepopulate_id, keyword, id))); | |
35 } | |
36 | |
37 }; // namespace | |
38 | |
39 TEST(TemplateURLServiceUtilTest, RemoveDuplicatePrepopulateIDs) { | |
40 ScopedVector<TemplateURLData> prepopulated_turls; | |
41 TemplateURLService::TemplateURLVector local_turls; | |
42 STLElementDeleter<TemplateURLService::TemplateURLVector> local_turls_deleter( | |
43 &local_turls); | |
44 | |
45 prepopulated_turls.push_back( | |
46 CreatePrepopulateTemplateURLData(1, "winner4", 1).release()); | |
47 prepopulated_turls.push_back( | |
48 CreatePrepopulateTemplateURLData(2, "xxx", 2).release()); | |
49 prepopulated_turls.push_back( | |
50 CreatePrepopulateTemplateURLData(3, "yyy", 3).release()); | |
51 | |
52 // Create a sets of different TURLs grouped by prepopulate ID. Each group | |
53 // will test a different heuristic of RemoveDuplicatePrepopulateIDs. | |
54 // Ignored set - These should be left alone as they do not have valid | |
55 // prepopulate IDs. | |
56 local_turls.push_back( | |
57 CreatePrepopulateTemplateURL(0, "winner1", 4).release()); | |
58 local_turls.push_back( | |
59 CreatePrepopulateTemplateURL(0, "winner2", 5).release()); | |
60 local_turls.push_back( | |
61 CreatePrepopulateTemplateURL(0, "winner3", 6).release()); | |
62 size_t num_non_prepopulated_urls = local_turls.size(); | |
63 | |
64 // Keyword match set - Prefer the one that matches the keyword of the | |
65 // prepopulate ID. | |
66 local_turls.push_back(CreatePrepopulateTemplateURL(1, "loser1", 7).release()); | |
67 local_turls.push_back(CreatePrepopulateTemplateURL(1, "loser2", 8).release()); | |
68 local_turls.push_back( | |
69 CreatePrepopulateTemplateURL(1, "winner4", 9).release()); | |
70 | |
71 // Default set - Prefer the default search engine over all other criteria. | |
72 // The last one is the default. It will be passed as the | |
73 // default_search_provider parameter to RemoveDuplicatePrepopulateIDs. | |
74 local_turls.push_back( | |
75 CreatePrepopulateTemplateURL(2, "loser3", 10).release()); | |
76 local_turls.push_back(CreatePrepopulateTemplateURL(2, "xxx", 11).release()); | |
77 TemplateURL* default_turl = | |
78 CreatePrepopulateTemplateURL(2, "winner5", 12).release(); | |
79 local_turls.push_back(default_turl); | |
80 | |
81 // ID set - Prefer the lowest TemplateURLID if the keywords don't match and if | |
82 // none are the default. | |
83 local_turls.push_back( | |
84 CreatePrepopulateTemplateURL(3, "winner6", 13).release()); | |
85 local_turls.push_back( | |
86 CreatePrepopulateTemplateURL(3, "loser5", 14).release()); | |
87 local_turls.push_back( | |
88 CreatePrepopulateTemplateURL(3, "loser6", 15).release()); | |
89 | |
90 RemoveDuplicatePrepopulateIDs(NULL, prepopulated_turls, default_turl, | |
91 &local_turls, SearchTermsData(), NULL); | |
92 | |
93 // Verify that the expected local TURLs survived the process. | |
94 EXPECT_EQ(local_turls.size(), | |
95 prepopulated_turls.size() + num_non_prepopulated_urls); | |
96 for (TemplateURLService::TemplateURLVector::const_iterator itr = | |
97 local_turls.begin(); itr != local_turls.end(); ++itr) { | |
98 EXPECT_TRUE( | |
99 StartsWith((*itr)->keyword(), base::ASCIIToUTF16("winner"), true)); | |
100 } | |
101 } | |
OLD | NEW |