| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CONTENT_PUBLIC_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ | 5 #ifndef CONTENT_PUBLIC_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ |
| 6 #define CONTENT_PUBLIC_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ | 6 #define CONTENT_PUBLIC_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ |
| 7 | 7 |
| 8 #include "content/public/browser/download_manager.h" | 8 #include "content/public/browser/download_manager.h" |
| 9 #include "content/public/browser/download_save_info.h" | 9 #include "content/public/browser/download_save_info.h" |
| 10 #include "content/public/browser/download_url_parameters.h" | 10 #include "content/public/browser/download_url_parameters.h" |
| 11 #include "googleurl/src/gurl.h" | 11 #include "googleurl/src/gurl.h" |
| 12 #include "testing/gmock/include/gmock/gmock.h" | 12 #include "testing/gmock/include/gmock/gmock.h" |
| 13 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 14 | 14 |
| 15 // To avoid leaking download_request_handle.h to embedders. | 15 // To avoid leaking download_request_handle.h to embedders. |
| 16 void PrintTo(const DownloadRequestHandle& params, std::ostream* os); | 16 void PrintTo(const DownloadRequestHandle& params, std::ostream* os); |
| 17 | 17 |
| 18 namespace content { | 18 namespace content { |
| 19 | 19 |
| 20 class MockDownloadManager : public content::DownloadManager { | 20 class MockDownloadManager : public DownloadManager { |
| 21 public: | 21 public: |
| 22 MockDownloadManager(); | 22 MockDownloadManager(); |
| 23 | 23 |
| 24 // DownloadManager: | 24 // DownloadManager: |
| 25 MOCK_METHOD1(SetDelegate, void(DownloadManagerDelegate* delegate)); | 25 MOCK_METHOD1(SetDelegate, void(DownloadManagerDelegate* delegate)); |
| 26 MOCK_CONST_METHOD0(GetDelegate, content::DownloadManagerDelegate*()); | 26 MOCK_CONST_METHOD0(GetDelegate, DownloadManagerDelegate*()); |
| 27 MOCK_METHOD0(Shutdown, void()); | 27 MOCK_METHOD0(Shutdown, void()); |
| 28 MOCK_METHOD2(GetTemporaryDownloads, void(const FilePath& dir_path, | 28 MOCK_METHOD2(GetTemporaryDownloads, void(const FilePath& dir_path, |
| 29 DownloadVector* result)); | 29 DownloadVector* result)); |
| 30 MOCK_METHOD2(GetAllDownloads, void(const FilePath& dir_path, | 30 MOCK_METHOD2(GetAllDownloads, void(const FilePath& dir_path, |
| 31 DownloadVector* result)); | 31 DownloadVector* result)); |
| 32 MOCK_METHOD2(SearchDownloads, void(const string16& query, | 32 MOCK_METHOD2(SearchDownloads, void(const string16& query, |
| 33 DownloadVector* result)); | 33 DownloadVector* result)); |
| 34 MOCK_METHOD1(Init, bool(content::BrowserContext* browser_context)); | 34 MOCK_METHOD1(Init, bool(BrowserContext* browser_context)); |
| 35 | 35 |
| 36 // Gasket for handling scoped_ptr arguments. | 36 // Gasket for handling scoped_ptr arguments. |
| 37 virtual content::DownloadId StartDownload( | 37 virtual DownloadId StartDownload( |
| 38 scoped_ptr<DownloadCreateInfo> info, | 38 scoped_ptr<DownloadCreateInfo> info, |
| 39 scoped_ptr<content::ByteStreamReader> stream) OVERRIDE; | 39 scoped_ptr<ByteStreamReader> stream) OVERRIDE; |
| 40 | 40 |
| 41 MOCK_METHOD2(MockStartDownload, | 41 MOCK_METHOD2(MockStartDownload, |
| 42 content::DownloadId(DownloadCreateInfo*, | 42 DownloadId(DownloadCreateInfo*, |
| 43 content::ByteStreamReader*)); | 43 ByteStreamReader*)); |
| 44 MOCK_METHOD4(UpdateDownload, void(int32 download_id, | 44 MOCK_METHOD4(UpdateDownload, void(int32 download_id, |
| 45 int64 bytes_so_far, | 45 int64 bytes_so_far, |
| 46 int64 bytes_per_sec, | 46 int64 bytes_per_sec, |
| 47 const std::string& hash_state)); | 47 const std::string& hash_state)); |
| 48 MOCK_METHOD3(OnResponseCompleted, void(int32 download_id, | 48 MOCK_METHOD3(OnResponseCompleted, void(int32 download_id, |
| 49 int64 size, | 49 int64 size, |
| 50 const std::string& hash)); | 50 const std::string& hash)); |
| 51 MOCK_METHOD1(CancelDownload, void(int32 download_id)); | 51 MOCK_METHOD1(CancelDownload, void(int32 download_id)); |
| 52 MOCK_METHOD2(OnDownloadInterrupted, | 52 MOCK_METHOD2(OnDownloadInterrupted, |
| 53 void(int32 download_id, | 53 void(int32 download_id, |
| 54 content::DownloadInterruptReason reason)); | 54 DownloadInterruptReason reason)); |
| 55 MOCK_METHOD2(RemoveDownloadsBetween, int(base::Time remove_begin, | 55 MOCK_METHOD2(RemoveDownloadsBetween, int(base::Time remove_begin, |
| 56 base::Time remove_end)); | 56 base::Time remove_end)); |
| 57 MOCK_METHOD1(RemoveDownloads, int(base::Time remove_begin)); | 57 MOCK_METHOD1(RemoveDownloads, int(base::Time remove_begin)); |
| 58 MOCK_METHOD0(RemoveAllDownloads, int()); | 58 MOCK_METHOD0(RemoveAllDownloads, int()); |
| 59 MOCK_METHOD1(DownloadUrlMock, void(DownloadUrlParameters*)); | 59 MOCK_METHOD1(DownloadUrlMock, void(DownloadUrlParameters*)); |
| 60 virtual void DownloadUrl(scoped_ptr<DownloadUrlParameters> params) OVERRIDE { | 60 virtual void DownloadUrl(scoped_ptr<DownloadUrlParameters> params) OVERRIDE { |
| 61 DownloadUrlMock(params.get()); | 61 DownloadUrlMock(params.get()); |
| 62 } | 62 } |
| 63 MOCK_METHOD1(AddObserver, void(Observer* observer)); | 63 MOCK_METHOD1(AddObserver, void(Observer* observer)); |
| 64 MOCK_METHOD1(RemoveObserver, void(Observer* observer)); | 64 MOCK_METHOD1(RemoveObserver, void(Observer* observer)); |
| 65 MOCK_METHOD1(OnPersistentStoreQueryComplete, void( | 65 MOCK_METHOD1(OnPersistentStoreQueryComplete, void( |
| 66 std::vector<DownloadPersistentStoreInfo>* entries)); | 66 std::vector<DownloadPersistentStoreInfo>* entries)); |
| 67 MOCK_METHOD2(OnItemAddedToPersistentStore, void(int32 download_id, | 67 MOCK_METHOD2(OnItemAddedToPersistentStore, void(int32 download_id, |
| 68 int64 db_handle)); | 68 int64 db_handle)); |
| 69 MOCK_CONST_METHOD0(InProgressCount, int()); | 69 MOCK_CONST_METHOD0(InProgressCount, int()); |
| 70 MOCK_CONST_METHOD0(GetBrowserContext, content::BrowserContext*()); | 70 MOCK_CONST_METHOD0(GetBrowserContext, BrowserContext*()); |
| 71 MOCK_METHOD0(LastDownloadPath, FilePath()); | |
| 72 MOCK_METHOD0(ClearLastDownloadPath, void()); | |
| 73 MOCK_METHOD2(FileSelected, void(const FilePath& path, int32 download_id)); | |
| 74 MOCK_METHOD1(FileSelectionCanceled, void(int32 download_id)); | |
| 75 MOCK_METHOD1(RestartDownload, void(int32 download_id)); | |
| 76 MOCK_METHOD0(CheckForHistoryFilesRemoval, void()); | 71 MOCK_METHOD0(CheckForHistoryFilesRemoval, void()); |
| 77 MOCK_METHOD1(GetDownloadItem, content::DownloadItem*(int id)); | 72 MOCK_METHOD1(GetDownloadItem, DownloadItem*(int id)); |
| 78 MOCK_METHOD1(GetDownload, content::DownloadItem*(int id)); | 73 MOCK_METHOD1(GetDownload, DownloadItem*(int id)); |
| 79 MOCK_METHOD1(SavePageDownloadFinished, void(content::DownloadItem* download)); | 74 MOCK_METHOD1(SavePageDownloadFinished, void(DownloadItem* download)); |
| 80 MOCK_METHOD1(GetActiveDownloadItem, content::DownloadItem*(int id)); | 75 MOCK_METHOD1(GetActiveDownloadItem, DownloadItem*(int id)); |
| 81 MOCK_METHOD0(GenerateFileHash, bool()); | 76 MOCK_METHOD0(GenerateFileHash, bool()); |
| 82 MOCK_METHOD2(ContinueDownloadWithPath, void(content::DownloadItem* download, | 77 MOCK_METHOD1(GetActiveDownload, DownloadItem*(int32 download_id)); |
| 83 const FilePath& chosen_file)); | |
| 84 MOCK_METHOD1(GetActiveDownload, content::DownloadItem*(int32 download_id)); | |
| 85 MOCK_METHOD1(SetFileManager, void(DownloadFileManager* file_manager)); | |
| 86 | 78 |
| 87 protected: | 79 protected: |
| 88 virtual ~MockDownloadManager(); | 80 virtual ~MockDownloadManager(); |
| 89 }; | 81 }; |
| 90 | 82 |
| 91 } // namespace content | 83 } // namespace content |
| 92 | 84 |
| 93 #endif // CONTENT_PUBLIC_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ | 85 #endif // CONTENT_PUBLIC_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ |
| OLD | NEW |