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 bytes_so_far, | 35 virtual void UpdateDownload(int32 download_id, int64 bytes_so_far, |
34 int64 bytes_per_sec) OVERRIDE; | 36 int64 bytes_per_sec) OVERRIDE; |
35 virtual void OnResponseCompleted(int32 download_id, int64 size, | 37 virtual void OnResponseCompleted(int32 download_id, int64 size, |
36 const std::string& hash) OVERRIDE; | 38 const std::string& hash) OVERRIDE; |
37 virtual void CancelDownload(int32 download_id) OVERRIDE; | 39 virtual void CancelDownload(int32 download_id) OVERRIDE; |
38 virtual void OnDownloadInterrupted(int32 download_id, int64 size, | 40 virtual void OnDownloadInterrupted(int32 download_id, int64 size, |
39 InterruptReason reason) OVERRIDE; | 41 InterruptReason reason) OVERRIDE; |
40 virtual void DownloadCancelledInternal(DownloadItem* download) OVERRIDE; | |
41 virtual void RemoveDownload(int64 download_handle) OVERRIDE; | |
42 virtual bool IsDownloadReadyForCompletion(DownloadItem* download) OVERRIDE; | |
43 virtual void MaybeCompleteDownload(DownloadItem* download) OVERRIDE; | |
44 virtual void OnDownloadRenamedToFinalName(int download_id, | 42 virtual void OnDownloadRenamedToFinalName(int download_id, |
45 const FilePath& full_path, | 43 const FilePath& full_path, |
46 int uniquifier) OVERRIDE; | 44 int uniquifier) OVERRIDE; |
47 virtual int RemoveDownloadsBetween(const base::Time remove_begin, | 45 virtual int RemoveDownloadsBetween(const base::Time remove_begin, |
48 const base::Time remove_end) OVERRIDE; | 46 const base::Time remove_end) OVERRIDE; |
49 virtual int RemoveDownloads(const base::Time remove_begin) OVERRIDE; | 47 virtual int RemoveDownloads(const base::Time remove_begin) OVERRIDE; |
50 virtual int RemoveAllDownloads() OVERRIDE; | 48 virtual int RemoveAllDownloads() OVERRIDE; |
51 virtual void DownloadCompleted(int32 download_id) OVERRIDE; | |
52 virtual void DownloadUrl(const GURL& url, | 49 virtual void DownloadUrl(const GURL& url, |
53 const GURL& referrer, | 50 const GURL& referrer, |
54 const std::string& referrer_encoding, | 51 const std::string& referrer_encoding, |
55 TabContents* tab_contents) OVERRIDE; | 52 TabContents* tab_contents) OVERRIDE; |
56 virtual void DownloadUrlToFile(const GURL& url, | 53 virtual void DownloadUrlToFile(const GURL& url, |
57 const GURL& referrer, | 54 const GURL& referrer, |
58 const std::string& referrer_encoding, | 55 const std::string& referrer_encoding, |
59 const DownloadSaveInfo& save_info, | 56 const DownloadSaveInfo& save_info, |
60 TabContents* tab_contents) OVERRIDE; | 57 TabContents* tab_contents) OVERRIDE; |
61 virtual void AddObserver(Observer* observer) OVERRIDE; | 58 virtual void AddObserver(Observer* observer) OVERRIDE; |
62 virtual void RemoveObserver(Observer* observer) OVERRIDE; | 59 virtual void RemoveObserver(Observer* observer) OVERRIDE; |
63 virtual void OnPersistentStoreQueryComplete( | 60 virtual void OnPersistentStoreQueryComplete( |
64 std::vector<DownloadPersistentStoreInfo>* entries) OVERRIDE; | 61 std::vector<DownloadPersistentStoreInfo>* entries) OVERRIDE; |
65 virtual void OnItemAddedToPersistentStore(int32 download_id, | 62 virtual void OnItemAddedToPersistentStore(int32 download_id, |
66 int64 db_handle) OVERRIDE; | 63 int64 db_handle) OVERRIDE; |
67 virtual void ShowDownloadInBrowser(DownloadItem* download) OVERRIDE; | |
68 virtual int InProgressCount() const OVERRIDE; | 64 virtual int InProgressCount() const OVERRIDE; |
69 virtual content::BrowserContext* BrowserContext() OVERRIDE; | 65 virtual content::BrowserContext* BrowserContext() const OVERRIDE; |
70 virtual FilePath LastDownloadPath() OVERRIDE; | 66 virtual FilePath LastDownloadPath() OVERRIDE; |
71 virtual void CreateDownloadItem( | 67 virtual void CreateDownloadItem( |
72 DownloadCreateInfo* info, | 68 DownloadCreateInfo* info, |
73 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; |
74 virtual void ClearLastDownloadPath() OVERRIDE; | 75 virtual void ClearLastDownloadPath() OVERRIDE; |
75 virtual void FileSelected(const FilePath& path, void* params) OVERRIDE; | 76 virtual void FileSelected(const FilePath& path, void* params) OVERRIDE; |
76 virtual void FileSelectionCanceled(void* params) OVERRIDE; | 77 virtual void FileSelectionCanceled(void* params) OVERRIDE; |
77 virtual void RestartDownload(int32 download_id) OVERRIDE; | 78 virtual void RestartDownload(int32 download_id) OVERRIDE; |
78 virtual void MarkDownloadOpened(DownloadItem* download) OVERRIDE; | |
79 virtual void CheckForHistoryFilesRemoval() OVERRIDE; | 79 virtual void CheckForHistoryFilesRemoval() OVERRIDE; |
80 virtual void CheckForFileRemoval(DownloadItem* download_item) OVERRIDE; | |
81 virtual void AssertQueueStateConsistent(DownloadItem* download) OVERRIDE; | |
82 virtual DownloadItem* GetDownloadItem(int id) OVERRIDE; | 80 virtual DownloadItem* GetDownloadItem(int id) OVERRIDE; |
83 virtual void SavePageDownloadStarted(DownloadItem* download) OVERRIDE; | |
84 virtual void SavePageDownloadFinished(DownloadItem* download) OVERRIDE; | 81 virtual void SavePageDownloadFinished(DownloadItem* download) OVERRIDE; |
85 virtual DownloadItem* GetActiveDownloadItem(int id) OVERRIDE; | 82 virtual DownloadItem* GetActiveDownloadItem(int id) OVERRIDE; |
86 virtual content::DownloadManagerDelegate* delegate() const OVERRIDE; | 83 virtual content::DownloadManagerDelegate* delegate() const OVERRIDE; |
87 virtual void SetDownloadManagerDelegate( | 84 virtual void SetDownloadManagerDelegate( |
88 content::DownloadManagerDelegate* delegate) OVERRIDE; | 85 content::DownloadManagerDelegate* delegate) OVERRIDE; |
89 virtual DownloadId GetNextId() OVERRIDE; | |
90 virtual void ContinueDownloadWithPath(DownloadItem* download, | 86 virtual void ContinueDownloadWithPath(DownloadItem* download, |
91 const FilePath& chosen_file) OVERRIDE; | 87 const FilePath& chosen_file) OVERRIDE; |
92 virtual DownloadItem* GetActiveDownload(int32 download_id) OVERRIDE; | 88 virtual DownloadItem* GetActiveDownload(int32 download_id) OVERRIDE; |
93 virtual void SetFileManager(DownloadFileManager* file_manager) OVERRIDE; | 89 virtual void SetFileManager(DownloadFileManager* file_manager) OVERRIDE; |
94 | 90 |
95 private: | 91 private: |
96 content::DownloadManagerDelegate* delegate_; | 92 content::DownloadManagerDelegate* delegate_; |
97 DownloadIdFactory* id_factory_; | 93 DownloadIdFactory* id_factory_; |
98 DownloadStatusUpdater* updater_; | 94 DownloadStatusUpdater* updater_; |
99 DownloadFileManager* file_manager_; | 95 DownloadFileManager* file_manager_; |
100 std::map<int32, DownloadItem*> item_map_; | 96 std::map<int32, DownloadItem*> item_map_; |
101 std::map<int32, DownloadItem*> inactive_item_map_; | 97 std::map<int32, DownloadItem*> inactive_item_map_; |
102 }; | 98 }; |
103 | 99 |
104 #endif // CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ | 100 #endif // CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_MANAGER_H_ |
OLD | NEW |