summaryrefslogtreecommitdiffstats
path: root/content/browser/download/download_manager_impl_unittest.cc
diff options
context:
space:
mode:
authorasanka@chromium.org <asanka@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2013-05-29 18:57:45 +0000
committerasanka@chromium.org <asanka@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2013-05-29 18:57:45 +0000
commitfe752272420b287bd97512d86668bae1371f0419 (patch)
treefab7814bba055c5592d0d8861a7c30fd88a3f9b4 /content/browser/download/download_manager_impl_unittest.cc
parentee54df992ca2a943a5f0b51daef56ae4e30426f5 (diff)
downloadchromium_src-fe752272420b287bd97512d86668bae1371f0419.zip
chromium_src-fe752272420b287bd97512d86668bae1371f0419.tar.gz
chromium_src-fe752272420b287bd97512d86668bae1371f0419.tar.bz2
[Downloads] Allow acquiring dangerous download file.
A dangerous download can be accepted by the user or rejected. If rejected, the downloaded file used to be deleted. This change adds DownloadItem::AcquireFileAndDeleteDownload() which allows the caller to acquire the dangerous file. The intended consumer of this feature is SafeBrowsing where the caller may want to acquire a dangerous file that's being discarded for the purpose of further analysis. Also change the logic during shutdown to no longer delete dangerous downloads, but to cancel them. TBR=rdsmith BUG=244604 Review URL: https://chromiumcodereview.appspot.com/14947007 git-svn-id: svn://svn.chromium.org/chrome/trunk/src@202925 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'content/browser/download/download_manager_impl_unittest.cc')
-rw-r--r--content/browser/download/download_manager_impl_unittest.cc10
1 files changed, 4 insertions, 6 deletions
diff --git a/content/browser/download/download_manager_impl_unittest.cc b/content/browser/download/download_manager_impl_unittest.cc
index 300eb2e..a6dd330 100644
--- a/content/browser/download/download_manager_impl_unittest.cc
+++ b/content/browser/download/download_manager_impl_unittest.cc
@@ -93,7 +93,8 @@ class MockDownloadItemImpl : public DownloadItemImpl {
MOCK_METHOD0(ShouldOpenFileBasedOnExtension, bool());
MOCK_METHOD0(OpenDownload, void());
MOCK_METHOD0(ShowDownloadInShell, void());
- MOCK_METHOD0(DangerousDownloadValidated, void());
+ MOCK_METHOD0(ValidateDangerousDownload, void());
+ MOCK_METHOD1(StealDangerousDownload, void(const AcquireFileCallback&));
MOCK_METHOD3(UpdateProgress, void(int64, int64, const std::string&));
MOCK_METHOD1(Cancel, void(bool));
MOCK_METHOD0(MarkAsComplete, void());
@@ -107,7 +108,6 @@ class MockDownloadItemImpl : public DownloadItemImpl {
MOCK_METHOD2(MockStart, void(DownloadFile*, DownloadRequestHandleInterface*));
- MOCK_METHOD1(Delete, void(DeleteReason));
MOCK_METHOD0(Remove, void());
MOCK_CONST_METHOD1(TimeRemaining, bool(base::TimeDelta*));
MOCK_CONST_METHOD0(CurrentSpeed, int64());
@@ -472,10 +472,8 @@ class DownloadManagerTest : public testing::Test {
virtual void TearDown() {
while (MockDownloadItemImpl*
item = mock_download_item_factory_->PopItem()) {
- EXPECT_CALL(*item, IsDangerous())
- .WillOnce(Return(false));
- EXPECT_CALL(*item, IsPartialDownload())
- .WillOnce(Return(false));
+ EXPECT_CALL(*item, GetState())
+ .WillOnce(Return(DownloadItem::CANCELLED));
}
EXPECT_CALL(GetMockObserver(), ManagerGoingDown(download_manager_.get()))
.WillOnce(Return());