Index: content/public/test/mock_download_manager.h |
diff --git a/content/public/test/mock_download_manager.h b/content/public/test/mock_download_manager.h |
index 9f090a92bc3b6493a4622f50fa5e887d9e32a079..adbdc2fa87e23953c1915d7348705a5f50bce2ed 100644 |
--- a/content/public/test/mock_download_manager.h |
+++ b/content/public/test/mock_download_manager.h |
@@ -7,6 +7,7 @@ |
#include <vector> |
+#include "content/public/browser/download_interrupt_reasons.h" |
#include "content/public/browser/download_manager.h" |
#include "content/public/browser/download_save_info.h" |
#include "content/public/browser/download_url_parameters.h" |
@@ -18,6 +19,9 @@ class DownloadRequestHandle; |
namespace content { |
+template <typename StatusType> |
+class ByteStreamReader; |
+ |
// To avoid leaking download_request_handle.h to embedders. |
void PrintTo(const DownloadRequestHandle& params, std::ostream* os); |
@@ -74,11 +78,12 @@ class MockDownloadManager : public DownloadManager { |
// Gasket for handling scoped_ptr arguments. |
virtual void StartDownload( |
scoped_ptr<DownloadCreateInfo> info, |
- scoped_ptr<ByteStreamReader> stream, |
+ scoped_ptr<ByteStreamReader<DownloadInterruptReason> > stream, |
const DownloadUrlParameters::OnStartedCallback& callback) OVERRIDE; |
MOCK_METHOD2(MockStartDownload, |
- void(DownloadCreateInfo*, ByteStreamReader*)); |
+ void(DownloadCreateInfo*, |
+ ByteStreamReader<DownloadInterruptReason>*)); |
MOCK_METHOD2(RemoveDownloadsBetween, int(base::Time remove_begin, |
base::Time remove_end)); |
MOCK_METHOD1(RemoveDownloads, int(base::Time remove_begin)); |