| 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 246 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 274 this->last_update_result()->item_statuses[1].second); | 274 this->last_update_result()->item_statuses[1].second); |
| 275 EXPECT_EQ(1UL, this->last_update_result()->updated_items.size()); | 275 EXPECT_EQ(1UL, this->last_update_result()->updated_items.size()); |
| 276 EXPECT_EQ(request1, this->last_update_result()->updated_items.at(0)); | 276 EXPECT_EQ(request1, this->last_update_result()->updated_items.at(0)); |
| 277 } | 277 } |
| 278 | 278 |
| 279 TYPED_TEST(RequestQueueStoreTest, AddRequest) { | 279 TYPED_TEST(RequestQueueStoreTest, AddRequest) { |
| 280 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); | 280 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); |
| 281 this->InitializeStore(store.get()); | 281 this->InitializeStore(store.get()); |
| 282 | 282 |
| 283 base::Time creation_time = base::Time::Now(); | 283 base::Time creation_time = base::Time::Now(); |
| 284 SavePageRequest request( | 284 SavePageRequest request(kRequestId, kUrl, kClientId, creation_time, |
| 285 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 285 kUserRequested); |
| 286 | 286 |
| 287 store->AddRequest(request, | 287 store->AddRequest(request, |
| 288 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 288 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 289 base::Unretained(this))); | 289 base::Unretained(this))); |
| 290 ASSERT_EQ(ItemActionStatus::NOT_FOUND, this->last_add_status()); | 290 ASSERT_EQ(ItemActionStatus::NOT_FOUND, this->last_add_status()); |
| 291 this->PumpLoop(); | 291 this->PumpLoop(); |
| 292 ASSERT_EQ(ItemActionStatus::SUCCESS, this->last_add_status()); | 292 ASSERT_EQ(ItemActionStatus::SUCCESS, this->last_add_status()); |
| 293 | 293 |
| 294 // Verifying get reqeust results after a request was added. | 294 // Verifying get reqeust results after a request was added. |
| 295 this->ClearResults(); | 295 this->ClearResults(); |
| (...skipping 22 matching lines...) Expand all Loading... |
| 318 this->PumpLoop(); | 318 this->PumpLoop(); |
| 319 ASSERT_EQ(LastResult::RESULT_TRUE, this->last_result()); | 319 ASSERT_EQ(LastResult::RESULT_TRUE, this->last_result()); |
| 320 ASSERT_EQ(1ul, this->last_requests().size()); | 320 ASSERT_EQ(1ul, this->last_requests().size()); |
| 321 } | 321 } |
| 322 | 322 |
| 323 TYPED_TEST(RequestQueueStoreTest, UpdateRequest) { | 323 TYPED_TEST(RequestQueueStoreTest, UpdateRequest) { |
| 324 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); | 324 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); |
| 325 this->InitializeStore(store.get()); | 325 this->InitializeStore(store.get()); |
| 326 | 326 |
| 327 base::Time creation_time = base::Time::Now(); | 327 base::Time creation_time = base::Time::Now(); |
| 328 SavePageRequest original_request( | 328 SavePageRequest original_request(kRequestId, kUrl, kClientId, creation_time, |
| 329 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 329 kUserRequested); |
| 330 store->AddRequest(original_request, | 330 store->AddRequest(original_request, |
| 331 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 331 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 332 base::Unretained(this))); | 332 base::Unretained(this))); |
| 333 this->PumpLoop(); | 333 this->PumpLoop(); |
| 334 this->ClearResults(); | 334 this->ClearResults(); |
| 335 | 335 |
| 336 base::Time new_creation_time = | 336 base::Time new_creation_time = |
| 337 creation_time + base::TimeDelta::FromMinutes(1); | 337 creation_time + base::TimeDelta::FromMinutes(1); |
| 338 base::Time activation_time = creation_time + base::TimeDelta::FromHours(6); | 338 base::Time activation_time = creation_time + base::TimeDelta::FromHours(6); |
| 339 // Try updating an existing request. | 339 // Try updating an existing request. |
| (...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 437 EXPECT_EQ(ItemActionStatus::NOT_FOUND, | 437 EXPECT_EQ(ItemActionStatus::NOT_FOUND, |
| 438 this->last_update_result()->item_statuses[1].second); | 438 this->last_update_result()->item_statuses[1].second); |
| 439 EXPECT_EQ(0UL, this->last_update_result()->updated_items.size()); | 439 EXPECT_EQ(0UL, this->last_update_result()->updated_items.size()); |
| 440 } | 440 } |
| 441 | 441 |
| 442 TYPED_TEST(RequestQueueStoreTest, ResetStore) { | 442 TYPED_TEST(RequestQueueStoreTest, ResetStore) { |
| 443 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); | 443 std::unique_ptr<RequestQueueStore> store(this->BuildStore()); |
| 444 this->InitializeStore(store.get()); | 444 this->InitializeStore(store.get()); |
| 445 | 445 |
| 446 base::Time creation_time = base::Time::Now(); | 446 base::Time creation_time = base::Time::Now(); |
| 447 SavePageRequest original_request( | 447 SavePageRequest original_request(kRequestId, kUrl, kClientId, creation_time, |
| 448 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 448 kUserRequested); |
| 449 store->AddRequest(original_request, | 449 store->AddRequest(original_request, |
| 450 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 450 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 451 base::Unretained(this))); | 451 base::Unretained(this))); |
| 452 this->PumpLoop(); | 452 this->PumpLoop(); |
| 453 this->ClearResults(); | 453 this->ClearResults(); |
| 454 | 454 |
| 455 store->Reset(base::Bind(&RequestQueueStoreTestBase::ResetDone, | 455 store->Reset(base::Bind(&RequestQueueStoreTestBase::ResetDone, |
| 456 base::Unretained(this))); | 456 base::Unretained(this))); |
| 457 ASSERT_EQ(LastResult::RESULT_NONE, this->last_result()); | 457 ASSERT_EQ(LastResult::RESULT_NONE, this->last_result()); |
| 458 this->PumpLoop(); | 458 this->PumpLoop(); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 470 class RequestQueueStoreSQLTest | 470 class RequestQueueStoreSQLTest |
| 471 : public RequestQueueStoreTest<RequestQueueStoreSQLFactory> {}; | 471 : public RequestQueueStoreTest<RequestQueueStoreSQLFactory> {}; |
| 472 | 472 |
| 473 // Makes sure that persistent DB is actually persisting requests across store | 473 // Makes sure that persistent DB is actually persisting requests across store |
| 474 // restarts. | 474 // restarts. |
| 475 TEST_F(RequestQueueStoreSQLTest, SaveCloseReopenRead) { | 475 TEST_F(RequestQueueStoreSQLTest, SaveCloseReopenRead) { |
| 476 std::unique_ptr<RequestQueueStore> store(BuildStore()); | 476 std::unique_ptr<RequestQueueStore> store(BuildStore()); |
| 477 this->InitializeStore(store.get()); | 477 this->InitializeStore(store.get()); |
| 478 | 478 |
| 479 base::Time creation_time = base::Time::Now(); | 479 base::Time creation_time = base::Time::Now(); |
| 480 SavePageRequest original_request( | 480 SavePageRequest original_request(kRequestId, kUrl, kClientId, creation_time, |
| 481 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 481 kUserRequested); |
| 482 store->AddRequest(original_request, | 482 store->AddRequest(original_request, |
| 483 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, | 483 base::Bind(&RequestQueueStoreTestBase::AddRequestDone, |
| 484 base::Unretained(this))); | 484 base::Unretained(this))); |
| 485 PumpLoop(); | 485 PumpLoop(); |
| 486 ClearResults(); | 486 ClearResults(); |
| 487 | 487 |
| 488 // Resets the store, using the same temp directory. The contents should be | 488 // Resets the store, using the same temp directory. The contents should be |
| 489 // intact. First reset is done separately to release DB lock. | 489 // intact. First reset is done separately to release DB lock. |
| 490 store.reset(); | 490 store.reset(); |
| 491 store = BuildStore(); | 491 store = BuildStore(); |
| 492 this->InitializeStore(store.get()); | 492 this->InitializeStore(store.get()); |
| 493 | 493 |
| 494 store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone, | 494 store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone, |
| 495 base::Unretained(this))); | 495 base::Unretained(this))); |
| 496 ASSERT_EQ(LastResult::RESULT_NONE, this->last_result()); | 496 ASSERT_EQ(LastResult::RESULT_NONE, this->last_result()); |
| 497 this->PumpLoop(); | 497 this->PumpLoop(); |
| 498 ASSERT_EQ(LastResult::RESULT_TRUE, this->last_result()); | 498 ASSERT_EQ(LastResult::RESULT_TRUE, this->last_result()); |
| 499 ASSERT_EQ(1ul, this->last_requests().size()); | 499 ASSERT_EQ(1ul, this->last_requests().size()); |
| 500 ASSERT_TRUE(original_request == *(this->last_requests().at(0).get())); | 500 ASSERT_TRUE(original_request == *(this->last_requests().at(0).get())); |
| 501 } | 501 } |
| 502 | 502 |
| 503 } // offline_pages | 503 } // offline_pages |
| OLD | NEW |