1 // Copyright 2013 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.
7 #include "base/memory/scoped_vector.h"
8 #include "base/strings/string16.h"
9 #include "base/strings/stringprintf.h"
10 #include "base/strings/utf_string_conversions.h"
11 #include "chrome/browser/ui/app_list/search/chrome_search_result.h"
12 #include "chrome/browser/ui/app_list/search/history_types.h"
13 #include "chrome/browser/ui/app_list/search/mixer.h"
14 #include "chrome/browser/ui/app_list/search/search_provider.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16 #include "ui/app_list/app_list_model.h"
21 class TestSearchResult : public ChromeSearchResult {
23 TestSearchResult(const std::string& id, double relevance) {
25 set_title(UTF8ToUTF16(id));
26 set_relevance(relevance);
28 virtual ~TestSearchResult() {}
31 // ChromeSearchResult overides:
32 virtual void Open(int event_flags) OVERRIDE {}
33 virtual void InvokeAction(int action_index, int event_flags) OVERRIDE {}
34 virtual scoped_ptr<ChromeSearchResult> Duplicate() OVERRIDE {
35 return scoped_ptr<ChromeSearchResult>(
36 new TestSearchResult(id(), relevance())).Pass();
38 virtual ChromeSearchResultType GetType() OVERRIDE {
39 return SEARCH_RESULT_TYPE_BOUNDARY;
42 DISALLOW_COPY_AND_ASSIGN(TestSearchResult);
45 class TestSearchProvider : public SearchProvider {
47 explicit TestSearchProvider(const std::string& prefix)
50 virtual ~TestSearchProvider() {}
52 // SearchProvider overrides:
53 virtual void Start(const string16& query) OVERRIDE {
55 for (size_t i = 0; i < count_; ++i) {
56 const std::string id =
57 base::StringPrintf("%s%d", prefix_.c_str(), static_cast<int>(i));
58 const double relevance = 1.0 - i / 10.0;
59 Add(scoped_ptr<ChromeSearchResult>(
60 new TestSearchResult(id, relevance)).Pass());
63 virtual void Stop() OVERRIDE {}
65 void set_prefix(const std::string& prefix) { prefix_ = prefix; }
66 void set_count(size_t count) { count_ = count; }
72 DISALLOW_COPY_AND_ASSIGN(TestSearchProvider);
75 class MixerTest : public testing::Test {
78 virtual ~MixerTest() {}
80 // testing::Test overrides:
81 virtual void SetUp() OVERRIDE {
82 results_.reset(new AppListModel::SearchResults);
84 providers_.push_back(new TestSearchProvider("app"));
85 providers_.push_back(new TestSearchProvider("omnibox"));
86 providers_.push_back(new TestSearchProvider("webstore"));
87 providers_.push_back(new TestSearchProvider("people"));
89 mixer_.reset(new Mixer(results_.get()));
91 mixer_->AddProviderToGroup(Mixer::MAIN_GROUP, providers_[0]);
92 mixer_->AddProviderToGroup(Mixer::OMNIBOX_GROUP, providers_[1]);
93 mixer_->AddProviderToGroup(Mixer::WEBSTORE_GROUP, providers_[2]);
94 mixer_->AddProviderToGroup(Mixer::PEOPLE_GROUP, providers_[3]);
100 for (size_t i = 0; i < providers_.size(); ++i) {
101 providers_[i]->Start(query);
102 providers_[i]->Stop();
105 mixer_->MixAndPublish(KnownResults());
108 std::string GetResults() const {
110 for (size_t i = 0; i < results_->item_count(); ++i) {
114 result += UTF16ToUTF8(results_->GetItemAt(i)->title());
120 Mixer* mixer() { return mixer_.get(); }
121 TestSearchProvider* app_provider() { return providers_[0]; }
122 TestSearchProvider* omnibox_provider() { return providers_[1]; }
123 TestSearchProvider* webstore_provider() { return providers_[2]; }
126 scoped_ptr<Mixer> mixer_;
127 scoped_ptr<AppListModel::SearchResults> results_;
129 ScopedVector<TestSearchProvider> providers_;
131 DISALLOW_COPY_AND_ASSIGN(MixerTest);
134 TEST_F(MixerTest, Basic) {
136 const size_t app_results;
137 const size_t omnibox_results;
138 const size_t webstore_results;
139 const char* expected;
142 {4, 6, 2, "app0,app1,app2,app3,omnibox0,webstore0"},
143 {10, 10, 10, "app0,app1,app2,app3,omnibox0,webstore0"},
144 {0, 10, 0, "omnibox0,omnibox1,omnibox2,omnibox3,omnibox4,omnibox5"},
145 {0, 10, 1, "omnibox0,omnibox1,omnibox2,omnibox3,omnibox4,webstore0"},
146 {0, 10, 2, "omnibox0,omnibox1,omnibox2,omnibox3,webstore0,webstore1"},
147 {1, 10, 0, "app0,omnibox0,omnibox1,omnibox2,omnibox3,omnibox4"},
148 {2, 10, 0, "app0,app1,omnibox0,omnibox1,omnibox2,omnibox3"},
149 {2, 10, 1, "app0,app1,omnibox0,omnibox1,omnibox2,webstore0"},
150 {2, 10, 2, "app0,app1,omnibox0,omnibox1,webstore0,webstore1"},
151 {2, 0, 2, "app0,app1,webstore0,webstore1"},
155 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(kTestCases); ++i) {
156 app_provider()->set_count(kTestCases[i].app_results);
157 omnibox_provider()->set_count(kTestCases[i].omnibox_results);
158 webstore_provider()->set_count(kTestCases[i].webstore_results);
161 EXPECT_EQ(kTestCases[i].expected, GetResults()) << "Case " << i;
165 TEST_F(MixerTest, RemoveDuplicates) {
166 const std::string dup = "dup";
168 // This gives "dup0,dup1,dup2".
169 app_provider()->set_prefix(dup);
170 app_provider()->set_count(3);
172 // This gives "dup0,dup1".
173 omnibox_provider()->set_prefix(dup);
174 omnibox_provider()->set_count(2);
176 // This gives "dup0".
177 webstore_provider()->set_prefix(dup);
178 webstore_provider()->set_count(1);
182 // Only three results with unique id are kept.
183 EXPECT_EQ("dup0,dup1,dup2", GetResults());
187 } // namespace app_list