| 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 #include "chrome/browser/safe_browsing/download_protection_service.h" | 5 #include "chrome/browser/safe_browsing/download_protection_service.h" |
| 6 | 6 |
| 7 #include <map> | 7 #include <map> |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/base_paths.h" | 10 #include "base/base_paths.h" |
| (...skipping 183 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 194 base::Bind(&OnSafeBrowsingResult, | 194 base::Bind(&OnSafeBrowsingResult, |
| 195 base::Owned(check))); | 195 base::Owned(check))); |
| 196 } | 196 } |
| 197 | 197 |
| 198 class DownloadProtectionServiceTest : public testing::Test { | 198 class DownloadProtectionServiceTest : public testing::Test { |
| 199 protected: | 199 protected: |
| 200 DownloadProtectionServiceTest() | 200 DownloadProtectionServiceTest() |
| 201 : test_browser_thread_bundle_( | 201 : test_browser_thread_bundle_( |
| 202 content::TestBrowserThreadBundle::IO_MAINLOOP) { | 202 content::TestBrowserThreadBundle::IO_MAINLOOP) { |
| 203 } | 203 } |
| 204 virtual void SetUp() { | 204 void SetUp() override { |
| 205 #if defined(OS_MACOSX) | 205 #if defined(OS_MACOSX) |
| 206 field_trial_list_.reset(new base::FieldTrialList( | 206 field_trial_list_.reset(new base::FieldTrialList( |
| 207 new metrics::SHA1EntropyProvider("42"))); | 207 new metrics::SHA1EntropyProvider("42"))); |
| 208 ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial( | 208 ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial( |
| 209 "SafeBrowsingOSXClientDownloadPings", | 209 "SafeBrowsingOSXClientDownloadPings", |
| 210 "Enabled")); | 210 "Enabled")); |
| 211 #endif | 211 #endif |
| 212 // Start real threads for the IO and File threads so that the DCHECKs | 212 // Start real threads for the IO and File threads so that the DCHECKs |
| 213 // to test that we're on the correct thread work. | 213 // to test that we're on the correct thread work. |
| 214 sb_service_ = new StrictMock<FakeSafeBrowsingService>(); | 214 sb_service_ = new StrictMock<FakeSafeBrowsingService>(); |
| 215 sb_service_->Initialize(); | 215 sb_service_->Initialize(); |
| 216 binary_feature_extractor_ = new StrictMock<MockBinaryFeatureExtractor>(); | 216 binary_feature_extractor_ = new StrictMock<MockBinaryFeatureExtractor>(); |
| 217 download_service_ = sb_service_->download_protection_service(); | 217 download_service_ = sb_service_->download_protection_service(); |
| 218 download_service_->binary_feature_extractor_ = binary_feature_extractor_; | 218 download_service_->binary_feature_extractor_ = binary_feature_extractor_; |
| 219 download_service_->SetEnabled(true); | 219 download_service_->SetEnabled(true); |
| 220 base::RunLoop().RunUntilIdle(); | 220 base::RunLoop().RunUntilIdle(); |
| 221 has_result_ = false; | 221 has_result_ = false; |
| 222 | 222 |
| 223 base::FilePath source_path; | 223 base::FilePath source_path; |
| 224 ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &source_path)); | 224 ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &source_path)); |
| 225 testdata_path_ = source_path | 225 testdata_path_ = source_path |
| 226 .AppendASCII("chrome") | 226 .AppendASCII("chrome") |
| 227 .AppendASCII("test") | 227 .AppendASCII("test") |
| 228 .AppendASCII("data") | 228 .AppendASCII("data") |
| 229 .AppendASCII("safe_browsing") | 229 .AppendASCII("safe_browsing") |
| 230 .AppendASCII("download_protection"); | 230 .AppendASCII("download_protection"); |
| 231 } | 231 } |
| 232 | 232 |
| 233 virtual void TearDown() { | 233 void TearDown() override { |
| 234 sb_service_->ShutDown(); | 234 sb_service_->ShutDown(); |
| 235 // Flush all of the thread message loops to ensure that there are no | 235 // Flush all of the thread message loops to ensure that there are no |
| 236 // tasks currently running. | 236 // tasks currently running. |
| 237 FlushThreadMessageLoops(); | 237 FlushThreadMessageLoops(); |
| 238 sb_service_ = NULL; | 238 sb_service_ = NULL; |
| 239 } | 239 } |
| 240 | 240 |
| 241 bool RequestContainsResource(const ClientDownloadRequest& request, | 241 bool RequestContainsResource(const ClientDownloadRequest& request, |
| 242 ClientDownloadRequest::ResourceType type, | 242 ClientDownloadRequest::ResourceType type, |
| 243 const std::string& url, | 243 const std::string& url, |
| (...skipping 1340 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1584 EXPECT_THAT(whitelist_strings, ElementsAre(cert_base + "/OU=unit")); | 1584 EXPECT_THAT(whitelist_strings, ElementsAre(cert_base + "/OU=unit")); |
| 1585 | 1585 |
| 1586 cert = ReadTestCertificate("test_c.pem"); | 1586 cert = ReadTestCertificate("test_c.pem"); |
| 1587 ASSERT_TRUE(cert.get()); | 1587 ASSERT_TRUE(cert.get()); |
| 1588 whitelist_strings.clear(); | 1588 whitelist_strings.clear(); |
| 1589 GetCertificateWhitelistStrings( | 1589 GetCertificateWhitelistStrings( |
| 1590 *cert.get(), *issuer_cert.get(), &whitelist_strings); | 1590 *cert.get(), *issuer_cert.get(), &whitelist_strings); |
| 1591 EXPECT_THAT(whitelist_strings, ElementsAre()); | 1591 EXPECT_THAT(whitelist_strings, ElementsAre()); |
| 1592 } | 1592 } |
| 1593 } // namespace safe_browsing | 1593 } // namespace safe_browsing |
| OLD | NEW |