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.
4
5 #include <string>
6
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 "testing/gtest/include/gtest/gtest.h"
15 #include "ui/app_list/app_list_model.h"
16 #include "ui/app_list/search_provider.h"
17
18 namespace app_list {
19 namespace test {
20
21 class TestSearchResult : public ChromeSearchResult {
22 public:
TestSearchResult(const std::string & id,double relevance)23 TestSearchResult(const std::string& id, double relevance) {
24 set_id(id);
25 set_title(base::UTF8ToUTF16(id));
26 set_relevance(relevance);
27 }
~TestSearchResult()28 virtual ~TestSearchResult() {}
29
30 private:
31 // ChromeSearchResult overides:
Open(int event_flags)32 virtual void Open(int event_flags) OVERRIDE {}
InvokeAction(int action_index,int event_flags)33 virtual void InvokeAction(int action_index, int event_flags) OVERRIDE {}
Duplicate()34 virtual scoped_ptr<ChromeSearchResult> Duplicate() OVERRIDE {
35 return scoped_ptr<ChromeSearchResult>(
36 new TestSearchResult(id(), relevance())).Pass();
37 }
GetType()38 virtual ChromeSearchResultType GetType() OVERRIDE {
39 return SEARCH_RESULT_TYPE_BOUNDARY;
40 }
41
42 DISALLOW_COPY_AND_ASSIGN(TestSearchResult);
43 };
44
45 class TestSearchProvider : public SearchProvider {
46 public:
TestSearchProvider(const std::string & prefix)47 explicit TestSearchProvider(const std::string& prefix)
48 : prefix_(prefix),
49 count_(0) {}
~TestSearchProvider()50 virtual ~TestSearchProvider() {}
51
52 // SearchProvider overrides:
Start(const base::string16 & query)53 virtual void Start(const base::string16& query) OVERRIDE {
54 ClearResults();
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<SearchResult>(new TestSearchResult(id, relevance)).Pass());
60 }
61 }
Stop()62 virtual void Stop() OVERRIDE {}
63
set_prefix(const std::string & prefix)64 void set_prefix(const std::string& prefix) { prefix_ = prefix; }
set_count(size_t count)65 void set_count(size_t count) { count_ = count; }
66
67 private:
68 std::string prefix_;
69 size_t count_;
70
71 DISALLOW_COPY_AND_ASSIGN(TestSearchProvider);
72 };
73
74 class MixerTest : public testing::Test {
75 public:
MixerTest()76 MixerTest() {}
~MixerTest()77 virtual ~MixerTest() {}
78
79 // testing::Test overrides:
SetUp()80 virtual void SetUp() OVERRIDE {
81 results_.reset(new AppListModel::SearchResults);
82
83 providers_.push_back(new TestSearchProvider("app"));
84 providers_.push_back(new TestSearchProvider("omnibox"));
85 providers_.push_back(new TestSearchProvider("webstore"));
86 providers_.push_back(new TestSearchProvider("people"));
87
88 mixer_.reset(new Mixer(results_.get()));
89 mixer_->Init();
90 mixer_->AddProviderToGroup(Mixer::MAIN_GROUP, providers_[0]);
91 mixer_->AddProviderToGroup(Mixer::OMNIBOX_GROUP, providers_[1]);
92 mixer_->AddProviderToGroup(Mixer::WEBSTORE_GROUP, providers_[2]);
93 mixer_->AddProviderToGroup(Mixer::PEOPLE_GROUP, providers_[3]);
94 }
95
RunQuery()96 void RunQuery() {
97 const base::string16 query;
98
99 for (size_t i = 0; i < providers_.size(); ++i) {
100 providers_[i]->Start(query);
101 providers_[i]->Stop();
102 }
103
104 mixer_->MixAndPublish(KnownResults());
105 }
106
GetResults() const107 std::string GetResults() const {
108 std::string result;
109 for (size_t i = 0; i < results_->item_count(); ++i) {
110 if (!result.empty())
111 result += ',';
112
113 result += base::UTF16ToUTF8(results_->GetItemAt(i)->title());
114 }
115
116 return result;
117 }
118
mixer()119 Mixer* mixer() { return mixer_.get(); }
app_provider()120 TestSearchProvider* app_provider() { return providers_[0]; }
omnibox_provider()121 TestSearchProvider* omnibox_provider() { return providers_[1]; }
webstore_provider()122 TestSearchProvider* webstore_provider() { return providers_[2]; }
123
124 private:
125 scoped_ptr<Mixer> mixer_;
126 scoped_ptr<AppListModel::SearchResults> results_;
127
128 ScopedVector<TestSearchProvider> providers_;
129
130 DISALLOW_COPY_AND_ASSIGN(MixerTest);
131 };
132
TEST_F(MixerTest,Basic)133 TEST_F(MixerTest, Basic) {
134 struct TestCase {
135 const size_t app_results;
136 const size_t omnibox_results;
137 const size_t webstore_results;
138 const char* expected;
139 } kTestCases[] = {
140 {0, 0, 0, ""},
141 {4, 6, 2, "app0,app1,app2,app3,omnibox0,webstore0"},
142 {10, 10, 10, "app0,app1,app2,app3,omnibox0,webstore0"},
143 {0, 10, 0, "omnibox0,omnibox1,omnibox2,omnibox3,omnibox4,omnibox5"},
144 {0, 10, 1, "omnibox0,omnibox1,omnibox2,omnibox3,omnibox4,webstore0"},
145 {0, 10, 2, "omnibox0,omnibox1,omnibox2,omnibox3,webstore0,webstore1"},
146 {1, 10, 0, "app0,omnibox0,omnibox1,omnibox2,omnibox3,omnibox4"},
147 {2, 10, 0, "app0,app1,omnibox0,omnibox1,omnibox2,omnibox3"},
148 {2, 10, 1, "app0,app1,omnibox0,omnibox1,omnibox2,webstore0"},
149 {2, 10, 2, "app0,app1,omnibox0,omnibox1,webstore0,webstore1"},
150 {2, 0, 2, "app0,app1,webstore0,webstore1"},
151 {0, 0, 0, ""},
152 };
153
154 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(kTestCases); ++i) {
155 app_provider()->set_count(kTestCases[i].app_results);
156 omnibox_provider()->set_count(kTestCases[i].omnibox_results);
157 webstore_provider()->set_count(kTestCases[i].webstore_results);
158 RunQuery();
159
160 EXPECT_EQ(kTestCases[i].expected, GetResults()) << "Case " << i;
161 }
162 }
163
TEST_F(MixerTest,RemoveDuplicates)164 TEST_F(MixerTest, RemoveDuplicates) {
165 const std::string dup = "dup";
166
167 // This gives "dup0,dup1,dup2".
168 app_provider()->set_prefix(dup);
169 app_provider()->set_count(3);
170
171 // This gives "dup0,dup1".
172 omnibox_provider()->set_prefix(dup);
173 omnibox_provider()->set_count(2);
174
175 // This gives "dup0".
176 webstore_provider()->set_prefix(dup);
177 webstore_provider()->set_count(1);
178
179 RunQuery();
180
181 // Only three results with unique id are kept.
182 EXPECT_EQ("dup0,dup1,dup2", GetResults());
183 }
184
185 } // namespace test
186 } // namespace app_list
187