| 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 |