| 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 // History unit tests come in two flavors: | 5 // History unit tests come in two flavors: |
| 6 // | 6 // |
| 7 // 1. The more complicated style is that the unit test creates a full history | 7 // 1. The more complicated style is that the unit test creates a full history |
| 8 // service. This spawns a background thread for the history backend, and | 8 // service. This spawns a background thread for the history backend, and |
| 9 // all communication is asynchronous. This is useful for testing more | 9 // all communication is asynchronous. This is useful for testing more |
| 10 // complicated things or end-to-end behavior. | 10 // complicated things or end-to-end behavior. |
| (...skipping 305 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 316 Time now = Time::Now(); | 316 Time now = Time::Now(); |
| 317 TimeDelta one_day = TimeDelta::FromDays(1); | 317 TimeDelta one_day = TimeDelta::FromDays(1); |
| 318 Time month_ago = now - TimeDelta::FromDays(30); | 318 Time month_ago = now - TimeDelta::FromDays(30); |
| 319 | 319 |
| 320 // Initially there should be nothing in the downloads database. | 320 // Initially there should be nothing in the downloads database. |
| 321 std::vector<DownloadPersistentStoreInfo> downloads; | 321 std::vector<DownloadPersistentStoreInfo> downloads; |
| 322 db_->QueryDownloads(&downloads); | 322 db_->QueryDownloads(&downloads); |
| 323 EXPECT_EQ(0U, downloads.size()); | 323 EXPECT_EQ(0U, downloads.size()); |
| 324 | 324 |
| 325 // Keep track of these as we need to update them later during the test. | 325 // Keep track of these as we need to update them later during the test. |
| 326 DownloadID in_progress, removing; | 326 DownloadID in_progress; |
| 327 | 327 |
| 328 // Create one with a 0 time. | 328 // Create one with a 0 time. |
| 329 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, Time())); | 329 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, Time())); |
| 330 // Create one for now and +/- 1 day. | 330 // Create one for now and +/- 1 day. |
| 331 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, now - one_day)); | 331 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, now - one_day)); |
| 332 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, now)); | 332 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, now)); |
| 333 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, now + one_day)); | 333 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, now + one_day)); |
| 334 // Try the other four states. | 334 // Try the other four states. |
| 335 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, month_ago)); | 335 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, month_ago)); |
| 336 EXPECT_NE(0, in_progress = AddDownload(DownloadItem::IN_PROGRESS, month_ago)); | 336 EXPECT_NE(0, in_progress = AddDownload(DownloadItem::IN_PROGRESS, month_ago)); |
| 337 EXPECT_NE(0, AddDownload(DownloadItem::CANCELLED, month_ago)); | 337 EXPECT_NE(0, AddDownload(DownloadItem::CANCELLED, month_ago)); |
| 338 EXPECT_NE(0, AddDownload(DownloadItem::INTERRUPTED, month_ago)); | 338 EXPECT_NE(0, AddDownload(DownloadItem::INTERRUPTED, month_ago)); |
| 339 EXPECT_NE(0, removing = AddDownload(DownloadItem::REMOVING, month_ago)); | |
| 340 | 339 |
| 341 // Test to see if inserts worked. | 340 // Test to see if inserts worked. |
| 342 db_->QueryDownloads(&downloads); | 341 db_->QueryDownloads(&downloads); |
| 343 EXPECT_EQ(9U, downloads.size()); | 342 EXPECT_EQ(8U, downloads.size()); |
| 344 | 343 |
| 345 // Try removing from current timestamp. This should delete the one in the | 344 // Try removing from current timestamp. This should delete the one in the |
| 346 // future and one very recent one. | 345 // future and one very recent one. |
| 347 db_->RemoveDownloadsBetween(now, Time()); | 346 db_->RemoveDownloadsBetween(now, Time()); |
| 348 db_->QueryDownloads(&downloads); | 347 db_->QueryDownloads(&downloads); |
| 349 EXPECT_EQ(7U, downloads.size()); | 348 EXPECT_EQ(6U, downloads.size()); |
| 350 | 349 |
| 351 // Try removing from two months ago. This should not delete items that are | 350 // Try removing from two months ago. This should not delete items that are |
| 352 // 'in progress' or in 'removing' state. | 351 // 'in progress' or in 'removing' state. |
| 353 db_->RemoveDownloadsBetween(now - TimeDelta::FromDays(60), Time()); | 352 db_->RemoveDownloadsBetween(now - TimeDelta::FromDays(60), Time()); |
| 354 db_->QueryDownloads(&downloads); | 353 db_->QueryDownloads(&downloads); |
| 355 EXPECT_EQ(3U, downloads.size()); | 354 EXPECT_EQ(2U, downloads.size()); |
| 356 | 355 |
| 357 // Download manager converts to TimeT, which is lossy, so we do the same | 356 // Download manager converts to TimeT, which is lossy, so we do the same |
| 358 // for comparison. | 357 // for comparison. |
| 359 Time month_ago_lossy = Time::FromTimeT(month_ago.ToTimeT()); | 358 Time month_ago_lossy = Time::FromTimeT(month_ago.ToTimeT()); |
| 360 | 359 |
| 361 // Make sure the right values remain. | 360 // Make sure the right values remain. |
| 362 EXPECT_EQ(DownloadItem::COMPLETE, downloads[0].state); | 361 EXPECT_EQ(DownloadItem::COMPLETE, downloads[0].state); |
| 363 EXPECT_EQ(0, downloads[0].start_time.ToInternalValue()); | 362 EXPECT_EQ(0, downloads[0].start_time.ToInternalValue()); |
| 364 EXPECT_EQ(DownloadItem::IN_PROGRESS, downloads[1].state); | 363 EXPECT_EQ(DownloadItem::IN_PROGRESS, downloads[1].state); |
| 365 EXPECT_EQ(month_ago_lossy.ToInternalValue(), | 364 EXPECT_EQ(month_ago_lossy.ToInternalValue(), |
| 366 downloads[1].start_time.ToInternalValue()); | 365 downloads[1].start_time.ToInternalValue()); |
| 367 EXPECT_EQ(DownloadItem::REMOVING, downloads[2].state); | |
| 368 EXPECT_EQ(month_ago_lossy.ToInternalValue(), | 366 EXPECT_EQ(month_ago_lossy.ToInternalValue(), |
| 369 downloads[2].start_time.ToInternalValue()); | 367 downloads[2].start_time.ToInternalValue()); |
| 370 | 368 |
| 371 // Change state so we can delete the downloads. | 369 // Change state so we can delete the downloads. |
| 372 DownloadPersistentStoreInfo data; | 370 DownloadPersistentStoreInfo data; |
| 373 data.received_bytes = 512; | 371 data.received_bytes = 512; |
| 374 data.state = DownloadItem::COMPLETE; | 372 data.state = DownloadItem::COMPLETE; |
| 375 data.end_time = base::Time::Now(); | 373 data.end_time = base::Time::Now(); |
| 376 data.opened = false; | 374 data.opened = false; |
| 377 data.db_handle = in_progress; | 375 data.db_handle = in_progress; |
| 378 EXPECT_TRUE(db_->UpdateDownload(data)); | 376 EXPECT_TRUE(db_->UpdateDownload(data)); |
| 379 data.state = DownloadItem::CANCELLED; | 377 data.state = DownloadItem::CANCELLED; |
| 380 data.db_handle = removing; | |
| 381 EXPECT_TRUE(db_->UpdateDownload(data)); | 378 EXPECT_TRUE(db_->UpdateDownload(data)); |
| 382 | 379 |
| 383 // Try removing from Time=0. This should delete all. | 380 // Try removing from Time=0. This should delete all. |
| 384 db_->RemoveDownloadsBetween(Time(), Time()); | 381 db_->RemoveDownloadsBetween(Time(), Time()); |
| 385 db_->QueryDownloads(&downloads); | 382 db_->QueryDownloads(&downloads); |
| 386 EXPECT_EQ(0U, downloads.size()); | 383 EXPECT_EQ(0U, downloads.size()); |
| 387 | 384 |
| 388 // Check removal of downloads stuck in IN_PROGRESS state. | 385 // Check removal of downloads stuck in IN_PROGRESS state. |
| 389 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, month_ago)); | 386 EXPECT_NE(0, AddDownload(DownloadItem::COMPLETE, month_ago)); |
| 390 EXPECT_NE(0, AddDownload(DownloadItem::IN_PROGRESS, month_ago)); | 387 EXPECT_NE(0, AddDownload(DownloadItem::IN_PROGRESS, month_ago)); |
| (...skipping 565 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 956 history_service_ = history; | 953 history_service_ = history; |
| 957 history->ScheduleDBTask(task.get(), &request_consumer); | 954 history->ScheduleDBTask(task.get(), &request_consumer); |
| 958 request_consumer.CancelAllRequests(); | 955 request_consumer.CancelAllRequests(); |
| 959 CleanupHistoryService(); | 956 CleanupHistoryService(); |
| 960 // WARNING: history has now been deleted. | 957 // WARNING: history has now been deleted. |
| 961 history = NULL; | 958 history = NULL; |
| 962 ASSERT_FALSE(task->done_invoked); | 959 ASSERT_FALSE(task->done_invoked); |
| 963 } | 960 } |
| 964 | 961 |
| 965 } // namespace history | 962 } // namespace history |
| OLD | NEW |