app_list: Added MixerTest.KnownResultsPriority test case.

ui/app_list/search/mixer_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <set>
 #include <string>
 
 #include "base/memory/scoped_vector.h"
 #include "base/strings/string16.h"
 #include "base/strings/stringprintf.h"
@@ skipping 102 matching lines @@
   }
 
   void RunQuery() {
     const base::string16 query;
 
     for (size_t i = 0; i < providers_.size(); ++i) {
       providers_[i]->Start(is_voice_query_, query);
       providers_[i]->Stop();
     }
 
-    mixer_->MixAndPublish(is_voice_query_, KnownResults());
+    mixer_->MixAndPublish(is_voice_query_, known_results_);
   }
 
   std::string GetResults() const {
     std::string result;
     for (size_t i = 0; i < results_->item_count(); ++i) {
       if (!result.empty())
         result += ',';
 
       result += base::UTF16ToUTF8(results_->GetItemAt(i)->title());
     }
 
     return result;
   }
 
   Mixer* mixer() { return mixer_.get(); }
   TestSearchProvider* app_provider() { return providers_[0]; }
   TestSearchProvider* omnibox_provider() { return providers_[1]; }
   TestSearchProvider* webstore_provider() { return providers_[2]; }
   TestSearchProvider* people_provider() { return providers_[3]; }
 
   // Sets whether test runs should be treated as a voice query.
   void set_is_voice_query(bool is_voice_query) {
     is_voice_query_ = is_voice_query;
   }
 
+  void AddKnownResult(const std::string& id, KnownResultType type) {
+    known_results_[id] = type;
+  }
+
  private:
   scoped_ptr<Mixer> mixer_;
   scoped_ptr<AppListModel::SearchResults> results_;
+  KnownResults known_results_;
 
   bool is_voice_query_;
 
   ScopedVector<TestSearchProvider> providers_;
 
   DISALLOW_COPY_AND_ASSIGN(MixerTest);
 };
 
 TEST_F(MixerTest, Basic) {
   struct TestCase {
@@ skipping 49 matching lines @@
   // This gives "dup0".
   webstore_provider()->set_prefix(dup);
   webstore_provider()->set_count(1);
 
   RunQuery();
 
   // Only three results with unique id are kept.
   EXPECT_EQ("dup0,dup1,dup2", GetResults());
 }
 
+// Tests that "known results" have priority over others.
+TEST_F(MixerTest, KnownResultsPriority) {
+  // This gives omnibox 0 -- 5.
+  omnibox_provider()->set_count(6);
+
+  // omnibox 1 -- 4 are "known results".
+  AddKnownResult("omnibox1", PREFIX_SECONDARY);
+  AddKnownResult("omnibox2", PERFECT_SECONDARY);
+  AddKnownResult("omnibox3", PREFIX_PRIMARY);
+  AddKnownResult("omnibox4", PERFECT_PRIMARY);
+
+  RunQuery();
+
+  // omnibox 1 -- 4 should be prioritised over the others. They should be
+  // ordered 4, 3, 2, 1 (in order of match quality).
+  EXPECT_EQ("omnibox4,omnibox3,omnibox2,omnibox1,omnibox0,omnibox5",
+            GetResults());
+}
+
 TEST_F(MixerTest, VoiceQuery) {
   omnibox_provider()->set_count(3);
   RunQuery();
   EXPECT_EQ("omnibox0,omnibox1,omnibox2", GetResults());
 
   // Set "omnibox1" as a voice result. Do not expect any changes (as this is not
   // a voice query).
   omnibox_provider()->set_as_voice_result(1);
   RunQuery();
   EXPECT_EQ("omnibox0,omnibox1,omnibox2", GetResults());
@@ skipping 92 matching lines @@
   EXPECT_EQ(old_ui_result_ids[0],
             TestSearchResult::GetInstanceId(ui_results.GetItemAt(3)));
   EXPECT_EQ(old_ui_result_ids[1],
             TestSearchResult::GetInstanceId(ui_results.GetItemAt(0)));
   EXPECT_EQ(old_ui_result_ids[2],
             TestSearchResult::GetInstanceId(ui_results.GetItemAt(2)));
 }
 
 }  // namespace test
 }  // namespace app_list