Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(5050)

Unified Diff: chrome/browser/download/chrome_download_manager_delegate_unittest.cc

Issue 1982723002: Use FileTypePolicies for download danger classifications. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@use_policies
Patch Set: Fix bad rebase Created 4 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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);
}
« no previous file with comments | « chrome/browser/download/chrome_download_manager_delegate.cc ('k') | chrome/browser/download/download_browsertest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698