| Index: chrome/browser/download/chrome_download_manager_delegate_unittest.cc
|
| diff --git a/chrome/browser/download/chrome_download_manager_delegate_unittest.cc b/chrome/browser/download/chrome_download_manager_delegate_unittest.cc
|
| index b04d24744f57e3aac2ad30217aa28c07f8841700..da500dcf04cf355b245c92357123214a96b7a285 100644
|
| --- a/chrome/browser/download/chrome_download_manager_delegate_unittest.cc
|
| +++ b/chrome/browser/download/chrome_download_manager_delegate_unittest.cc
|
| @@ -47,6 +47,7 @@ using ::testing::SetArgPointee;
|
| using ::testing::WithArg;
|
| using ::testing::_;
|
| using content::DownloadItem;
|
| +using safe_browsing::DownloadFileType;
|
|
|
| namespace {
|
|
|
| @@ -423,7 +424,7 @@ TEST_F(ChromeDownloadManagerDelegateTest, MaybeDangerousContent) {
|
| DownloadTargetInfo target_info;
|
| DetermineDownloadTarget(download_item.get(), &target_info);
|
|
|
| - EXPECT_EQ(download_util::DANGEROUS,
|
| + EXPECT_EQ(DownloadFileType::DANGEROUS,
|
| DownloadItemModel(download_item.get()).GetDangerLevel());
|
| EXPECT_EQ(content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| target_info.danger_type);
|
| @@ -436,7 +437,7 @@ TEST_F(ChromeDownloadManagerDelegateTest, MaybeDangerousContent) {
|
| .WillRepeatedly(Return(kSafeContentDisposition));
|
| DownloadTargetInfo target_info;
|
| DetermineDownloadTarget(download_item.get(), &target_info);
|
| - EXPECT_EQ(download_util::NOT_DANGEROUS,
|
| + EXPECT_EQ(DownloadFileType::NOT_DANGEROUS,
|
| DownloadItemModel(download_item.get()).GetDangerLevel());
|
| EXPECT_EQ(content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| target_info.danger_type);
|
| @@ -449,7 +450,7 @@ TEST_F(ChromeDownloadManagerDelegateTest, MaybeDangerousContent) {
|
| .WillRepeatedly(Return(kModerateContentDisposition));
|
| DownloadTargetInfo target_info;
|
| DetermineDownloadTarget(download_item.get(), &target_info);
|
| - EXPECT_EQ(download_util::ALLOW_ON_USER_GESTURE,
|
| + EXPECT_EQ(DownloadFileType::ALLOW_ON_USER_GESTURE,
|
| DownloadItemModel(download_item.get()).GetDangerLevel());
|
| EXPECT_EQ(content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| target_info.danger_type);
|
| @@ -481,7 +482,7 @@ namespace {
|
|
|
| struct SafeBrowsingTestParameters {
|
| content::DownloadDangerType initial_danger_type;
|
| - download_util::DownloadDangerLevel initial_danger_level;
|
| + DownloadFileType::DangerLevel initial_danger_level;
|
| safe_browsing::DownloadProtectionService::DownloadCheckResult verdict;
|
|
|
| content::DownloadDangerType expected_danger_type;
|
| @@ -530,100 +531,106 @@ void ChromeDownloadManagerDelegateTestWithSafeBrowsing::TearDown() {
|
|
|
| const SafeBrowsingTestParameters kSafeBrowsingTestCases[] = {
|
| // SAFE verdict for a safe file.
|
| - {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, download_util::NOT_DANGEROUS,
|
| + {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
|
| + DownloadFileType::NOT_DANGEROUS,
|
| safe_browsing::DownloadProtectionService::SAFE,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS},
|
|
|
| // UNKNOWN verdict for a safe file.
|
| - {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, download_util::NOT_DANGEROUS,
|
| + {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
|
| + DownloadFileType::NOT_DANGEROUS,
|
| safe_browsing::DownloadProtectionService::UNKNOWN,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS},
|
|
|
| // DANGEROUS verdict for a safe file.
|
| - {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, download_util::NOT_DANGEROUS,
|
| + {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
|
| + DownloadFileType::NOT_DANGEROUS,
|
| safe_browsing::DownloadProtectionService::DANGEROUS,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT},
|
|
|
| // UNCOMMON verdict for a safe file.
|
| - {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, download_util::NOT_DANGEROUS,
|
| + {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
|
| + DownloadFileType::NOT_DANGEROUS,
|
| safe_browsing::DownloadProtectionService::UNCOMMON,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT},
|
|
|
| // POTENTIALLY_UNWANTED verdict for a safe file.
|
| - {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, download_util::NOT_DANGEROUS,
|
| + {content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
|
| + DownloadFileType::NOT_DANGEROUS,
|
| safe_browsing::DownloadProtectionService::POTENTIALLY_UNWANTED,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED},
|
|
|
| // SAFE verdict for a potentially dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::ALLOW_ON_USER_GESTURE,
|
| + DownloadFileType::ALLOW_ON_USER_GESTURE,
|
| safe_browsing::DownloadProtectionService::SAFE,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS},
|
|
|
| // UNKNOWN verdict for a potentially dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::ALLOW_ON_USER_GESTURE,
|
| + DownloadFileType::ALLOW_ON_USER_GESTURE,
|
| safe_browsing::DownloadProtectionService::UNKNOWN,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE},
|
|
|
| // DANGEROUS verdict for a potentially dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::ALLOW_ON_USER_GESTURE,
|
| + DownloadFileType::ALLOW_ON_USER_GESTURE,
|
| safe_browsing::DownloadProtectionService::DANGEROUS,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT},
|
|
|
| // UNCOMMON verdict for a potentially dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::ALLOW_ON_USER_GESTURE,
|
| + DownloadFileType::ALLOW_ON_USER_GESTURE,
|
| safe_browsing::DownloadProtectionService::UNCOMMON,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT},
|
|
|
| // POTENTIALLY_UNWANTED verdict for a potentially dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::ALLOW_ON_USER_GESTURE,
|
| + DownloadFileType::ALLOW_ON_USER_GESTURE,
|
| safe_browsing::DownloadProtectionService::POTENTIALLY_UNWANTED,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED},
|
|
|
| // SAFE verdict for a dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::DANGEROUS, safe_browsing::DownloadProtectionService::SAFE,
|
| + DownloadFileType::DANGEROUS,
|
| + safe_browsing::DownloadProtectionService::SAFE,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE},
|
|
|
| // UNKNOWN verdict for a dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::DANGEROUS,
|
| + DownloadFileType::DANGEROUS,
|
| safe_browsing::DownloadProtectionService::UNKNOWN,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE},
|
|
|
| // DANGEROUS verdict for a dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::DANGEROUS,
|
| + DownloadFileType::DANGEROUS,
|
| safe_browsing::DownloadProtectionService::DANGEROUS,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT},
|
|
|
| // UNCOMMON verdict for a dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::DANGEROUS,
|
| + DownloadFileType::DANGEROUS,
|
| safe_browsing::DownloadProtectionService::UNCOMMON,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT},
|
|
|
| // POTENTIALLY_UNWANTED verdict for a dangerous file.
|
| {content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT,
|
| - download_util::DANGEROUS,
|
| + DownloadFileType::DANGEROUS,
|
| safe_browsing::DownloadProtectionService::POTENTIALLY_UNWANTED,
|
|
|
| content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED},
|
| @@ -646,7 +653,7 @@ TEST_P(ChromeDownloadManagerDelegateTestWithSafeBrowsing, CheckClientDownload) {
|
| EXPECT_CALL(*download_item, GetDangerType())
|
| .WillRepeatedly(Return(kParameters.initial_danger_type));
|
|
|
| - if (kParameters.initial_danger_level != download_util::NOT_DANGEROUS) {
|
| + if (kParameters.initial_danger_level != DownloadFileType::NOT_DANGEROUS) {
|
| DownloadItemModel(download_item.get())
|
| .SetDangerLevel(kParameters.initial_danger_level);
|
| }
|
|
|