| 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 #include <set> | 5 #include <set> |
| 6 #include <string> | 6 #include <string> |
| 7 | 7 |
| 8 #include "base/bind.h" | 8 #include "base/bind.h" |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| (...skipping 124 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 135 MOCK_METHOD1(SetOpenWhenComplete, void(bool)); | 135 MOCK_METHOD1(SetOpenWhenComplete, void(bool)); |
| 136 MOCK_CONST_METHOD0(GetFileExternallyRemoved, bool()); | 136 MOCK_CONST_METHOD0(GetFileExternallyRemoved, bool()); |
| 137 MOCK_CONST_METHOD0(GetSafetyState, SafetyState()); | 137 MOCK_CONST_METHOD0(GetSafetyState, SafetyState()); |
| 138 MOCK_CONST_METHOD0(GetDangerType, content::DownloadDangerType()); | 138 MOCK_CONST_METHOD0(GetDangerType, content::DownloadDangerType()); |
| 139 MOCK_CONST_METHOD0(IsDangerous, bool()); | 139 MOCK_CONST_METHOD0(IsDangerous, bool()); |
| 140 MOCK_METHOD0(GetAutoOpened, bool()); | 140 MOCK_METHOD0(GetAutoOpened, bool()); |
| 141 MOCK_CONST_METHOD0(GetTargetName, FilePath()); | 141 MOCK_CONST_METHOD0(GetTargetName, FilePath()); |
| 142 MOCK_CONST_METHOD0(GetForcedFilePath, const FilePath&()); | 142 MOCK_CONST_METHOD0(GetForcedFilePath, const FilePath&()); |
| 143 MOCK_CONST_METHOD0(HasUserGesture, bool()); | 143 MOCK_CONST_METHOD0(HasUserGesture, bool()); |
| 144 MOCK_CONST_METHOD0(GetTransitionType, content::PageTransition()); | 144 MOCK_CONST_METHOD0(GetTransitionType, content::PageTransition()); |
| 145 MOCK_CONST_METHOD0(IsOtr, bool()); | |
| 146 MOCK_CONST_METHOD0(IsTemporary, bool()); | 145 MOCK_CONST_METHOD0(IsTemporary, bool()); |
| 147 MOCK_METHOD1(SetIsTemporary, void(bool)); | 146 MOCK_METHOD1(SetIsTemporary, void(bool)); |
| 148 MOCK_METHOD1(SetOpened, void(bool)); | 147 MOCK_METHOD1(SetOpened, void(bool)); |
| 149 MOCK_CONST_METHOD0(GetOpened, bool()); | 148 MOCK_CONST_METHOD0(GetOpened, bool()); |
| 150 MOCK_CONST_METHOD0(GetLastModifiedTime, const std::string&()); | 149 MOCK_CONST_METHOD0(GetLastModifiedTime, const std::string&()); |
| 151 MOCK_CONST_METHOD0(GetETag, const std::string&()); | 150 MOCK_CONST_METHOD0(GetETag, const std::string&()); |
| 152 MOCK_CONST_METHOD0(GetLastReason, DownloadInterruptReason()); | 151 MOCK_CONST_METHOD0(GetLastReason, DownloadInterruptReason()); |
| 153 MOCK_CONST_METHOD0(GetPersistentStoreInfo, | 152 MOCK_CONST_METHOD0(GetPersistentStoreInfo, |
| 154 content::DownloadPersistentStoreInfo()); | 153 content::DownloadPersistentStoreInfo()); |
| 155 MOCK_CONST_METHOD0(GetBrowserContext, content::BrowserContext*()); | 154 MOCK_CONST_METHOD0(GetBrowserContext, content::BrowserContext*()); |
| (...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 269 // Overridden methods from DownloadItemFactory. | 268 // Overridden methods from DownloadItemFactory. |
| 270 virtual DownloadItemImpl* CreatePersistedItem( | 269 virtual DownloadItemImpl* CreatePersistedItem( |
| 271 DownloadItemImplDelegate* delegate, | 270 DownloadItemImplDelegate* delegate, |
| 272 content::DownloadId download_id, | 271 content::DownloadId download_id, |
| 273 const content::DownloadPersistentStoreInfo& info, | 272 const content::DownloadPersistentStoreInfo& info, |
| 274 const net::BoundNetLog& bound_net_log) OVERRIDE; | 273 const net::BoundNetLog& bound_net_log) OVERRIDE; |
| 275 virtual DownloadItemImpl* CreateActiveItem( | 274 virtual DownloadItemImpl* CreateActiveItem( |
| 276 DownloadItemImplDelegate* delegate, | 275 DownloadItemImplDelegate* delegate, |
| 277 const DownloadCreateInfo& info, | 276 const DownloadCreateInfo& info, |
| 278 scoped_ptr<DownloadRequestHandleInterface> request_handle, | 277 scoped_ptr<DownloadRequestHandleInterface> request_handle, |
| 279 bool is_otr, | |
| 280 const net::BoundNetLog& bound_net_log) OVERRIDE; | 278 const net::BoundNetLog& bound_net_log) OVERRIDE; |
| 281 virtual DownloadItemImpl* CreateSavePageItem( | 279 virtual DownloadItemImpl* CreateSavePageItem( |
| 282 DownloadItemImplDelegate* delegate, | 280 DownloadItemImplDelegate* delegate, |
| 283 const FilePath& path, | 281 const FilePath& path, |
| 284 const GURL& url, | 282 const GURL& url, |
| 285 bool is_otr, | |
| 286 content::DownloadId download_id, | 283 content::DownloadId download_id, |
| 287 const std::string& mime_type, | 284 const std::string& mime_type, |
| 288 const net::BoundNetLog& bound_net_log) OVERRIDE; | 285 const net::BoundNetLog& bound_net_log) OVERRIDE; |
| 289 | 286 |
| 290 private: | 287 private: |
| 291 std::map<int32, MockDownloadItemImpl*> items_; | 288 std::map<int32, MockDownloadItemImpl*> items_; |
| 292 DownloadItemImplDelegate item_delegate_; | 289 DownloadItemImplDelegate item_delegate_; |
| 293 | 290 |
| 294 DISALLOW_COPY_AND_ASSIGN(MockDownloadItemFactory); | 291 DISALLOW_COPY_AND_ASSIGN(MockDownloadItemFactory); |
| 295 }; | 292 }; |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 334 .WillRepeatedly(Return(local_id)); | 331 .WillRepeatedly(Return(local_id)); |
| 335 items_[local_id] = result; | 332 items_[local_id] = result; |
| 336 | 333 |
| 337 return result; | 334 return result; |
| 338 } | 335 } |
| 339 | 336 |
| 340 DownloadItemImpl* MockDownloadItemFactory::CreateActiveItem( | 337 DownloadItemImpl* MockDownloadItemFactory::CreateActiveItem( |
| 341 DownloadItemImplDelegate* delegate, | 338 DownloadItemImplDelegate* delegate, |
| 342 const DownloadCreateInfo& info, | 339 const DownloadCreateInfo& info, |
| 343 scoped_ptr<DownloadRequestHandleInterface> request_handle, | 340 scoped_ptr<DownloadRequestHandleInterface> request_handle, |
| 344 bool is_otr, | |
| 345 const net::BoundNetLog& bound_net_log) { | 341 const net::BoundNetLog& bound_net_log) { |
| 346 int local_id = info.download_id.local(); | 342 int local_id = info.download_id.local(); |
| 347 DCHECK(items_.find(local_id) == items_.end()); | 343 DCHECK(items_.find(local_id) == items_.end()); |
| 348 | 344 |
| 349 MockDownloadItemImpl* result = | 345 MockDownloadItemImpl* result = |
| 350 new StrictMock<MockDownloadItemImpl>(&item_delegate_); | 346 new StrictMock<MockDownloadItemImpl>(&item_delegate_); |
| 351 EXPECT_CALL(*result, GetId()) | 347 EXPECT_CALL(*result, GetId()) |
| 352 .WillRepeatedly(Return(local_id)); | 348 .WillRepeatedly(Return(local_id)); |
| 353 items_[local_id] = result; | 349 items_[local_id] = result; |
| 354 | 350 |
| 355 return result; | 351 return result; |
| 356 } | 352 } |
| 357 | 353 |
| 358 DownloadItemImpl* MockDownloadItemFactory::CreateSavePageItem( | 354 DownloadItemImpl* MockDownloadItemFactory::CreateSavePageItem( |
| 359 DownloadItemImplDelegate* delegate, | 355 DownloadItemImplDelegate* delegate, |
| 360 const FilePath& path, | 356 const FilePath& path, |
| 361 const GURL& url, | 357 const GURL& url, |
| 362 bool is_otr, | |
| 363 content::DownloadId download_id, | 358 content::DownloadId download_id, |
| 364 const std::string& mime_type, | 359 const std::string& mime_type, |
| 365 const net::BoundNetLog& bound_net_log) { | 360 const net::BoundNetLog& bound_net_log) { |
| 366 int local_id = download_id.local(); | 361 int local_id = download_id.local(); |
| 367 DCHECK(items_.find(local_id) == items_.end()); | 362 DCHECK(items_.find(local_id) == items_.end()); |
| 368 | 363 |
| 369 MockDownloadItemImpl* result = | 364 MockDownloadItemImpl* result = |
| 370 new StrictMock<MockDownloadItemImpl>(&item_delegate_); | 365 new StrictMock<MockDownloadItemImpl>(&item_delegate_); |
| 371 EXPECT_CALL(*result, GetId()) | 366 EXPECT_CALL(*result, GetId()) |
| 372 .WillRepeatedly(Return(local_id)); | 367 .WillRepeatedly(Return(local_id)); |
| (...skipping 300 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 673 UpdateItemInPersistentStore(&item)); | 668 UpdateItemInPersistentStore(&item)); |
| 674 EXPECT_CALL(item, GetState()) | 669 EXPECT_CALL(item, GetState()) |
| 675 .WillRepeatedly(Return(DownloadItem::CANCELLED)); | 670 .WillRepeatedly(Return(DownloadItem::CANCELLED)); |
| 676 EXPECT_CALL(item, GetDbHandle()) | 671 EXPECT_CALL(item, GetDbHandle()) |
| 677 .WillRepeatedly(Return(db_handle)); | 672 .WillRepeatedly(Return(db_handle)); |
| 678 | 673 |
| 679 EXPECT_CALL(item, OffThreadCancel(&GetMockDownloadFileManager())); | 674 EXPECT_CALL(item, OffThreadCancel(&GetMockDownloadFileManager())); |
| 680 DownloadStopped(&item); | 675 DownloadStopped(&item); |
| 681 EXPECT_EQ(NULL, download_manager_->GetActiveDownloadItem(download_id)); | 676 EXPECT_EQ(NULL, download_manager_->GetActiveDownloadItem(download_id)); |
| 682 } | 677 } |
| OLD | NEW |