history_quick_provider_unittest.cc revision dc0f95d653279beabeb9817299e2902918ba123e
1// Copyright (c) 2011 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 "chrome/browser/autocomplete/history_quick_provider.h" 6 7#include <algorithm> 8#include <functional> 9#include <set> 10#include <string> 11#include <vector> 12 13#include "base/message_loop.h" 14#include "base/scoped_ptr.h" 15#include "base/utf_string_conversions.h" 16#include "chrome/browser/autocomplete/autocomplete_match.h" 17#include "chrome/browser/history/history.h" 18#include "chrome/browser/history/url_database.h" 19#include "chrome/browser/prefs/pref_service.h" 20#include "chrome/common/pref_names.h" 21#include "chrome/test/testing_browser_process.h" 22#include "chrome/test/testing_browser_process_test.h" 23#include "chrome/test/testing_profile.h" 24#include "content/browser/browser_thread.h" 25#include "testing/gtest/include/gtest/gtest.h" 26 27using base::Time; 28using base::TimeDelta; 29 30struct TestURLInfo { 31 std::string url; 32 std::string title; 33 int visit_count; 34 int typed_count; 35 int days_from_now; 36} quick_test_db[] = { 37 {"http://www.google.com/", "Google", 3, 3, 0}, 38 {"http://slashdot.org/favorite_page.html", "Favorite page", 200, 100, 0}, 39 {"http://kerneltrap.org/not_very_popular.html", "Less popular", 4, 0, 0}, 40 {"http://freshmeat.net/unpopular.html", "Unpopular", 1, 1, 0}, 41 {"http://news.google.com/?ned=us&topic=n", "Google News - U.S.", 2, 2, 0}, 42 {"http://news.google.com/", "Google News", 1, 1, 0}, 43 {"http://foo.com/", "Dir", 5, 5, 0}, 44 {"http://foo.com/dir/", "Dir", 2, 2, 0}, 45 {"http://foo.com/dir/another/", "Dir", 5, 1, 0}, 46 {"http://foo.com/dir/another/again/", "Dir", 10, 0, 0}, 47 {"http://foo.com/dir/another/again/myfile.html", "File", 10, 2, 0}, 48 {"http://startest.com/y/a", "A", 5, 2, 0}, 49 {"http://startest.com/y/b", "B", 1, 2, 0}, 50 {"http://startest.com/x/c", "C", 1, 1, 1}, 51 {"http://startest.com/x/d", "D", 1, 1, 1}, 52 {"http://startest.com/y/e", "E", 1, 1, 2}, 53 {"http://startest.com/y/f", "F", 1, 1, 2}, 54 {"http://startest.com/y/g", "G", 1, 1, 4}, 55 {"http://startest.com/y/h", "H", 1, 1, 4}, 56 {"http://startest.com/y/i", "I", 1, 1, 5}, 57 {"http://startest.com/y/j", "J", 1, 1, 5}, 58 {"http://startest.com/y/k", "K", 1, 1, 6}, 59 {"http://startest.com/y/l", "L", 1, 1, 6}, 60 {"http://startest.com/y/m", "M", 1, 1, 6}, 61 {"http://abcdefghixyzjklmnopqrstuvw.com/a", "An XYZ", 1, 1, 0}, 62 {"http://spaces.com/path%20with%20spaces/foo.html", "Spaces", 2, 2, 0}, 63 {"http://abcdefghijklxyzmnopqrstuvw.com/a", "An XYZ", 1, 1, 0}, 64 {"http://abcdefxyzghijklmnopqrstuvw.com/a", "An XYZ", 1, 1, 0}, 65 {"http://abcxyzdefghijklmnopqrstuvw.com/a", "An XYZ", 1, 1, 0}, 66 {"http://xyzabcdefghijklmnopqrstuvw.com/a", "An XYZ", 1, 1, 0}, 67}; 68 69class HistoryQuickProviderTest : public TestingBrowserProcessTest, 70 public ACProviderListener { 71 public: 72 HistoryQuickProviderTest() 73 : ui_thread_(BrowserThread::UI, &message_loop_), 74 file_thread_(BrowserThread::FILE, &message_loop_) {} 75 76 // ACProviderListener 77 virtual void OnProviderUpdate(bool updated_matches); 78 79 protected: 80 void SetUp() { 81 profile_.reset(new TestingProfile()); 82 profile_->CreateHistoryService(true, false); 83 profile_->CreateBookmarkModel(true); 84 profile_->BlockUntilBookmarkModelLoaded(); 85 history_service_ = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS); 86 EXPECT_TRUE(history_service_); 87 provider_ = new HistoryQuickProvider(this, profile_.get()); 88 FillData(); 89 } 90 91 void TearDown() { 92 provider_ = NULL; 93 } 94 95 // Fills test data into the history system. 96 void FillData(); 97 98 // Runs an autocomplete query on |text| and checks to see that the returned 99 // results' destination URLs match those provided. |expected_urls| does not 100 // need to be in sorted order. 101 void RunTest(const string16 text, 102 std::vector<std::string> expected_urls, 103 std::string expected_top_result); 104 105 MessageLoopForUI message_loop_; 106 BrowserThread ui_thread_; 107 BrowserThread file_thread_; 108 109 scoped_ptr<TestingProfile> profile_; 110 HistoryService* history_service_; 111 112 private: 113 scoped_refptr<HistoryQuickProvider> provider_; 114}; 115 116void HistoryQuickProviderTest::OnProviderUpdate(bool updated_matches) { 117 MessageLoop::current()->Quit(); 118} 119 120void HistoryQuickProviderTest::FillData() { 121 history::URLDatabase* db = history_service_->InMemoryDatabase(); 122 ASSERT_TRUE(db != NULL); 123 for (size_t i = 0; i < arraysize(quick_test_db); ++i) { 124 const TestURLInfo& cur = quick_test_db[i]; 125 const GURL current_url(cur.url); 126 Time visit_time = Time::Now() - TimeDelta::FromDays(cur.days_from_now); 127 128 history::URLRow url_info(current_url); 129 url_info.set_title(UTF8ToUTF16(cur.title)); 130 url_info.set_visit_count(cur.visit_count); 131 url_info.set_typed_count(cur.typed_count); 132 url_info.set_last_visit(visit_time); 133 url_info.set_hidden(false); 134 EXPECT_TRUE(db->AddURL(url_info)); 135 136 history_service_->AddPageWithDetails(current_url, UTF8ToUTF16(cur.title), 137 cur.visit_count, cur.typed_count, 138 visit_time, false, 139 history::SOURCE_BROWSED); 140 } 141 142 history::InMemoryURLIndex* index = 143 new history::InMemoryURLIndex(FilePath(FILE_PATH_LITERAL("/dummy"))); 144 PrefService* prefs = profile_->GetPrefs(); 145 std::string languages(prefs->GetString(prefs::kAcceptLanguages)); 146 index->Init(db, languages); 147 provider_->SetIndexForTesting(index); 148} 149 150class SetShouldContain : public std::unary_function<const std::string&, 151 std::set<std::string> > { 152 public: 153 explicit SetShouldContain(const ACMatches& matched_urls) { 154 for (ACMatches::const_iterator iter = matched_urls.begin(); 155 iter != matched_urls.end(); ++iter) 156 matches_.insert(iter->destination_url.spec()); 157 } 158 159 void operator()(const std::string& expected) { 160 EXPECT_EQ(1U, matches_.erase(expected)); 161 } 162 163 std::set<std::string> LeftOvers() const { return matches_; } 164 165 private: 166 std::set<std::string> matches_; 167}; 168 169void HistoryQuickProviderTest::RunTest(const string16 text, 170 std::vector<std::string> expected_urls, 171 std::string expected_top_result) { 172 std::sort(expected_urls.begin(), expected_urls.end()); 173 174 MessageLoop::current()->RunAllPending(); 175 AutocompleteInput input(text, string16(), false, false, true, false); 176 provider_->Start(input, false); 177 EXPECT_TRUE(provider_->done()); 178 179 ACMatches matches = provider_->matches(); 180 // If the number of expected and actual matches aren't equal then we need 181 // test no further, but let's do anyway so that we know which URLs failed. 182 EXPECT_EQ(expected_urls.size(), matches.size()); 183 184 // Verify that all expected URLs were found and that all found URLs 185 // were expected. 186 std::set<std::string> leftovers = 187 for_each(expected_urls.begin(), expected_urls.end(), 188 SetShouldContain(matches)).LeftOvers(); 189 EXPECT_TRUE(leftovers.empty()); 190 191 // See if we got the expected top scorer. 192 if (!matches.empty()) { 193 std::partial_sort(matches.begin(), matches.begin() + 1, 194 matches.end(), AutocompleteMatch::MoreRelevant); 195 EXPECT_EQ(expected_top_result, matches[0].destination_url.spec()); 196 } 197} 198 199TEST_F(HistoryQuickProviderTest, SimpleSingleMatch) { 200 string16 text(ASCIIToUTF16("slashdot")); 201 std::string expected_url("http://slashdot.org/favorite_page.html"); 202 std::vector<std::string> expected_urls; 203 expected_urls.push_back(expected_url); 204 RunTest(text, expected_urls, expected_url); 205} 206 207TEST_F(HistoryQuickProviderTest, MultiMatch) { 208 string16 text(ASCIIToUTF16("foo")); 209 std::vector<std::string> expected_urls; 210 expected_urls.push_back("http://foo.com/"); 211 expected_urls.push_back("http://foo.com/dir/"); 212 expected_urls.push_back("http://foo.com/dir/another/"); 213 expected_urls.push_back("http://foo.com/dir/another/again/"); 214 expected_urls.push_back("http://foo.com/dir/another/again/myfile.html"); 215 expected_urls.push_back("http://spaces.com/path%20with%20spaces/foo.html"); 216 RunTest(text, expected_urls, "http://foo.com/"); 217} 218 219TEST_F(HistoryQuickProviderTest, StartRelativeMatch) { 220 string16 text(ASCIIToUTF16("xyz")); 221 std::vector<std::string> expected_urls; 222 expected_urls.push_back("http://xyzabcdefghijklmnopqrstuvw.com/a"); 223 expected_urls.push_back("http://abcxyzdefghijklmnopqrstuvw.com/a"); 224 expected_urls.push_back("http://abcdefxyzghijklmnopqrstuvw.com/a"); 225 expected_urls.push_back("http://abcdefghixyzjklmnopqrstuvw.com/a"); 226 expected_urls.push_back("http://abcdefghijklxyzmnopqrstuvw.com/a"); 227 RunTest(text, expected_urls, "http://xyzabcdefghijklmnopqrstuvw.com/a"); 228} 229 230TEST_F(HistoryQuickProviderTest, RecencyMatch) { 231 string16 text(ASCIIToUTF16("startest")); 232 std::vector<std::string> expected_urls; 233 expected_urls.push_back("http://startest.com/y/a"); 234 expected_urls.push_back("http://startest.com/y/b"); 235 expected_urls.push_back("http://startest.com/x/c"); 236 expected_urls.push_back("http://startest.com/x/d"); 237 expected_urls.push_back("http://startest.com/y/e"); 238 expected_urls.push_back("http://startest.com/y/f"); 239 RunTest(text, expected_urls, "http://startest.com/y/a"); 240} 241