OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 // | 4 // |
5 // Unit tests for the SafeBrowsing storage system. | 5 // Unit tests for the SafeBrowsing storage system. |
6 | 6 |
7 #include "chrome/browser/safe_browsing/safe_browsing_database.h" | 7 #include "chrome/browser/safe_browsing/safe_browsing_database.h" |
8 | 8 |
9 #include "base/files/file_util.h" | 9 #include "base/files/file_util.h" |
10 #include "base/files/scoped_temp_dir.h" | 10 #include "base/files/scoped_temp_dir.h" |
11 #include "base/logging.h" | 11 #include "base/logging.h" |
12 #include "base/memory/scoped_vector.h" | 12 #include "base/memory/scoped_vector.h" |
13 #include "base/message_loop/message_loop.h" | 13 #include "base/message_loop/message_loop.h" |
14 #include "base/sha1.h" | 14 #include "base/sha1.h" |
15 #include "base/strings/string_number_conversions.h" | 15 #include "base/strings/string_number_conversions.h" |
16 #include "base/strings/string_split.h" | 16 #include "base/strings/string_split.h" |
17 #include "base/time/time.h" | 17 #include "base/time/time.h" |
18 #include "chrome/browser/safe_browsing/chunk.pb.h" | 18 #include "chrome/browser/safe_browsing/chunk.pb.h" |
19 #include "chrome/browser/safe_browsing/safe_browsing_store_file.h" | 19 #include "chrome/browser/safe_browsing/safe_browsing_store_file.h" |
20 #include "content/public/test/test_browser_thread_bundle.h" | |
21 #include "crypto/sha2.h" | 20 #include "crypto/sha2.h" |
22 #include "net/base/net_util.h" | 21 #include "net/base/net_util.h" |
23 #include "sql/connection.h" | 22 #include "sql/connection.h" |
24 #include "sql/statement.h" | 23 #include "sql/statement.h" |
25 #include "testing/gtest/include/gtest/gtest.h" | 24 #include "testing/gtest/include/gtest/gtest.h" |
26 #include "testing/platform_test.h" | 25 #include "testing/platform_test.h" |
27 #include "url/gurl.h" | 26 #include "url/gurl.h" |
28 | 27 |
29 using base::Time; | 28 using base::Time; |
30 using base::TimeDelta; | 29 using base::TimeDelta; |
(...skipping 1201 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1232 urls.push_back(GURL(std::string("https://") + kEvil1Url2)); | 1231 urls.push_back(GURL(std::string("https://") + kEvil1Url2)); |
1233 EXPECT_TRUE(database_->ContainsDownloadUrl(urls, &prefix_hits)); | 1232 EXPECT_TRUE(database_->ContainsDownloadUrl(urls, &prefix_hits)); |
1234 ASSERT_EQ(2U, prefix_hits.size()); | 1233 ASSERT_EQ(2U, prefix_hits.size()); |
1235 EXPECT_EQ(SBPrefixForString(kEvil1Url1), prefix_hits[0]); | 1234 EXPECT_EQ(SBPrefixForString(kEvil1Url1), prefix_hits[0]); |
1236 EXPECT_EQ(SBPrefixForString(kEvil1Url2), prefix_hits[1]); | 1235 EXPECT_EQ(SBPrefixForString(kEvil1Url2), prefix_hits[1]); |
1237 database_.reset(); | 1236 database_.reset(); |
1238 } | 1237 } |
1239 | 1238 |
1240 // Checks that the whitelists are handled properly. | 1239 // Checks that the whitelists are handled properly. |
1241 TEST_F(SafeBrowsingDatabaseTest, Whitelists) { | 1240 TEST_F(SafeBrowsingDatabaseTest, Whitelists) { |
1242 database_.reset(); | |
1243 | |
1244 // We expect all calls to ContainsCsdWhitelistedUrl in particular to be made | |
1245 // from the IO thread. In general the whitelist lookups are thread-safe. | |
1246 content::TestBrowserThreadBundle thread_bundle_; | |
1247 | |
1248 // If the whitelist is disabled everything should match the whitelist. | 1241 // If the whitelist is disabled everything should match the whitelist. |
1249 database_.reset(new SafeBrowsingDatabaseNew(new SafeBrowsingStoreFile(), | 1242 database_.reset(new SafeBrowsingDatabaseNew(new SafeBrowsingStoreFile(), |
1250 NULL, NULL, NULL, NULL, NULL, | 1243 NULL, NULL, NULL, NULL, NULL, |
1251 NULL, NULL)); | 1244 NULL, NULL)); |
1252 database_->Init(database_filename_); | 1245 database_->Init(database_filename_); |
1253 EXPECT_TRUE(database_->ContainsDownloadWhitelistedUrl( | 1246 EXPECT_TRUE(database_->ContainsDownloadWhitelistedUrl( |
1254 GURL(std::string("http://www.phishing.com/")))); | 1247 GURL(std::string("http://www.phishing.com/")))); |
1255 EXPECT_TRUE(database_->ContainsDownloadWhitelistedUrl( | 1248 EXPECT_TRUE(database_->ContainsDownloadWhitelistedUrl( |
1256 GURL(std::string("http://www.phishing.com/")))); | 1249 GURL(std::string("http://www.phishing.com/")))); |
1257 EXPECT_TRUE(database_->ContainsDownloadWhitelistedString("asdf")); | 1250 EXPECT_TRUE(database_->ContainsDownloadWhitelistedString("asdf")); |
(...skipping 879 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
2137 GURL(std::string("http://") + kExampleCollision), | 2130 GURL(std::string("http://") + kExampleCollision), |
2138 &prefix_hits, &cache_hits)); | 2131 &prefix_hits, &cache_hits)); |
2139 ASSERT_EQ(1U, prefix_hits.size()); | 2132 ASSERT_EQ(1U, prefix_hits.size()); |
2140 EXPECT_EQ(SBPrefixForString(kExampleCollision), prefix_hits[0]); | 2133 EXPECT_EQ(SBPrefixForString(kExampleCollision), prefix_hits[0]); |
2141 EXPECT_TRUE(cache_hits.empty()); | 2134 EXPECT_TRUE(cache_hits.empty()); |
2142 | 2135 |
2143 // This prefix collides, but no full hash match. | 2136 // This prefix collides, but no full hash match. |
2144 EXPECT_FALSE(database_->ContainsBrowseUrl( | 2137 EXPECT_FALSE(database_->ContainsBrowseUrl( |
2145 GURL(std::string("http://") + kExampleFine), &prefix_hits, &cache_hits)); | 2138 GURL(std::string("http://") + kExampleFine), &prefix_hits, &cache_hits)); |
2146 } | 2139 } |
OLD | NEW |