diff options
author | asanka@chromium.org <asanka@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2013-05-29 18:57:45 +0000 |
---|---|---|
committer | asanka@chromium.org <asanka@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2013-05-29 18:57:45 +0000 |
commit | fe752272420b287bd97512d86668bae1371f0419 (patch) | |
tree | fab7814bba055c5592d0d8861a7c30fd88a3f9b4 /content/browser/download/download_manager_impl_unittest.cc | |
parent | ee54df992ca2a943a5f0b51daef56ae4e30426f5 (diff) | |
download | chromium_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.cc | 10 |
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()); |