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 "base/bind.h" | 5 #include "base/bind.h" |
6 #include "base/command_line.h" | 6 #include "base/command_line.h" |
7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
8 #include "base/macros.h" | 8 #include "base/macros.h" |
| 9 #include "base/memory/ptr_util.h" |
9 #include "build/build_config.h" | 10 #include "build/build_config.h" |
10 #include "chrome/browser/download/download_danger_prompt.h" | 11 #include "chrome/browser/download/download_danger_prompt.h" |
11 #include "chrome/browser/profiles/profile.h" | 12 #include "chrome/browser/profiles/profile.h" |
12 #include "chrome/browser/safe_browsing/download_protection_service.h" | 13 #include "chrome/browser/safe_browsing/download_protection_service.h" |
13 #include "chrome/browser/safe_browsing/test_safe_browsing_service.h" | 14 #include "chrome/browser/safe_browsing/test_safe_browsing_service.h" |
14 #include "chrome/browser/ui/browser.h" | 15 #include "chrome/browser/ui/browser.h" |
15 #include "chrome/browser/ui/browser_commands.h" | 16 #include "chrome/browser/ui/browser_commands.h" |
16 #include "chrome/browser/ui/browser_tabstrip.h" | 17 #include "chrome/browser/ui/browser_tabstrip.h" |
17 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 18 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
18 #include "chrome/test/base/in_process_browser_test.h" | 19 #include "chrome/test/base/in_process_browser_test.h" |
(...skipping 130 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
149 | 150 |
150 DownloadDangerPrompt* prompt() { return prompt_; } | 151 DownloadDangerPrompt* prompt() { return prompt_; } |
151 | 152 |
152 private: | 153 private: |
153 void SetUpDownloadItemExpectations( | 154 void SetUpDownloadItemExpectations( |
154 const content::DownloadDangerType& danger_type, | 155 const content::DownloadDangerType& danger_type, |
155 const std::string& token) { | 156 const std::string& token) { |
156 EXPECT_CALL(download_, GetFileNameToReportUser()).WillRepeatedly(Return( | 157 EXPECT_CALL(download_, GetFileNameToReportUser()).WillRepeatedly(Return( |
157 base::FilePath(FILE_PATH_LITERAL("evil.exe")))); | 158 base::FilePath(FILE_PATH_LITERAL("evil.exe")))); |
158 EXPECT_CALL(download_, GetDangerType()).WillRepeatedly(Return(danger_type)); | 159 EXPECT_CALL(download_, GetDangerType()).WillRepeatedly(Return(danger_type)); |
159 DownloadProtectionService::DownloadPingToken* token_obj = | 160 auto token_obj = |
160 new DownloadProtectionService::DownloadPingToken(token); | 161 base::MakeUnique<DownloadProtectionService::DownloadPingToken>(token); |
161 download_.SetUserData(DownloadProtectionService::kDownloadPingTokenKey, | 162 download_.SetUserData(DownloadProtectionService::kDownloadPingTokenKey, |
162 token_obj); | 163 std::move(token_obj)); |
163 } | 164 } |
164 | 165 |
165 void SetUpSafeBrowsingReportExpectations( | 166 void SetUpSafeBrowsingReportExpectations( |
166 bool did_proceed, | 167 bool did_proceed, |
167 const ClientDownloadResponse::Verdict& download_verdict, | 168 const ClientDownloadResponse::Verdict& download_verdict, |
168 const std::string& token, | 169 const std::string& token, |
169 bool from_download_api) { | 170 bool from_download_api) { |
170 ClientSafeBrowsingReportRequest expected_report; | 171 ClientSafeBrowsingReportRequest expected_report; |
171 expected_report.set_url(GURL(kTestDownloadUrl).spec()); | 172 expected_report.set_url(GURL(kTestDownloadUrl).spec()); |
172 if (from_download_api) | 173 if (from_download_api) |
(...skipping 161 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
334 | 335 |
335 // Prefix for test instantiations intentionally left blank since the test | 336 // Prefix for test instantiations intentionally left blank since the test |
336 // fixture class has a single parameterization. | 337 // fixture class has a single parameterization. |
337 INSTANTIATE_TEST_CASE_P(, | 338 INSTANTIATE_TEST_CASE_P(, |
338 DownloadDangerPromptTest, | 339 DownloadDangerPromptTest, |
339 ::testing::Values(SecondaryUiMd::ENABLED, | 340 ::testing::Values(SecondaryUiMd::ENABLED, |
340 SecondaryUiMd::DISABLED), | 341 SecondaryUiMd::DISABLED), |
341 &SecondaryUiMdStatusToString); | 342 &SecondaryUiMdStatusToString); |
342 | 343 |
343 } // namespace safe_browsing | 344 } // namespace safe_browsing |
OLD | NEW |