| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "components/offline_pages/background/request_queue_store.h" | 5 #include "components/offline_pages/core/background/request_queue_store.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| 11 #include "base/files/scoped_temp_dir.h" | 11 #include "base/files/scoped_temp_dir.h" |
| 12 #include "base/test/test_simple_task_runner.h" | 12 #include "base/test/test_simple_task_runner.h" |
| 13 #include "base/threading/thread_task_runner_handle.h" | 13 #include "base/threading/thread_task_runner_handle.h" |
| 14 #include "components/offline_pages/background/request_queue.h" | 14 #include "components/offline_pages/core/background/request_queue.h" |
| 15 #include "components/offline_pages/background/request_queue_in_memory_store.h" | 15 #include "components/offline_pages/core/background/request_queue_in_memory_store
.h" |
| 16 #include "components/offline_pages/background/request_queue_store_sql.h" | 16 #include "components/offline_pages/core/background/request_queue_store_sql.h" |
| 17 #include "components/offline_pages/background/save_page_request.h" | 17 #include "components/offline_pages/core/background/save_page_request.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
| 19 | 19 |
| 20 namespace offline_pages { | 20 namespace offline_pages { |
| 21 | 21 |
| 22 using UpdateStatus = RequestQueueStore::UpdateStatus; | 22 using UpdateStatus = RequestQueueStore::UpdateStatus; |
| 23 | 23 |
| 24 namespace { | 24 namespace { |
| 25 const int64_t kRequestId = 42; | 25 const int64_t kRequestId = 42; |
| 26 const int64_t kRequestId2 = 44; | 26 const int64_t kRequestId2 = 44; |
| 27 const int64_t kRequestId3 = 47; | 27 const int64_t kRequestId3 = 47; |
| (...skipping 226 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 254 EXPECT_EQ(kRequestId3, this->last_update_result()->item_statuses[1].first); | 254 EXPECT_EQ(kRequestId3, this->last_update_result()->item_statuses[1].first); |
| 255 EXPECT_EQ(ItemActionStatus::NOT_FOUND, | 255 EXPECT_EQ(ItemActionStatus::NOT_FOUND, |
| 256 this->last_update_result()->item_statuses[1].second); | 256 this->last_update_result()->item_statuses[1].second); |
| 257 EXPECT_EQ(1UL, this->last_update_result()->updated_items.size()); | 257 EXPECT_EQ(1UL, this->last_update_result()->updated_items.size()); |
| 258 EXPECT_EQ(request1, this->last_update_result()->updated_items.at(0)); | 258 EXPECT_EQ(request1, this->last_update_result()->updated_items.at(0)); |
| 259 } | 259 } |
| 260 | 260 |
| 261 TYPED_TEST(RequestQueueStoreTest, AddRequest) { | 261 TYPED_TEST(RequestQueueStoreTest, AddRequest) { |
| 262 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); | 262 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); |
| 263 base::Time creation_time = base::Time::Now(); | 263 base::Time creation_time = base::Time::Now(); |
| 264 SavePageRequest request( | 264 SavePageRequest request(kRequestId, kUrl, kClientId, creation_time, |
| 265 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 265 kUserRequested); |
| 266 | 266 |
| 267 store->AddRequest(request, | 267 store->AddRequest(request, |
| 268 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 268 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 269 base::Unretained(this))); | 269 base::Unretained(this))); |
| 270 ASSERT_EQ(ItemActionStatus::NOT_FOUND, this->last_add_status()); | 270 ASSERT_EQ(ItemActionStatus::NOT_FOUND, this->last_add_status()); |
| 271 this->PumpLoop(); | 271 this->PumpLoop(); |
| 272 ASSERT_EQ(ItemActionStatus::SUCCESS, this->last_add_status()); | 272 ASSERT_EQ(ItemActionStatus::SUCCESS, this->last_add_status()); |
| 273 | 273 |
| 274 // Verifying get reqeust results after a request was added. | 274 // Verifying get reqeust results after a request was added. |
| 275 this->ClearResults(); | 275 this->ClearResults(); |
| (...skipping 20 matching lines...) Expand all Loading... |
| 296 base::Unretained(this))); | 296 base::Unretained(this))); |
| 297 ASSERT_EQ(LastResult::kNone, this->last_result()); | 297 ASSERT_EQ(LastResult::kNone, this->last_result()); |
| 298 this->PumpLoop(); | 298 this->PumpLoop(); |
| 299 ASSERT_EQ(LastResult::kTrue, this->last_result()); | 299 ASSERT_EQ(LastResult::kTrue, this->last_result()); |
| 300 ASSERT_EQ(1ul, this->last_requests().size()); | 300 ASSERT_EQ(1ul, this->last_requests().size()); |
| 301 } | 301 } |
| 302 | 302 |
| 303 TYPED_TEST(RequestQueueStoreTest, UpdateRequest) { | 303 TYPED_TEST(RequestQueueStoreTest, UpdateRequest) { |
| 304 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); | 304 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); |
| 305 base::Time creation_time = base::Time::Now(); | 305 base::Time creation_time = base::Time::Now(); |
| 306 SavePageRequest original_request( | 306 SavePageRequest original_request(kRequestId, kUrl, kClientId, creation_time, |
| 307 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 307 kUserRequested); |
| 308 store->AddRequest(original_request, | 308 store->AddRequest(original_request, |
| 309 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 309 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 310 base::Unretained(this))); | 310 base::Unretained(this))); |
| 311 this->PumpLoop(); | 311 this->PumpLoop(); |
| 312 this->ClearResults(); | 312 this->ClearResults(); |
| 313 | 313 |
| 314 base::Time new_creation_time = | 314 base::Time new_creation_time = |
| 315 creation_time + base::TimeDelta::FromMinutes(1); | 315 creation_time + base::TimeDelta::FromMinutes(1); |
| 316 base::Time activation_time = creation_time + base::TimeDelta::FromHours(6); | 316 base::Time activation_time = creation_time + base::TimeDelta::FromHours(6); |
| 317 // Try updating an existing request. | 317 // Try updating an existing request. |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 411 this->last_update_result()->item_statuses[0].second); | 411 this->last_update_result()->item_statuses[0].second); |
| 412 EXPECT_EQ(kRequestId2, this->last_update_result()->item_statuses[1].first); | 412 EXPECT_EQ(kRequestId2, this->last_update_result()->item_statuses[1].first); |
| 413 EXPECT_EQ(ItemActionStatus::NOT_FOUND, | 413 EXPECT_EQ(ItemActionStatus::NOT_FOUND, |
| 414 this->last_update_result()->item_statuses[1].second); | 414 this->last_update_result()->item_statuses[1].second); |
| 415 EXPECT_EQ(0UL, this->last_update_result()->updated_items.size()); | 415 EXPECT_EQ(0UL, this->last_update_result()->updated_items.size()); |
| 416 } | 416 } |
| 417 | 417 |
| 418 TYPED_TEST(RequestQueueStoreTest, ResetStore) { | 418 TYPED_TEST(RequestQueueStoreTest, ResetStore) { |
| 419 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); | 419 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); |
| 420 base::Time creation_time = base::Time::Now(); | 420 base::Time creation_time = base::Time::Now(); |
| 421 SavePageRequest original_request( | 421 SavePageRequest original_request(kRequestId, kUrl, kClientId, creation_time, |
| 422 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 422 kUserRequested); |
| 423 store->AddRequest(original_request, | 423 store->AddRequest(original_request, |
| 424 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 424 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 425 base::Unretained(this))); | 425 base::Unretained(this))); |
| 426 this->PumpLoop(); | 426 this->PumpLoop(); |
| 427 this->ClearResults(); | 427 this->ClearResults(); |
| 428 | 428 |
| 429 store->Reset(base::Bind(&RequestQueueStoreTestBase::ResetDone, | 429 store->Reset(base::Bind(&RequestQueueStoreTestBase::ResetDone, |
| 430 base::Unretained(this))); | 430 base::Unretained(this))); |
| 431 ASSERT_EQ(LastResult::kNone, this->last_result()); | 431 ASSERT_EQ(LastResult::kNone, this->last_result()); |
| 432 this->PumpLoop(); | 432 this->PumpLoop(); |
| 433 ASSERT_EQ(LastResult::kTrue, this->last_result()); | 433 ASSERT_EQ(LastResult::kTrue, this->last_result()); |
| 434 this->ClearResults(); | 434 this->ClearResults(); |
| 435 | 435 |
| 436 store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone, | 436 store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone, |
| 437 base::Unretained(this))); | 437 base::Unretained(this))); |
| 438 this->PumpLoop(); | 438 this->PumpLoop(); |
| 439 ASSERT_EQ(LastResult::kTrue, this->last_result()); | 439 ASSERT_EQ(LastResult::kTrue, this->last_result()); |
| 440 ASSERT_TRUE(this->last_requests().empty()); | 440 ASSERT_TRUE(this->last_requests().empty()); |
| 441 } | 441 } |
| 442 | 442 |
| 443 class RequestQueueStoreSQLTest | 443 class RequestQueueStoreSQLTest |
| 444 : public RequestQueueStoreTest<RequestQueueStoreSQLFactory> {}; | 444 : public RequestQueueStoreTest<RequestQueueStoreSQLFactory> {}; |
| 445 | 445 |
| 446 // Makes sure that persistent DB is actually persisting requests across store | 446 // Makes sure that persistent DB is actually persisting requests across store |
| 447 // restarts. | 447 // restarts. |
| 448 TEST_F(RequestQueueStoreSQLTest, SaveCloseReopenRead) { | 448 TEST_F(RequestQueueStoreSQLTest, SaveCloseReopenRead) { |
| 449 std::unique_ptr<RequestQueueStore> store(BuildStore()); | 449 std::unique_ptr<RequestQueueStore> store(BuildStore()); |
| 450 base::Time creation_time = base::Time::Now(); | 450 base::Time creation_time = base::Time::Now(); |
| 451 SavePageRequest original_request( | 451 SavePageRequest original_request(kRequestId, kUrl, kClientId, creation_time, |
| 452 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 452 kUserRequested); |
| 453 store->AddRequest(original_request, | 453 store->AddRequest(original_request, |
| 454 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 454 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 455 base::Unretained(this))); | 455 base::Unretained(this))); |
| 456 PumpLoop(); | 456 PumpLoop(); |
| 457 ClearResults(); | 457 ClearResults(); |
| 458 | 458 |
| 459 // Resets the store, using the same temp directory. The contents should be | 459 // Resets the store, using the same temp directory. The contents should be |
| 460 // intact. First reset is done separately to release DB lock. | 460 // intact. First reset is done separately to release DB lock. |
| 461 store.reset(); | 461 store.reset(); |
| 462 store = BuildStore(); | 462 store = BuildStore(); |
| 463 store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone, | 463 store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone, |
| 464 base::Unretained(this))); | 464 base::Unretained(this))); |
| 465 ASSERT_EQ(LastResult::kNone, this->last_result()); | 465 ASSERT_EQ(LastResult::kNone, this->last_result()); |
| 466 this->PumpLoop(); | 466 this->PumpLoop(); |
| 467 ASSERT_EQ(LastResult::kTrue, this->last_result()); | 467 ASSERT_EQ(LastResult::kTrue, this->last_result()); |
| 468 ASSERT_EQ(1ul, this->last_requests().size()); | 468 ASSERT_EQ(1ul, this->last_requests().size()); |
| 469 ASSERT_TRUE(original_request == *(this->last_requests().at(0).get())); | 469 ASSERT_TRUE(original_request == *(this->last_requests().at(0).get())); |
| 470 } | 470 } |
| 471 | 471 |
| 472 } // offline_pages | 472 } // offline_pages |
| OLD | NEW |