| Index: content/browser/download/download_item_impl_unittest.cc
|
| diff --git a/content/browser/download/download_item_impl_unittest.cc b/content/browser/download/download_item_impl_unittest.cc
|
| index 3f223fc12e2d3167d7102ff18755325a65914fb3..abe5c57165d581a26a99a62285b77c64b4f8ed57 100644
|
| --- a/content/browser/download/download_item_impl_unittest.cc
|
| +++ b/content/browser/download/download_item_impl_unittest.cc
|
| @@ -5,6 +5,7 @@
|
| #include "base/message_loop.h"
|
| #include "base/stl_util.h"
|
| #include "base/threading/thread.h"
|
| +#include "content/browser/download/byte_stream.h"
|
| #include "content/browser/download/download_create_info.h"
|
| #include "content/browser/download/download_file_manager.h"
|
| #include "content/browser/download/download_item_impl.h"
|
| @@ -58,8 +59,21 @@ class MockRequestHandle : public DownloadRequestHandleInterface {
|
| class MockDownloadFileFactory
|
| : public DownloadFileManager::DownloadFileFactory {
|
| public:
|
| - MOCK_METHOD5(CreateFile,
|
| + content::DownloadFile* CreateFile(
|
| + DownloadCreateInfo* info,
|
| + scoped_ptr<content::ByteStreamReader> stream_reader,
|
| + const DownloadRequestHandle& request_handle,
|
| + DownloadManager* mgr,
|
| + bool calculate_hash,
|
| + const net::BoundNetLog& bound_net_log) {
|
| + return MockCreateFile(
|
| + info, stream_reader.get(), request_handle, mgr, calculate_hash,
|
| + bound_net_log);
|
| + }
|
| +
|
| + MOCK_METHOD6(MockCreateFile,
|
| content::DownloadFile*(DownloadCreateInfo*,
|
| + content::ByteStreamReader*,
|
| const DownloadRequestHandle&,
|
| DownloadManager*,
|
| bool,
|
| @@ -70,12 +84,15 @@ class MockDownloadFileManager : public DownloadFileManager {
|
| public:
|
| MockDownloadFileManager();
|
| MOCK_METHOD0(Shutdown, void());
|
| - MOCK_METHOD2(StartDownload,
|
| - void(DownloadCreateInfo*, const DownloadRequestHandle&));
|
| - MOCK_METHOD2(UpdateDownload, void(DownloadId, content::DownloadBuffer*));
|
| - MOCK_METHOD3(OnResponseCompleted,
|
| - void(DownloadId, content::DownloadInterruptReason,
|
| - const std::string&));
|
| + MOCK_METHOD3(MockStartDownload,
|
| + void(DownloadCreateInfo*, content::ByteStreamReader*,
|
| + const DownloadRequestHandle&));
|
| + virtual void StartDownload(scoped_ptr<DownloadCreateInfo> info,
|
| + scoped_ptr<content::ByteStreamReader> stream,
|
| + const DownloadRequestHandle& request_handle) {
|
| + MockStartDownload(info.release(), stream.release(), request_handle);
|
| + }
|
| +
|
| MOCK_METHOD1(CancelDownload, void(DownloadId));
|
| MOCK_METHOD1(CompleteDownload, void(DownloadId));
|
| MOCK_METHOD1(OnDownloadManagerShutdown, void(DownloadManager*));
|
|
|