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 "base/message_loop.h" | 5 #include "base/message_loop.h" |
6 #include "base/stl_util.h" | 6 #include "base/stl_util.h" |
7 #include "base/threading/thread.h" | 7 #include "base/threading/thread.h" |
8 #include "content/browser/download/byte_stream.h" | 8 #include "content/browser/download/byte_stream.h" |
9 #include "content/browser/download/download_create_info.h" | 9 #include "content/browser/download/download_create_info.h" |
10 #include "content/browser/download/download_file_factory.h" | 10 #include "content/browser/download/download_file_manager.h" |
11 #include "content/browser/download/download_item_impl.h" | 11 #include "content/browser/download/download_item_impl.h" |
12 #include "content/browser/download/download_item_impl_delegate.h" | 12 #include "content/browser/download/download_item_impl_delegate.h" |
13 #include "content/browser/download/download_request_handle.h" | 13 #include "content/browser/download/download_request_handle.h" |
14 #include "content/browser/download/mock_download_file.h" | |
15 #include "content/public/browser/download_id.h" | 14 #include "content/public/browser/download_id.h" |
16 #include "content/public/browser/download_destination_observer.h" | |
17 #include "content/public/browser/download_interrupt_reasons.h" | 15 #include "content/public/browser/download_interrupt_reasons.h" |
18 #include "content/public/test/mock_download_item.h" | 16 #include "content/public/test/mock_download_item.h" |
19 #include "content/public/test/test_browser_thread.h" | 17 #include "content/public/test/test_browser_thread.h" |
20 #include "testing/gmock/include/gmock/gmock.h" | 18 #include "testing/gmock/include/gmock/gmock.h" |
21 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
22 | 20 |
23 using content::BrowserThread; | 21 using content::BrowserThread; |
24 using content::DownloadId; | 22 using content::DownloadId; |
25 using content::DownloadItem; | 23 using content::DownloadItem; |
26 using content::DownloadManager; | 24 using content::DownloadManager; |
27 using content::MockDownloadItem; | 25 using content::MockDownloadItem; |
28 using content::WebContents; | 26 using content::WebContents; |
29 using ::testing::_; | 27 using ::testing::_; |
30 using ::testing::AllOf; | 28 using ::testing::AllOf; |
31 using ::testing::Property; | 29 using ::testing::Property; |
32 using ::testing::Return; | 30 using ::testing::Return; |
33 using ::testing::StrictMock; | |
34 | 31 |
35 DownloadId::Domain kValidDownloadItemIdDomain = "valid DownloadId::Domain"; | 32 DownloadId::Domain kValidDownloadItemIdDomain = "valid DownloadId::Domain"; |
36 | 33 |
37 namespace { | 34 namespace { |
38 class MockDelegate : public DownloadItemImplDelegate { | 35 class MockDelegate : public DownloadItemImplDelegate { |
39 public: | 36 public: |
40 MOCK_METHOD1(DelegateStart, void(DownloadItemImpl* download)); | 37 MockDelegate(DownloadFileManager* file_manager) |
| 38 : file_manager_(file_manager) { |
| 39 } |
| 40 MOCK_METHOD1(ShouldOpenFileBasedOnExtension, bool(const FilePath& path)); |
41 MOCK_METHOD1(ShouldOpenDownload, bool(DownloadItemImpl* download)); | 41 MOCK_METHOD1(ShouldOpenDownload, bool(DownloadItemImpl* download)); |
42 MOCK_METHOD1(ShouldOpenFileBasedOnExtension, bool(const FilePath& path)); | |
43 MOCK_METHOD1(CheckForFileRemoval, void(DownloadItemImpl* download)); | 42 MOCK_METHOD1(CheckForFileRemoval, void(DownloadItemImpl* download)); |
44 MOCK_METHOD1(MaybeCompleteDownload, void(DownloadItemImpl* download)); | 43 MOCK_METHOD1(MaybeCompleteDownload, void(DownloadItemImpl* download)); |
45 MOCK_CONST_METHOD0(GetBrowserContext, content::BrowserContext*()); | 44 MOCK_CONST_METHOD0(GetBrowserContext, content::BrowserContext*()); |
46 MOCK_METHOD1(DownloadStopped, void(DownloadItemImpl* download)); | 45 MOCK_METHOD1(DownloadStopped, void(DownloadItemImpl* download)); |
47 MOCK_METHOD1(DownloadCompleted, void(DownloadItemImpl* download)); | 46 MOCK_METHOD1(DownloadCompleted, void(DownloadItemImpl* download)); |
48 MOCK_METHOD1(DownloadOpened, void(DownloadItemImpl* download)); | 47 MOCK_METHOD1(DownloadOpened, void(DownloadItemImpl* download)); |
49 MOCK_METHOD1(DownloadRemoved, void(DownloadItemImpl* download)); | 48 MOCK_METHOD1(DownloadRemoved, void(DownloadItemImpl* download)); |
50 MOCK_METHOD1(DownloadRenamedToIntermediateName, | 49 MOCK_METHOD1(DownloadRenamedToIntermediateName, |
51 void(DownloadItemImpl* download)); | 50 void(DownloadItemImpl* download)); |
52 MOCK_METHOD1(DownloadRenamedToFinalName, void(DownloadItemImpl* download)); | 51 MOCK_METHOD1(DownloadRenamedToFinalName, void(DownloadItemImpl* download)); |
53 MOCK_CONST_METHOD1(AssertStateConsistent, void(DownloadItemImpl* download)); | 52 MOCK_CONST_METHOD1(AssertStateConsistent, void(DownloadItemImpl* download)); |
54 virtual DownloadFileManager* GetDownloadFileManager() OVERRIDE { | 53 virtual DownloadFileManager* GetDownloadFileManager() OVERRIDE { |
55 return file_manager_; | 54 return file_manager_; |
56 } | 55 } |
57 private: | 56 private: |
58 DownloadFileManager* file_manager_; | 57 DownloadFileManager* file_manager_; |
59 }; | 58 }; |
60 | 59 |
61 class MockRequestHandle : public DownloadRequestHandleInterface { | 60 class MockRequestHandle : public DownloadRequestHandleInterface { |
62 public: | 61 public: |
63 MOCK_CONST_METHOD0(GetWebContents, WebContents*()); | 62 MOCK_CONST_METHOD0(GetWebContents, WebContents*()); |
64 MOCK_CONST_METHOD0(GetDownloadManager, DownloadManager*()); | 63 MOCK_CONST_METHOD0(GetDownloadManager, DownloadManager*()); |
65 MOCK_CONST_METHOD0(PauseRequest, void()); | 64 MOCK_CONST_METHOD0(PauseRequest, void()); |
66 MOCK_CONST_METHOD0(ResumeRequest, void()); | 65 MOCK_CONST_METHOD0(ResumeRequest, void()); |
67 MOCK_CONST_METHOD0(CancelRequest, void()); | 66 MOCK_CONST_METHOD0(CancelRequest, void()); |
68 MOCK_CONST_METHOD0(DebugString, std::string()); | 67 MOCK_CONST_METHOD0(DebugString, std::string()); |
69 }; | 68 }; |
70 | 69 |
| 70 class MockDownloadFileFactory |
| 71 : public DownloadFileManager::DownloadFileFactory { |
| 72 public: |
| 73 content::DownloadFile* CreateFile( |
| 74 DownloadCreateInfo* info, |
| 75 scoped_ptr<content::ByteStreamReader> stream_reader, |
| 76 DownloadManager* mgr, |
| 77 bool calculate_hash, |
| 78 const net::BoundNetLog& bound_net_log) { |
| 79 return MockCreateFile( |
| 80 info, stream_reader.get(), info->request_handle, mgr, calculate_hash, |
| 81 bound_net_log); |
| 82 } |
| 83 |
| 84 MOCK_METHOD6(MockCreateFile, |
| 85 content::DownloadFile*(DownloadCreateInfo*, |
| 86 content::ByteStreamReader*, |
| 87 const DownloadRequestHandle&, |
| 88 DownloadManager*, |
| 89 bool, |
| 90 const net::BoundNetLog&)); |
| 91 }; |
| 92 |
| 93 class MockDownloadFileManager : public DownloadFileManager { |
| 94 public: |
| 95 MockDownloadFileManager(); |
| 96 MOCK_METHOD0(Shutdown, void()); |
| 97 MOCK_METHOD1(CancelDownload, void(DownloadId)); |
| 98 MOCK_METHOD2(CompleteDownload, void(DownloadId, const base::Closure&)); |
| 99 MOCK_METHOD1(OnDownloadManagerShutdown, void(DownloadManager*)); |
| 100 MOCK_METHOD4(RenameDownloadFile, void(DownloadId, const FilePath&, bool, |
| 101 const RenameCompletionCallback&)); |
| 102 MOCK_CONST_METHOD0(NumberOfActiveDownloads, int()); |
| 103 private: |
| 104 ~MockDownloadFileManager() {} |
| 105 }; |
| 106 |
71 // Schedules a task to invoke the RenameCompletionCallback with |new_path| on | 107 // Schedules a task to invoke the RenameCompletionCallback with |new_path| on |
72 // the UI thread. Should only be used as the action for | 108 // the UI thread. Should only be used as the action for |
73 // MockDownloadFile::Rename as follows: | 109 // MockDownloadFileManager::Rename*DownloadFile as follows: |
74 // EXPECT_CALL(download_file, Rename(_,_,_)) | 110 // EXPECT_CALL(mock_download_file_manager, |
| 111 // RenameDownloadFile(_,_,_,_)) |
75 // .WillOnce(ScheduleRenameCallback(new_path)); | 112 // .WillOnce(ScheduleRenameCallback(new_path)); |
76 ACTION_P(ScheduleRenameCallback, new_path) { | 113 ACTION_P(ScheduleRenameCallback, new_path) { |
77 BrowserThread::PostTask( | 114 BrowserThread::PostTask( |
78 BrowserThread::UI, FROM_HERE, | 115 BrowserThread::UI, FROM_HERE, |
79 base::Bind(arg2, content::DOWNLOAD_INTERRUPT_REASON_NONE, new_path)); | 116 base::Bind(arg3, content::DOWNLOAD_INTERRUPT_REASON_NONE, new_path)); |
80 } | 117 } |
81 | 118 |
82 // Similarly for scheduling a completion callback. | 119 // Similarly for scheduling a completion callback. |
83 ACTION(ScheduleCompleteCallback) { | 120 ACTION(ScheduleCompleteCallback) { |
84 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, base::Bind(arg1)); | 121 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, base::Bind(arg1)); |
85 } | 122 } |
86 | 123 |
| 124 MockDownloadFileManager::MockDownloadFileManager() |
| 125 : DownloadFileManager(new MockDownloadFileFactory) { |
| 126 } |
| 127 |
87 } // namespace | 128 } // namespace |
88 | 129 |
89 class DownloadItemTest : public testing::Test { | 130 class DownloadItemTest : public testing::Test { |
90 public: | 131 public: |
91 class MockObserver : public DownloadItem::Observer { | 132 class MockObserver : public DownloadItem::Observer { |
92 public: | 133 public: |
93 explicit MockObserver(DownloadItem* item) | 134 explicit MockObserver(DownloadItem* item) |
94 : item_(item), | 135 : item_(item), |
95 removed_(false), | 136 removed_(false), |
96 destroyed_(false), | 137 destroyed_(false), |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
136 private: | 177 private: |
137 DownloadItem* item_; | 178 DownloadItem* item_; |
138 bool removed_; | 179 bool removed_; |
139 bool destroyed_; | 180 bool destroyed_; |
140 bool updated_; | 181 bool updated_; |
141 }; | 182 }; |
142 | 183 |
143 DownloadItemTest() | 184 DownloadItemTest() |
144 : ui_thread_(BrowserThread::UI, &loop_), | 185 : ui_thread_(BrowserThread::UI, &loop_), |
145 file_thread_(BrowserThread::FILE, &loop_), | 186 file_thread_(BrowserThread::FILE, &loop_), |
146 delegate_() { | 187 file_manager_(new MockDownloadFileManager), |
| 188 delegate_(file_manager_.get()) { |
147 } | 189 } |
148 | 190 |
149 ~DownloadItemTest() { | 191 ~DownloadItemTest() { |
150 } | 192 } |
151 | 193 |
152 virtual void SetUp() { | 194 virtual void SetUp() { |
153 } | 195 } |
154 | 196 |
155 virtual void TearDown() { | 197 virtual void TearDown() { |
156 ui_thread_.DeprecatedGetThreadObject()->message_loop()->RunAllPending(); | 198 ui_thread_.DeprecatedGetThreadObject()->message_loop()->RunAllPending(); |
(...skipping 20 matching lines...) Expand all Loading... |
177 | 219 |
178 scoped_ptr<DownloadRequestHandleInterface> request_handle( | 220 scoped_ptr<DownloadRequestHandleInterface> request_handle( |
179 new testing::NiceMock<MockRequestHandle>); | 221 new testing::NiceMock<MockRequestHandle>); |
180 DownloadItemImpl* download = | 222 DownloadItemImpl* download = |
181 new DownloadItemImpl(&delegate_, *(info_.get()), | 223 new DownloadItemImpl(&delegate_, *(info_.get()), |
182 request_handle.Pass(), net::BoundNetLog()); | 224 request_handle.Pass(), net::BoundNetLog()); |
183 allocated_downloads_.insert(download); | 225 allocated_downloads_.insert(download); |
184 return download; | 226 return download; |
185 } | 227 } |
186 | 228 |
187 // Add DownloadFile to DownloadItem | |
188 MockDownloadFile* AddDownloadFileToDownloadItem(DownloadItemImpl* item) { | |
189 MockDownloadFile* mock_download_file(new StrictMock<MockDownloadFile>); | |
190 scoped_ptr<content::DownloadFile> download_file(mock_download_file); | |
191 EXPECT_CALL(*mock_download_file, Initialize(_)); | |
192 item->Start(download_file.Pass()); | |
193 loop_.RunAllPending(); | |
194 return mock_download_file; | |
195 } | |
196 | |
197 // Cleanup a download item (specifically get rid of the DownloadFile on it). | |
198 // The item must be in the IN_PROGRESS state. | |
199 void CleanupItem(DownloadItemImpl* item, MockDownloadFile* download_file) { | |
200 EXPECT_EQ(content::DownloadItem::IN_PROGRESS, item->GetState()); | |
201 | |
202 EXPECT_CALL(*download_file, Cancel()); | |
203 EXPECT_CALL(delegate_, DownloadStopped(item)); | |
204 item->Cancel(true); | |
205 loop_.RunAllPending(); | |
206 } | |
207 | |
208 // Destroy a previously created download item. | 229 // Destroy a previously created download item. |
209 void DestroyDownloadItem(DownloadItem* item) { | 230 void DestroyDownloadItem(DownloadItem* item) { |
210 allocated_downloads_.erase(item); | 231 allocated_downloads_.erase(item); |
211 delete item; | 232 delete item; |
212 } | 233 } |
213 | 234 |
214 void RunAllPendingInMessageLoops() { | 235 void RunAllPendingInMessageLoops() { |
215 loop_.RunAllPending(); | 236 loop_.RunAllPending(); |
216 } | 237 } |
217 | 238 |
218 MockDelegate* mock_delegate() { | 239 MockDelegate* mock_delegate() { |
219 return &delegate_; | 240 return &delegate_; |
220 } | 241 } |
221 | 242 |
| 243 MockDownloadFileManager* mock_file_manager() { |
| 244 return file_manager_.get(); |
| 245 } |
| 246 |
222 private: | 247 private: |
223 MessageLoopForUI loop_; | 248 MessageLoopForUI loop_; |
224 content::TestBrowserThread ui_thread_; // UI thread | 249 content::TestBrowserThread ui_thread_; // UI thread |
225 content::TestBrowserThread file_thread_; // FILE thread | 250 content::TestBrowserThread file_thread_; // FILE thread |
| 251 scoped_refptr<MockDownloadFileManager> file_manager_; |
226 testing::NiceMock<MockDelegate> delegate_; | 252 testing::NiceMock<MockDelegate> delegate_; |
227 std::set<DownloadItem*> allocated_downloads_; | 253 std::set<DownloadItem*> allocated_downloads_; |
228 }; | 254 }; |
229 | 255 |
230 namespace { | 256 namespace { |
231 | 257 |
232 const int kDownloadChunkSize = 1000; | 258 const int kDownloadChunkSize = 1000; |
233 const int kDownloadSpeed = 1000; | 259 const int kDownloadSpeed = 1000; |
234 const int kDummyDBHandle = 10; | 260 const int kDummyDBHandle = 10; |
235 const FilePath::CharType kDummyPath[] = FILE_PATH_LITERAL("/testpath"); | 261 const FilePath::CharType kDummyPath[] = FILE_PATH_LITERAL("/testpath"); |
(...skipping 13 matching lines...) Expand all Loading... |
249 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 275 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
250 MockObserver observer(item); | 276 MockObserver observer(item); |
251 | 277 |
252 item->UpdateProgress(kDownloadChunkSize, kDownloadSpeed, ""); | 278 item->UpdateProgress(kDownloadChunkSize, kDownloadSpeed, ""); |
253 ASSERT_TRUE(observer.CheckUpdated()); | 279 ASSERT_TRUE(observer.CheckUpdated()); |
254 EXPECT_EQ(kDownloadSpeed, item->CurrentSpeed()); | 280 EXPECT_EQ(kDownloadSpeed, item->CurrentSpeed()); |
255 } | 281 } |
256 | 282 |
257 TEST_F(DownloadItemTest, NotificationAfterCancel) { | 283 TEST_F(DownloadItemTest, NotificationAfterCancel) { |
258 DownloadItemImpl* user_cancel = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 284 DownloadItemImpl* user_cancel = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
259 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(user_cancel); | |
260 EXPECT_CALL(*download_file, Cancel()); | |
261 MockObserver observer1(user_cancel); | 285 MockObserver observer1(user_cancel); |
262 | 286 |
263 user_cancel->Cancel(true); | 287 user_cancel->Cancel(true); |
264 ASSERT_TRUE(observer1.CheckUpdated()); | 288 ASSERT_TRUE(observer1.CheckUpdated()); |
265 RunAllPendingInMessageLoops(); | |
266 | 289 |
267 DownloadItemImpl* system_cancel = | 290 DownloadItemImpl* system_cancel = |
268 CreateDownloadItem(DownloadItem::IN_PROGRESS); | 291 CreateDownloadItem(DownloadItem::IN_PROGRESS); |
269 download_file = AddDownloadFileToDownloadItem(system_cancel); | |
270 EXPECT_CALL(*download_file, Cancel()); | |
271 MockObserver observer2(system_cancel); | 292 MockObserver observer2(system_cancel); |
272 | 293 |
273 system_cancel->Cancel(false); | 294 system_cancel->Cancel(false); |
274 ASSERT_TRUE(observer2.CheckUpdated()); | 295 ASSERT_TRUE(observer2.CheckUpdated()); |
275 RunAllPendingInMessageLoops(); | |
276 } | 296 } |
277 | 297 |
278 TEST_F(DownloadItemTest, NotificationAfterComplete) { | 298 TEST_F(DownloadItemTest, NotificationAfterComplete) { |
279 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 299 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
280 MockObserver observer(item); | 300 MockObserver observer(item); |
281 | 301 |
282 item->OnAllDataSaved(DownloadItem::kEmptyFileHash); | 302 item->OnAllDataSaved(kDownloadChunkSize, DownloadItem::kEmptyFileHash); |
283 ASSERT_TRUE(observer.CheckUpdated()); | 303 ASSERT_TRUE(observer.CheckUpdated()); |
284 | 304 |
285 item->MarkAsComplete(); | 305 item->MarkAsComplete(); |
286 ASSERT_TRUE(observer.CheckUpdated()); | 306 ASSERT_TRUE(observer.CheckUpdated()); |
287 } | 307 } |
288 | 308 |
289 TEST_F(DownloadItemTest, NotificationAfterDownloadedFileRemoved) { | 309 TEST_F(DownloadItemTest, NotificationAfterDownloadedFileRemoved) { |
290 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 310 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
291 MockObserver observer(item); | 311 MockObserver observer(item); |
292 | 312 |
293 item->OnDownloadedFileRemoved(); | 313 item->OnDownloadedFileRemoved(); |
294 ASSERT_TRUE(observer.CheckUpdated()); | 314 ASSERT_TRUE(observer.CheckUpdated()); |
295 } | 315 } |
296 | 316 |
297 TEST_F(DownloadItemTest, NotificationAfterInterrupted) { | 317 TEST_F(DownloadItemTest, NotificationAfterInterrupted) { |
298 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 318 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
299 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
300 EXPECT_CALL(*download_file, Cancel()); | |
301 MockObserver observer(item); | 319 MockObserver observer(item); |
302 | 320 |
303 item->Interrupt(content::DOWNLOAD_INTERRUPT_REASON_NONE); | 321 item->Interrupt(content::DOWNLOAD_INTERRUPT_REASON_NONE); |
304 ASSERT_TRUE(observer.CheckUpdated()); | 322 ASSERT_TRUE(observer.CheckUpdated()); |
305 RunAllPendingInMessageLoops(); | |
306 } | 323 } |
307 | 324 |
308 TEST_F(DownloadItemTest, NotificationAfterDelete) { | 325 TEST_F(DownloadItemTest, NotificationAfterDelete) { |
309 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 326 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
310 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
311 EXPECT_CALL(*download_file, Cancel()); | |
312 MockObserver observer(item); | 327 MockObserver observer(item); |
313 | 328 |
314 item->Delete(DownloadItem::DELETE_DUE_TO_BROWSER_SHUTDOWN); | 329 item->Delete(DownloadItem::DELETE_DUE_TO_BROWSER_SHUTDOWN); |
315 ASSERT_TRUE(observer.CheckUpdated()); | 330 ASSERT_TRUE(observer.CheckUpdated()); |
316 RunAllPendingInMessageLoops(); | |
317 } | 331 } |
318 | 332 |
319 TEST_F(DownloadItemTest, NotificationAfterDestroyed) { | 333 TEST_F(DownloadItemTest, NotificationAfterDestroyed) { |
320 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 334 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
321 MockObserver observer(item); | 335 MockObserver observer(item); |
322 | 336 |
323 DestroyDownloadItem(item); | 337 DestroyDownloadItem(item); |
324 ASSERT_TRUE(observer.CheckDestroyed()); | 338 ASSERT_TRUE(observer.CheckDestroyed()); |
325 } | 339 } |
326 | 340 |
327 TEST_F(DownloadItemTest, NotificationAfterRemove) { | 341 TEST_F(DownloadItemTest, NotificationAfterRemove) { |
328 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 342 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
329 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
330 EXPECT_CALL(*download_file, Cancel()); | |
331 MockObserver observer(item); | 343 MockObserver observer(item); |
332 | 344 |
333 item->Remove(); | 345 item->Remove(); |
334 ASSERT_TRUE(observer.CheckUpdated()); | 346 ASSERT_TRUE(observer.CheckUpdated()); |
335 ASSERT_TRUE(observer.CheckRemoved()); | 347 ASSERT_TRUE(observer.CheckRemoved()); |
336 } | 348 } |
337 | 349 |
338 TEST_F(DownloadItemTest, NotificationAfterOnContentCheckCompleted) { | 350 TEST_F(DownloadItemTest, NotificationAfterOnContentCheckCompleted) { |
339 // Setting to NOT_DANGEROUS does not trigger a notification. | 351 // Setting to NOT_DANGEROUS does not trigger a notification. |
340 DownloadItemImpl* safe_item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 352 DownloadItemImpl* safe_item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
341 MockObserver safe_observer(safe_item); | 353 MockObserver safe_observer(safe_item); |
342 | 354 |
343 safe_item->OnAllDataSaved(""); | 355 safe_item->OnAllDataSaved(1, ""); |
344 EXPECT_TRUE(safe_observer.CheckUpdated()); | 356 EXPECT_TRUE(safe_observer.CheckUpdated()); |
345 safe_item->OnContentCheckCompleted( | 357 safe_item->OnContentCheckCompleted( |
346 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS); | 358 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS); |
347 EXPECT_TRUE(safe_observer.CheckUpdated()); | 359 EXPECT_TRUE(safe_observer.CheckUpdated()); |
348 | 360 |
349 // Setting to unsafe url or unsafe file should trigger a notification. | 361 // Setting to unsafe url or unsafe file should trigger a notification. |
350 DownloadItemImpl* unsafeurl_item = | 362 DownloadItemImpl* unsafeurl_item = |
351 CreateDownloadItem(DownloadItem::IN_PROGRESS); | 363 CreateDownloadItem(DownloadItem::IN_PROGRESS); |
352 MockObserver unsafeurl_observer(unsafeurl_item); | 364 MockObserver unsafeurl_observer(unsafeurl_item); |
353 | 365 |
354 unsafeurl_item->OnAllDataSaved(""); | 366 unsafeurl_item->OnAllDataSaved(1, ""); |
355 EXPECT_TRUE(unsafeurl_observer.CheckUpdated()); | 367 EXPECT_TRUE(unsafeurl_observer.CheckUpdated()); |
356 unsafeurl_item->OnContentCheckCompleted( | 368 unsafeurl_item->OnContentCheckCompleted( |
357 content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL); | 369 content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL); |
358 EXPECT_TRUE(unsafeurl_observer.CheckUpdated()); | 370 EXPECT_TRUE(unsafeurl_observer.CheckUpdated()); |
359 | 371 |
360 unsafeurl_item->DangerousDownloadValidated(); | 372 unsafeurl_item->DangerousDownloadValidated(); |
361 EXPECT_TRUE(unsafeurl_observer.CheckUpdated()); | 373 EXPECT_TRUE(unsafeurl_observer.CheckUpdated()); |
362 | 374 |
363 DownloadItemImpl* unsafefile_item = | 375 DownloadItemImpl* unsafefile_item = |
364 CreateDownloadItem(DownloadItem::IN_PROGRESS); | 376 CreateDownloadItem(DownloadItem::IN_PROGRESS); |
365 MockObserver unsafefile_observer(unsafefile_item); | 377 MockObserver unsafefile_observer(unsafefile_item); |
366 | 378 |
367 unsafefile_item->OnAllDataSaved(""); | 379 unsafefile_item->OnAllDataSaved(1, ""); |
368 EXPECT_TRUE(unsafefile_observer.CheckUpdated()); | 380 EXPECT_TRUE(unsafefile_observer.CheckUpdated()); |
369 unsafefile_item->OnContentCheckCompleted( | 381 unsafefile_item->OnContentCheckCompleted( |
370 content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE); | 382 content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE); |
371 EXPECT_TRUE(unsafefile_observer.CheckUpdated()); | 383 EXPECT_TRUE(unsafefile_observer.CheckUpdated()); |
372 | 384 |
373 unsafefile_item->DangerousDownloadValidated(); | 385 unsafefile_item->DangerousDownloadValidated(); |
374 EXPECT_TRUE(unsafefile_observer.CheckUpdated()); | 386 EXPECT_TRUE(unsafefile_observer.CheckUpdated()); |
375 } | 387 } |
376 | 388 |
377 // DownloadItemImpl::OnDownloadTargetDetermined will schedule a task to run | 389 // DownloadItemImpl::OnDownloadTargetDetermined will schedule a task to run |
378 // DownloadFile::Rename(). Once the rename | 390 // DownloadFileManager::RenameDownloadFile(). Once the rename |
379 // completes, DownloadItemImpl receives a notification with the new file | 391 // completes, DownloadItemImpl receives a notification with the new file |
380 // name. Check that observers are updated when the new filename is available and | 392 // name. Check that observers are updated when the new filename is available and |
381 // not before. | 393 // not before. |
382 TEST_F(DownloadItemTest, NotificationAfterOnDownloadTargetDetermined) { | 394 TEST_F(DownloadItemTest, NotificationAfterOnDownloadTargetDetermined) { |
383 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 395 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
384 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
385 MockObserver observer(item); | 396 MockObserver observer(item); |
386 FilePath target_path(kDummyPath); | 397 FilePath target_path(kDummyPath); |
387 FilePath intermediate_path(target_path.InsertBeforeExtensionASCII("x")); | 398 FilePath intermediate_path(target_path.InsertBeforeExtensionASCII("x")); |
388 FilePath new_intermediate_path(target_path.InsertBeforeExtensionASCII("y")); | 399 FilePath new_intermediate_path(target_path.InsertBeforeExtensionASCII("y")); |
389 EXPECT_CALL(*download_file, Rename(intermediate_path, false, _)) | 400 EXPECT_CALL(*mock_file_manager(), |
| 401 RenameDownloadFile(_,intermediate_path,false,_)) |
390 .WillOnce(ScheduleRenameCallback(new_intermediate_path)); | 402 .WillOnce(ScheduleRenameCallback(new_intermediate_path)); |
391 | 403 |
392 // Currently, a notification would be generated if the danger type is anything | 404 // Currently, a notification would be generated if the danger type is anything |
393 // other than NOT_DANGEROUS. | 405 // other than NOT_DANGEROUS. |
394 item->OnDownloadTargetDetermined(target_path, | 406 item->OnDownloadTargetDetermined(target_path, |
395 DownloadItem::TARGET_DISPOSITION_OVERWRITE, | 407 DownloadItem::TARGET_DISPOSITION_OVERWRITE, |
396 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, | 408 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, |
397 intermediate_path); | 409 intermediate_path); |
398 EXPECT_FALSE(observer.CheckUpdated()); | 410 EXPECT_FALSE(observer.CheckUpdated()); |
399 RunAllPendingInMessageLoops(); | 411 RunAllPendingInMessageLoops(); |
400 EXPECT_TRUE(observer.CheckUpdated()); | 412 EXPECT_TRUE(observer.CheckUpdated()); |
401 EXPECT_EQ(new_intermediate_path, item->GetFullPath()); | 413 EXPECT_EQ(new_intermediate_path, item->GetFullPath()); |
402 | |
403 CleanupItem(item, download_file); | |
404 } | 414 } |
405 | 415 |
406 TEST_F(DownloadItemTest, NotificationAfterTogglePause) { | 416 TEST_F(DownloadItemTest, NotificationAfterTogglePause) { |
407 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 417 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
408 MockObserver observer(item); | 418 MockObserver observer(item); |
409 | 419 |
410 item->TogglePause(); | 420 item->TogglePause(); |
411 ASSERT_TRUE(observer.CheckUpdated()); | 421 ASSERT_TRUE(observer.CheckUpdated()); |
412 | 422 |
413 item->TogglePause(); | 423 item->TogglePause(); |
414 ASSERT_TRUE(observer.CheckUpdated()); | 424 ASSERT_TRUE(observer.CheckUpdated()); |
415 } | 425 } |
416 | 426 |
417 TEST_F(DownloadItemTest, DisplayName) { | 427 TEST_F(DownloadItemTest, DisplayName) { |
418 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 428 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
419 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
420 FilePath target_path(FilePath(kDummyPath).AppendASCII("foo.bar")); | 429 FilePath target_path(FilePath(kDummyPath).AppendASCII("foo.bar")); |
421 FilePath intermediate_path(target_path.InsertBeforeExtensionASCII("x")); | 430 FilePath intermediate_path(target_path.InsertBeforeExtensionASCII("x")); |
422 EXPECT_EQ(FILE_PATH_LITERAL(""), | 431 EXPECT_EQ(FILE_PATH_LITERAL(""), |
423 item->GetFileNameToReportUser().value()); | 432 item->GetFileNameToReportUser().value()); |
424 EXPECT_CALL(*download_file, Rename(_, false, _)) | 433 EXPECT_CALL(*mock_file_manager(), |
| 434 RenameDownloadFile(_,_,false,_)) |
425 .WillOnce(ScheduleRenameCallback(intermediate_path)); | 435 .WillOnce(ScheduleRenameCallback(intermediate_path)); |
426 item->OnDownloadTargetDetermined(target_path, | 436 item->OnDownloadTargetDetermined(target_path, |
427 DownloadItem::TARGET_DISPOSITION_OVERWRITE, | 437 DownloadItem::TARGET_DISPOSITION_OVERWRITE, |
428 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, | 438 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, |
429 intermediate_path); | 439 intermediate_path); |
430 RunAllPendingInMessageLoops(); | 440 RunAllPendingInMessageLoops(); |
431 EXPECT_EQ(FILE_PATH_LITERAL("foo.bar"), | 441 EXPECT_EQ(FILE_PATH_LITERAL("foo.bar"), |
432 item->GetFileNameToReportUser().value()); | 442 item->GetFileNameToReportUser().value()); |
433 item->SetDisplayName(FilePath(FILE_PATH_LITERAL("new.name"))); | 443 item->SetDisplayName(FilePath(FILE_PATH_LITERAL("new.name"))); |
434 EXPECT_EQ(FILE_PATH_LITERAL("new.name"), | 444 EXPECT_EQ(FILE_PATH_LITERAL("new.name"), |
435 item->GetFileNameToReportUser().value()); | 445 item->GetFileNameToReportUser().value()); |
436 CleanupItem(item, download_file); | |
437 } | |
438 | |
439 // Test to make sure that Start method calls DF initialize properly. | |
440 TEST_F(DownloadItemTest, Start) { | |
441 MockDownloadFile* mock_download_file(new MockDownloadFile); | |
442 scoped_ptr<content::DownloadFile> download_file(mock_download_file); | |
443 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | |
444 EXPECT_CALL(*mock_download_file, Initialize(_)); | |
445 item->Start(download_file.Pass()); | |
446 | |
447 CleanupItem(item, mock_download_file); | |
448 } | 446 } |
449 | 447 |
450 // Test that the delegate is invoked after the download file is renamed. | 448 // Test that the delegate is invoked after the download file is renamed. |
451 // Delegate::DownloadRenamedToIntermediateName() should be invoked when the | 449 // Delegate::DownloadRenamedToIntermediateName() should be invoked when the |
452 // download is renamed to the intermediate name. | 450 // download is renamed to the intermediate name. |
453 // Delegate::DownloadRenamedToFinalName() should be invoked after the final | 451 // Delegate::DownloadRenamedToFinalName() should be invoked after the final |
454 // rename. | 452 // rename. |
455 TEST_F(DownloadItemTest, CallbackAfterRename) { | 453 TEST_F(DownloadItemTest, CallbackAfterRename) { |
456 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 454 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
457 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
458 FilePath final_path(FilePath(kDummyPath).AppendASCII("foo.bar")); | 455 FilePath final_path(FilePath(kDummyPath).AppendASCII("foo.bar")); |
459 FilePath intermediate_path(final_path.InsertBeforeExtensionASCII("x")); | 456 FilePath intermediate_path(final_path.InsertBeforeExtensionASCII("x")); |
460 FilePath new_intermediate_path(final_path.InsertBeforeExtensionASCII("y")); | 457 FilePath new_intermediate_path(final_path.InsertBeforeExtensionASCII("y")); |
461 EXPECT_CALL(*download_file, Rename(intermediate_path, false, _)) | 458 EXPECT_CALL(*mock_file_manager(), |
| 459 RenameDownloadFile(item->GetGlobalId(), |
| 460 intermediate_path, false, _)) |
462 .WillOnce(ScheduleRenameCallback(new_intermediate_path)); | 461 .WillOnce(ScheduleRenameCallback(new_intermediate_path)); |
463 | |
464 // DownloadItemImpl should invoke this callback on the delegate once the | 462 // DownloadItemImpl should invoke this callback on the delegate once the |
465 // download is renamed to the intermediate name. Also check that GetFullPath() | 463 // download is renamed to the intermediate name. Also check that GetFullPath() |
466 // returns the intermediate path at the time of the call. | 464 // returns the intermediate path at the time of the call. |
467 EXPECT_CALL(*mock_delegate(), | 465 EXPECT_CALL(*mock_delegate(), |
468 DownloadRenamedToIntermediateName( | 466 DownloadRenamedToIntermediateName( |
469 AllOf(item, | 467 AllOf(item, |
470 Property(&DownloadItem::GetFullPath, | 468 Property(&DownloadItem::GetFullPath, |
471 new_intermediate_path)))); | 469 new_intermediate_path)))); |
472 item->OnDownloadTargetDetermined(final_path, | 470 item->OnDownloadTargetDetermined(final_path, |
473 DownloadItem::TARGET_DISPOSITION_OVERWRITE, | 471 DownloadItem::TARGET_DISPOSITION_OVERWRITE, |
474 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, | 472 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, |
475 intermediate_path); | 473 intermediate_path); |
476 RunAllPendingInMessageLoops(); | 474 RunAllPendingInMessageLoops(); |
477 // All the callbacks should have happened by now. | 475 // All the callbacks should have happened by now. |
478 ::testing::Mock::VerifyAndClearExpectations(download_file); | 476 ::testing::Mock::VerifyAndClearExpectations(mock_file_manager()); |
479 ::testing::Mock::VerifyAndClearExpectations(mock_delegate()); | 477 ::testing::Mock::VerifyAndClearExpectations(mock_delegate()); |
480 | 478 |
481 item->OnAllDataSaved(""); | 479 item->OnAllDataSaved(10, ""); |
482 EXPECT_CALL(*download_file, Rename(final_path, true, _)) | 480 EXPECT_CALL(*mock_file_manager(), |
| 481 RenameDownloadFile(item->GetGlobalId(), |
| 482 final_path, true, _)) |
483 .WillOnce(ScheduleRenameCallback(final_path)); | 483 .WillOnce(ScheduleRenameCallback(final_path)); |
| 484 EXPECT_CALL(*mock_file_manager(), |
| 485 CompleteDownload(item->GetGlobalId(), _)) |
| 486 .WillOnce(ScheduleCompleteCallback()); |
484 // DownloadItemImpl should invoke this callback on the delegate after the | 487 // DownloadItemImpl should invoke this callback on the delegate after the |
485 // final rename has completed. Also check that GetFullPath() and | 488 // final rename has completed. Also check that GetFullPath() and |
486 // GetTargetFilePath() return the final path at the time of the call. | 489 // GetTargetFilePath() return the final path at the time of the call. |
487 EXPECT_CALL(*mock_delegate(), | 490 EXPECT_CALL(*mock_delegate(), |
488 DownloadRenamedToFinalName( | 491 DownloadRenamedToFinalName( |
489 AllOf(item, | 492 AllOf(item, |
490 Property(&DownloadItem::GetFullPath, final_path), | 493 Property(&DownloadItem::GetFullPath, final_path), |
491 Property(&DownloadItem::GetTargetFilePath, | 494 Property(&DownloadItem::GetTargetFilePath, |
492 final_path)))); | 495 final_path)))); |
493 EXPECT_CALL(*mock_delegate(), DownloadCompleted(item)); | 496 EXPECT_CALL(*mock_delegate(), DownloadCompleted(item)); |
494 EXPECT_CALL(*mock_delegate(), ShouldOpenDownload(item)) | 497 EXPECT_CALL(*mock_delegate(), ShouldOpenDownload(item)) |
495 .WillOnce(Return(true)); | 498 .WillOnce(Return(true)); |
496 EXPECT_CALL(*download_file, Detach()); | |
497 item->OnDownloadCompleting(); | 499 item->OnDownloadCompleting(); |
498 RunAllPendingInMessageLoops(); | 500 RunAllPendingInMessageLoops(); |
499 ::testing::Mock::VerifyAndClearExpectations(download_file); | 501 ::testing::Mock::VerifyAndClearExpectations(mock_file_manager()); |
500 ::testing::Mock::VerifyAndClearExpectations(mock_delegate()); | 502 ::testing::Mock::VerifyAndClearExpectations(mock_delegate()); |
501 } | 503 } |
502 | 504 |
503 TEST_F(DownloadItemTest, Interrupted) { | 505 TEST_F(DownloadItemTest, Interrupted) { |
504 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 506 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
505 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
506 | 507 |
507 const content::DownloadInterruptReason reason( | 508 const content::DownloadInterruptReason reason( |
508 content::DOWNLOAD_INTERRUPT_REASON_FILE_ACCESS_DENIED); | 509 content::DOWNLOAD_INTERRUPT_REASON_FILE_ACCESS_DENIED); |
509 | 510 |
510 // Confirm interrupt sets state properly. | 511 // Confirm interrupt sets state properly. |
511 EXPECT_CALL(*download_file, Cancel()); | |
512 item->Interrupt(reason); | 512 item->Interrupt(reason); |
513 RunAllPendingInMessageLoops(); | |
514 EXPECT_EQ(DownloadItem::INTERRUPTED, item->GetState()); | 513 EXPECT_EQ(DownloadItem::INTERRUPTED, item->GetState()); |
515 EXPECT_EQ(reason, item->GetLastReason()); | 514 EXPECT_EQ(reason, item->GetLastReason()); |
516 | 515 |
517 // Cancel should result in no change. | 516 // Cancel should result in no change. |
518 item->Cancel(true); | 517 item->Cancel(true); |
519 EXPECT_EQ(DownloadItem::INTERRUPTED, item->GetState()); | 518 EXPECT_EQ(DownloadItem::INTERRUPTED, item->GetState()); |
520 EXPECT_EQ(content::DOWNLOAD_INTERRUPT_REASON_USER_CANCELED, | 519 EXPECT_EQ(content::DOWNLOAD_INTERRUPT_REASON_USER_CANCELED, |
521 item->GetLastReason()); | 520 item->GetLastReason()); |
522 } | 521 } |
523 | 522 |
524 TEST_F(DownloadItemTest, Canceled) { | 523 TEST_F(DownloadItemTest, Canceled) { |
525 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 524 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
526 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
527 | 525 |
528 // Confirm cancel sets state properly. | 526 // Confirm cancel sets state properly. |
529 EXPECT_CALL(*mock_delegate(), DownloadStopped(item)); | 527 EXPECT_CALL(*mock_delegate(), DownloadStopped(item)); |
530 EXPECT_CALL(*download_file, Cancel()); | |
531 item->Cancel(true); | 528 item->Cancel(true); |
532 EXPECT_EQ(DownloadItem::CANCELLED, item->GetState()); | 529 EXPECT_EQ(DownloadItem::CANCELLED, item->GetState()); |
533 RunAllPendingInMessageLoops(); | |
534 } | 530 } |
535 | 531 |
536 TEST_F(DownloadItemTest, FileRemoved) { | 532 TEST_F(DownloadItemTest, FileRemoved) { |
537 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | 533 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); |
538 | 534 |
539 EXPECT_FALSE(item->GetFileExternallyRemoved()); | 535 EXPECT_FALSE(item->GetFileExternallyRemoved()); |
540 item->OnDownloadedFileRemoved(); | 536 item->OnDownloadedFileRemoved(); |
541 EXPECT_TRUE(item->GetFileExternallyRemoved()); | 537 EXPECT_TRUE(item->GetFileExternallyRemoved()); |
542 } | 538 } |
543 | 539 |
544 TEST_F(DownloadItemTest, DestinationUpdate) { | |
545 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | |
546 base::WeakPtr<content::DownloadDestinationObserver> as_observer( | |
547 item->DestinationObserverAsWeakPtr()); | |
548 MockObserver observer(item); | |
549 | |
550 EXPECT_EQ(0l, item->CurrentSpeed()); | |
551 EXPECT_EQ("", item->GetHashState()); | |
552 EXPECT_EQ(0l, item->GetReceivedBytes()); | |
553 EXPECT_EQ(0l, item->GetTotalBytes()); | |
554 EXPECT_FALSE(observer.CheckUpdated()); | |
555 item->SetTotalBytes(100l); | |
556 EXPECT_EQ(100l, item->GetTotalBytes()); | |
557 | |
558 as_observer->DestinationUpdate(10, 20, "deadbeef"); | |
559 EXPECT_EQ(20l, item->CurrentSpeed()); | |
560 EXPECT_EQ("deadbeef", item->GetHashState()); | |
561 EXPECT_EQ(10l, item->GetReceivedBytes()); | |
562 EXPECT_EQ(100l, item->GetTotalBytes()); | |
563 EXPECT_TRUE(observer.CheckUpdated()); | |
564 | |
565 as_observer->DestinationUpdate(200, 20, "livebeef"); | |
566 EXPECT_EQ(20l, item->CurrentSpeed()); | |
567 EXPECT_EQ("livebeef", item->GetHashState()); | |
568 EXPECT_EQ(200l, item->GetReceivedBytes()); | |
569 EXPECT_EQ(0l, item->GetTotalBytes()); | |
570 EXPECT_TRUE(observer.CheckUpdated()); | |
571 } | |
572 | |
573 TEST_F(DownloadItemTest, DestinationError) { | |
574 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | |
575 MockDownloadFile* download_file = AddDownloadFileToDownloadItem(item); | |
576 base::WeakPtr<content::DownloadDestinationObserver> as_observer( | |
577 item->DestinationObserverAsWeakPtr()); | |
578 MockObserver observer(item); | |
579 | |
580 EXPECT_EQ(content::DownloadItem::IN_PROGRESS, item->GetState()); | |
581 EXPECT_EQ(content::DOWNLOAD_INTERRUPT_REASON_NONE, item->GetLastReason()); | |
582 EXPECT_FALSE(observer.CheckUpdated()); | |
583 | |
584 EXPECT_CALL(*mock_delegate(), DownloadStopped(item)); | |
585 EXPECT_CALL(*download_file, Cancel()); | |
586 as_observer->DestinationError( | |
587 content::DOWNLOAD_INTERRUPT_REASON_FILE_ACCESS_DENIED); | |
588 ::testing::Mock::VerifyAndClearExpectations(mock_delegate()); | |
589 EXPECT_TRUE(observer.CheckUpdated()); | |
590 EXPECT_EQ(content::DownloadItem::INTERRUPTED, item->GetState()); | |
591 EXPECT_EQ(content::DOWNLOAD_INTERRUPT_REASON_FILE_ACCESS_DENIED, | |
592 item->GetLastReason()); | |
593 RunAllPendingInMessageLoops(); | |
594 } | |
595 | |
596 TEST_F(DownloadItemTest, DestinationCompleted) { | |
597 DownloadItemImpl* item = CreateDownloadItem(DownloadItem::IN_PROGRESS); | |
598 base::WeakPtr<content::DownloadDestinationObserver> as_observer( | |
599 item->DestinationObserverAsWeakPtr()); | |
600 MockObserver observer(item); | |
601 | |
602 EXPECT_EQ(content::DownloadItem::IN_PROGRESS, item->GetState()); | |
603 EXPECT_EQ("", item->GetHash()); | |
604 EXPECT_EQ("", item->GetHashState()); | |
605 EXPECT_FALSE(item->AllDataSaved()); | |
606 EXPECT_FALSE(observer.CheckUpdated()); | |
607 | |
608 as_observer->DestinationUpdate(10, 20, "deadbeef"); | |
609 EXPECT_TRUE(observer.CheckUpdated()); | |
610 EXPECT_FALSE(observer.CheckUpdated()); // Confirm reset. | |
611 EXPECT_EQ(content::DownloadItem::IN_PROGRESS, item->GetState()); | |
612 EXPECT_EQ("", item->GetHash()); | |
613 EXPECT_EQ("deadbeef", item->GetHashState()); | |
614 EXPECT_FALSE(item->AllDataSaved()); | |
615 | |
616 EXPECT_CALL(*mock_delegate(), MaybeCompleteDownload(item)); | |
617 as_observer->DestinationCompleted("livebeef"); | |
618 ::testing::Mock::VerifyAndClearExpectations(mock_delegate()); | |
619 EXPECT_EQ(content::DownloadItem::IN_PROGRESS, item->GetState()); | |
620 EXPECT_TRUE(observer.CheckUpdated()); | |
621 EXPECT_EQ("livebeef", item->GetHash()); | |
622 EXPECT_EQ("", item->GetHashState()); | |
623 EXPECT_TRUE(item->AllDataSaved()); | |
624 } | |
625 | |
626 TEST(MockDownloadItem, Compiles) { | 540 TEST(MockDownloadItem, Compiles) { |
627 MockDownloadItem mock_item; | 541 MockDownloadItem mock_item; |
628 } | 542 } |
629 | |
OLD | NEW |