Chromium Code Reviews| Index: chrome/browser/autocomplete/in_memory_url_index_unittest.cc |
| diff --git a/chrome/browser/autocomplete/in_memory_url_index_unittest.cc b/chrome/browser/autocomplete/in_memory_url_index_unittest.cc |
| index 401c5bf107662589af0f8fcfea05dcb21088384f..54be09fa73cef4f4b478dab617bfb8f8f9107aa5 100644 |
| --- a/chrome/browser/autocomplete/in_memory_url_index_unittest.cc |
| +++ b/chrome/browser/autocomplete/in_memory_url_index_unittest.cc |
| @@ -14,6 +14,7 @@ |
| #include "base/run_loop.h" |
| #include "base/strings/string16.h" |
| #include "base/strings/string_util.h" |
| +#include "base/strings/stringprintf.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "chrome/browser/autocomplete/in_memory_url_index.h" |
| #include "chrome/browser/autocomplete/url_index_private_data.h" |
| @@ -27,6 +28,7 @@ |
| #include "components/history/core/browser/history_database.h" |
| #include "components/history/core/browser/history_service.h" |
| #include "components/omnibox/in_memory_url_index_types.h" |
| +#include "content/public/browser/browser_thread.h" |
| #include "content/public/test/test_browser_thread_bundle.h" |
| #include "sql/transaction.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| @@ -48,6 +50,7 @@ namespace { |
| const size_t kInvalid = base::string16::npos; |
| const size_t kMaxMatches = 3; |
| const char kTestLanguages[] = "en,ja,hi,zh"; |
| +const char kClientWhitelistedScheme[] = "chrome"; |
|
Peter Kasting
2015/06/16 22:18:07
Nit: Let's use "xyz" here or something to make it
blundell
2015/06/17 07:55:44
Done.
|
| // Helper function to set lower case |lower_string| and |lower_terms| (words |
| // list) based on supplied |search_string| and |cursor_position|. If |
| @@ -321,8 +324,12 @@ bool InMemoryURLIndexTest::InitializeInMemoryURLIndexInSetUp() const { |
| void InMemoryURLIndexTest::InitializeInMemoryURLIndex() { |
| DCHECK(!url_index_); |
| - url_index_.reset(new InMemoryURLIndex(nullptr, history_service_, |
| - base::FilePath(), kTestLanguages)); |
| + |
| + std::set<std::string> client_schemes_to_whitelist; |
| + client_schemes_to_whitelist.insert(kClientWhitelistedScheme); |
| + url_index_.reset(new InMemoryURLIndex( |
| + nullptr, history_service_, content::BrowserThread::GetBlockingPool(), |
| + base::FilePath(), kTestLanguages, client_schemes_to_whitelist)); |
| url_index_->Init(); |
| url_index_->RebuildFromHistory(history_database_); |
| } |
| @@ -942,18 +949,20 @@ TEST_F(InMemoryURLIndexTest, ExpireRow) { |
| } |
| TEST_F(InMemoryURLIndexTest, WhitelistedURLs) { |
| + std::string client_whitelisted_url = |
| + base::StringPrintf("%s://foo", kClientWhitelistedScheme); |
| struct TestData { |
| const std::string url_spec; |
| const bool expected_is_whitelisted; |
| } data[] = { |
| // URLs with whitelisted schemes. |
| { "about:histograms", true }, |
| - { "chrome://settings", true }, |
| { "file://localhost/Users/joeschmoe/sekrets", true }, |
| { "ftp://public.mycompany.com/myfile.txt", true }, |
| { "http://www.google.com/translate", true }, |
| { "https://www.gmail.com/", true }, |
| { "mailto:support@google.com", true }, |
| + { client_whitelisted_url, true }, |
| // URLs with unacceptable schemes. |
| { "aaa://www.dummyhost.com;frammy", false }, |
| { "aaas://www.dummyhost.com;frammy", false }, |
| @@ -1314,8 +1323,9 @@ class InMemoryURLIndexCacheTest : public testing::Test { |
| void InMemoryURLIndexCacheTest::SetUp() { |
| ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| base::FilePath path(temp_dir_.path()); |
| - url_index_.reset( |
| - new InMemoryURLIndex(nullptr, nullptr, path, kTestLanguages)); |
| + url_index_.reset(new InMemoryURLIndex( |
| + nullptr, nullptr, content::BrowserThread::GetBlockingPool(), path, |
| + kTestLanguages, std::set<std::string>())); |
| } |
| void InMemoryURLIndexCacheTest::TearDown() { |