Index: chrome/browser/safe_browsing/safe_browsing_database_unittest.cc |
diff --git a/chrome/browser/safe_browsing/safe_browsing_database_unittest.cc b/chrome/browser/safe_browsing/safe_browsing_database_unittest.cc |
index f3b094b0b56fa3de1914afeffac0234baff60a85..8d80e82366caf5462102e0eac922113189a6d2d0 100644 |
--- a/chrome/browser/safe_browsing/safe_browsing_database_unittest.cc |
+++ b/chrome/browser/safe_browsing/safe_browsing_database_unittest.cc |
@@ -280,11 +280,13 @@ class SafeBrowsingDatabaseTest : public PlatformTest { |
new SafeBrowsingStoreFile(task_runner_); |
SafeBrowsingStoreFile* unwanted_software_store = |
new SafeBrowsingStoreFile(task_runner_); |
+ SafeBrowsingStoreFile* module_whitelist_store = |
+ new SafeBrowsingStoreFile(task_runner_); |
database_.reset(new SafeBrowsingDatabaseNew( |
task_runner_, browse_store, download_store, csd_whitelist_store, |
download_whitelist_store, inclusion_whitelist_store, |
- extension_blacklist_store, ip_blacklist_store, |
- unwanted_software_store)); |
+ extension_blacklist_store, ip_blacklist_store, unwanted_software_store, |
+ module_whitelist_store)); |
database_->Init(database_filename_); |
} |
@@ -440,10 +442,14 @@ TEST_F(SafeBrowsingDatabaseTest, ListNames) { |
chunks.push_back(AddChunkPrefixValue(11, "www.unwanted.com/software.html")); |
database_->InsertChunks(kUnwantedUrlList, chunks); |
+ chunks.clear(); |
+ chunks.push_back(AddChunkPrefixValue(12, "chrome.dll")); |
+ database_->InsertChunks(kModuleWhitelist, chunks); |
+ |
database_->UpdateFinished(true); |
GetListsInfo(&lists); |
- ASSERT_EQ(9U, lists.size()); |
+ ASSERT_EQ(10U, lists.size()); |
EXPECT_EQ(kMalwareList, lists[0].name); |
EXPECT_EQ("1", lists[0].adds); |
EXPECT_TRUE(lists[0].subs.empty()); |
@@ -471,6 +477,9 @@ TEST_F(SafeBrowsingDatabaseTest, ListNames) { |
EXPECT_EQ(kUnwantedUrlList, lists[8].name); |
EXPECT_EQ("11", lists[8].adds); |
EXPECT_TRUE(lists[8].subs.empty()); |
+ EXPECT_EQ(kModuleWhitelist, lists[9].name); |
+ EXPECT_EQ("12", lists[9].adds); |
+ EXPECT_TRUE(lists[9].subs.empty()); |
database_.reset(); |
} |
@@ -1153,7 +1162,7 @@ TEST_F(SafeBrowsingDatabaseTest, DISABLED_FileCorruptionHandling) { |
base::MessageLoop loop; |
SafeBrowsingStoreFile* store = new SafeBrowsingStoreFile(task_runner_); |
database_.reset(new SafeBrowsingDatabaseNew( |
- task_runner_, store, NULL, NULL, NULL, NULL, NULL, NULL, NULL)); |
+ task_runner_, store, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL)); |
database_->Init(database_filename_); |
// This will cause an empty database to be created. |
@@ -1328,7 +1337,7 @@ TEST_F(SafeBrowsingDatabaseTest, Whitelists) { |
// If the whitelist is disabled everything should match the whitelist. |
database_.reset(new SafeBrowsingDatabaseNew( |
task_runner_, new SafeBrowsingStoreFile(task_runner_), NULL, NULL, NULL, |
- NULL, NULL, NULL, NULL)); |
+ NULL, NULL, NULL, NULL, NULL)); |
database_->Init(database_filename_); |
for (const auto& test_case : kTestCases) { |
SCOPED_TRACE(std::string("Tested list at fault => ") + |