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 |