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 |