| 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 |