OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ | 5 #ifndef CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ |
6 #define CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ | 6 #define CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
| 9 #include "content/browser/download/download_item_impl.h" |
9 #include "content/browser/download/download_manager.h" | 10 #include "content/browser/download/download_manager.h" |
10 #include "content/browser/download/download_id.h" | 11 #include "content/browser/download/download_id.h" |
11 #include "content/browser/download/download_id_factory.h" | 12 #include "content/browser/download/download_id_factory.h" |
12 | 13 |
13 class DownloadStatusUpdater; | 14 class DownloadStatusUpdater; |
14 class DownloadItem; | 15 class DownloadItem; |
15 | 16 |
16 class MockDownloadManager : public DownloadManager { | 17 class MockDownloadManager |
| 18 : public DownloadManager { |
17 public: | 19 public: |
18 explicit MockDownloadManager(content::DownloadManagerDelegate* delegate, | 20 explicit MockDownloadManager(content::DownloadManagerDelegate* delegate, |
19 DownloadIdFactory* id_factory, | 21 DownloadIdFactory* id_factory, |
20 DownloadStatusUpdater* updater); | 22 DownloadStatusUpdater* updater); |
21 virtual ~MockDownloadManager(); | 23 virtual ~MockDownloadManager(); |
22 | 24 |
23 // DownloadManager: | 25 // DownloadManager: |
24 virtual void Shutdown() OVERRIDE; | 26 virtual void Shutdown() OVERRIDE; |
25 virtual void GetTemporaryDownloads(const FilePath& dir_path, | 27 virtual void GetTemporaryDownloads(const FilePath& dir_path, |
26 DownloadVector* result) OVERRIDE; | 28 DownloadVector* result) OVERRIDE; |
27 virtual void GetAllDownloads(const FilePath& dir_path, | 29 virtual void GetAllDownloads(const FilePath& dir_path, |
28 DownloadVector* result) OVERRIDE; | 30 DownloadVector* result) OVERRIDE; |
29 virtual void SearchDownloads(const string16& query, | 31 virtual void SearchDownloads(const string16& query, |
30 DownloadVector* result) OVERRIDE; | 32 DownloadVector* result) OVERRIDE; |
31 virtual bool Init(content::BrowserContext* browser_context) OVERRIDE; | 33 virtual bool Init(content::BrowserContext* browser_context) OVERRIDE; |
32 virtual void StartDownload(int32 id) OVERRIDE; | 34 virtual void StartDownload(int32 id) OVERRIDE; |
33 virtual void UpdateDownload(int32 download_id, int64 size) OVERRIDE; | 35 virtual void UpdateDownload(int32 download_id, int64 size) OVERRIDE; |
34 virtual void OnResponseCompleted(int32 download_id, int64 size, | 36 virtual void OnResponseCompleted(int32 download_id, int64 size, |
35 const std::string& hash) OVERRIDE; | 37 const std::string& hash) OVERRIDE; |
36 virtual void CancelDownload(int32 download_id) OVERRIDE; | 38 virtual void CancelDownload(int32 download_id) OVERRIDE; |
37 virtual void OnDownloadInterrupted(int32 download_id, int64 size, | 39 virtual void OnDownloadInterrupted(int32 download_id, int64 size, |
38 InterruptReason reason) OVERRIDE; | 40 InterruptReason reason) OVERRIDE; |
39 virtual void DownloadCancelledInternal(DownloadItem* download) OVERRIDE; | |
40 virtual void RemoveDownload(int64 download_handle) OVERRIDE; | |
41 virtual bool IsDownloadReadyForCompletion(DownloadItem* download) OVERRIDE; | |
42 virtual void MaybeCompleteDownload(DownloadItem* download) OVERRIDE; | 41 virtual void MaybeCompleteDownload(DownloadItem* download) OVERRIDE; |
43 virtual void OnDownloadRenamedToFinalName(int download_id, | 42 virtual void OnDownloadRenamedToFinalName(int download_id, |
44 const FilePath& full_path, | 43 const FilePath& full_path, |
45 int uniquifier) OVERRIDE; | 44 int uniquifier) OVERRIDE; |
46 virtual int RemoveDownloadsBetween(const base::Time remove_begin, | 45 virtual int RemoveDownloadsBetween(const base::Time remove_begin, |
47 const base::Time remove_end) OVERRIDE; | 46 const base::Time remove_end) OVERRIDE; |
48 virtual int RemoveDownloads(const base::Time remove_begin) OVERRIDE; | 47 virtual int RemoveDownloads(const base::Time remove_begin) OVERRIDE; |
49 virtual int RemoveAllDownloads() OVERRIDE; | 48 virtual int RemoveAllDownloads() OVERRIDE; |
50 virtual void DownloadCompleted(int32 download_id) OVERRIDE; | |
51 virtual void DownloadUrl(const GURL& url, | 49 virtual void DownloadUrl(const GURL& url, |
52 const GURL& referrer, | 50 const GURL& referrer, |
53 const std::string& referrer_encoding, | 51 const std::string& referrer_encoding, |
54 TabContents* tab_contents) OVERRIDE; | 52 TabContents* tab_contents) OVERRIDE; |
55 virtual void DownloadUrlToFile(const GURL& url, | 53 virtual void DownloadUrlToFile(const GURL& url, |
56 const GURL& referrer, | 54 const GURL& referrer, |
57 const std::string& referrer_encoding, | 55 const std::string& referrer_encoding, |
58 const DownloadSaveInfo& save_info, | 56 const DownloadSaveInfo& save_info, |
59 TabContents* tab_contents) OVERRIDE; | 57 TabContents* tab_contents) OVERRIDE; |
60 virtual void AddObserver(Observer* observer) OVERRIDE; | 58 virtual void AddObserver(Observer* observer) OVERRIDE; |
61 virtual void RemoveObserver(Observer* observer) OVERRIDE; | 59 virtual void RemoveObserver(Observer* observer) OVERRIDE; |
62 virtual void OnPersistentStoreQueryComplete( | 60 virtual void OnPersistentStoreQueryComplete( |
63 std::vector<DownloadPersistentStoreInfo>* entries) OVERRIDE; | 61 std::vector<DownloadPersistentStoreInfo>* entries) OVERRIDE; |
64 virtual void OnItemAddedToPersistentStore(int32 download_id, | 62 virtual void OnItemAddedToPersistentStore(int32 download_id, |
65 int64 db_handle) OVERRIDE; | 63 int64 db_handle) OVERRIDE; |
66 virtual void ShowDownloadInBrowser(DownloadItem* download) OVERRIDE; | |
67 virtual int InProgressCount() const OVERRIDE; | 64 virtual int InProgressCount() const OVERRIDE; |
68 virtual content::BrowserContext* BrowserContext() OVERRIDE; | 65 virtual content::BrowserContext* BrowserContext() const OVERRIDE; |
69 virtual FilePath LastDownloadPath() OVERRIDE; | 66 virtual FilePath LastDownloadPath() OVERRIDE; |
70 virtual void CreateDownloadItem( | 67 virtual void CreateDownloadItem( |
71 DownloadCreateInfo* info, | 68 DownloadCreateInfo* info, |
72 const DownloadRequestHandle& request_handle) OVERRIDE; | 69 const DownloadRequestHandle& request_handle) OVERRIDE; |
| 70 virtual DownloadItem* CreateSavePackageDownloadItem( |
| 71 const FilePath& main_file_path, |
| 72 const GURL& page_url, |
| 73 bool is_otr, |
| 74 DownloadItem::Observer* observer) OVERRIDE; |
73 virtual void ClearLastDownloadPath() OVERRIDE; | 75 virtual void ClearLastDownloadPath() OVERRIDE; |
74 virtual void FileSelected(const FilePath& path, void* params) OVERRIDE; | 76 virtual void FileSelected(const FilePath& path, void* params) OVERRIDE; |
75 virtual void FileSelectionCanceled(void* params) OVERRIDE; | 77 virtual void FileSelectionCanceled(void* params) OVERRIDE; |
76 virtual void RestartDownload(int32 download_id) OVERRIDE; | 78 virtual void RestartDownload(int32 download_id) OVERRIDE; |
77 virtual void MarkDownloadOpened(DownloadItem* download) OVERRIDE; | |
78 virtual void CheckForHistoryFilesRemoval() OVERRIDE; | 79 virtual void CheckForHistoryFilesRemoval() OVERRIDE; |
79 virtual void CheckForFileRemoval(DownloadItem* download_item) OVERRIDE; | |
80 virtual void AssertQueueStateConsistent(DownloadItem* download) OVERRIDE; | |
81 virtual DownloadItem* GetDownloadItem(int id) OVERRIDE; | 80 virtual DownloadItem* GetDownloadItem(int id) OVERRIDE; |
82 virtual void SavePageDownloadStarted(DownloadItem* download) OVERRIDE; | |
83 virtual void SavePageDownloadFinished(DownloadItem* download) OVERRIDE; | 81 virtual void SavePageDownloadFinished(DownloadItem* download) OVERRIDE; |
84 virtual DownloadItem* GetActiveDownloadItem(int id) OVERRIDE; | 82 virtual DownloadItem* GetActiveDownloadItem(int id) OVERRIDE; |
85 virtual content::DownloadManagerDelegate* delegate() const OVERRIDE; | 83 virtual content::DownloadManagerDelegate* delegate() const OVERRIDE; |
86 virtual void SetDownloadManagerDelegate( | 84 virtual void SetDownloadManagerDelegate( |
87 content::DownloadManagerDelegate* delegate) OVERRIDE; | 85 content::DownloadManagerDelegate* delegate) OVERRIDE; |
88 virtual DownloadId GetNextId() OVERRIDE; | |
89 virtual void ContinueDownloadWithPath(DownloadItem* download, | 86 virtual void ContinueDownloadWithPath(DownloadItem* download, |
90 const FilePath& chosen_file) OVERRIDE; | 87 const FilePath& chosen_file) OVERRIDE; |
91 virtual DownloadItem* GetActiveDownload(int32 download_id) OVERRIDE; | 88 virtual DownloadItem* GetActiveDownload(int32 download_id) OVERRIDE; |
92 virtual void SetFileManager(DownloadFileManager* file_manager) OVERRIDE; | 89 virtual void SetFileManager(DownloadFileManager* file_manager) OVERRIDE; |
93 | 90 |
94 private: | 91 private: |
95 content::DownloadManagerDelegate* delegate_; | 92 content::DownloadManagerDelegate* delegate_; |
96 DownloadIdFactory* id_factory_; | 93 DownloadIdFactory* id_factory_; |
97 DownloadStatusUpdater* updater_; | 94 DownloadStatusUpdater* updater_; |
98 DownloadFileManager* file_manager_; | 95 DownloadFileManager* file_manager_; |
99 std::map<int32, DownloadItem*> item_map_; | 96 std::map<int32, DownloadItem*> item_map_; |
100 std::map<int32, DownloadItem*> inactive_item_map_; | 97 std::map<int32, DownloadItem*> inactive_item_map_; |
101 }; | 98 }; |
102 | 99 |
103 #endif // CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ | 100 #endif // CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ |
OLD | NEW |