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.h" | 5 #include "components/offline_pages/core/background/request_queue.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 #include <utility> | 8 #include <utility> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
11 #include "base/test/test_simple_task_runner.h" | 11 #include "base/test/test_simple_task_runner.h" |
12 #include "base/threading/thread_task_runner_handle.h" | 12 #include "base/threading/thread_task_runner_handle.h" |
13 #include "components/offline_pages/background/device_conditions.h" | 13 #include "components/offline_pages/core/background/device_conditions.h" |
14 #include "components/offline_pages/background/offliner_policy.h" | 14 #include "components/offline_pages/core/background/offliner_policy.h" |
15 #include "components/offline_pages/background/request_coordinator.h" | 15 #include "components/offline_pages/core/background/request_coordinator.h" |
16 #include "components/offline_pages/background/request_coordinator_event_logger.h
" | 16 #include "components/offline_pages/core/background/request_coordinator_event_log
ger.h" |
17 #include "components/offline_pages/background/request_notifier.h" | 17 #include "components/offline_pages/core/background/request_notifier.h" |
18 #include "components/offline_pages/background/request_queue_in_memory_store.h" | 18 #include "components/offline_pages/core/background/request_queue_in_memory_store
.h" |
19 #include "components/offline_pages/background/save_page_request.h" | 19 #include "components/offline_pages/core/background/save_page_request.h" |
20 #include "testing/gtest/include/gtest/gtest.h" | 20 #include "testing/gtest/include/gtest/gtest.h" |
21 | 21 |
22 namespace offline_pages { | 22 namespace offline_pages { |
23 | 23 |
24 using AddRequestResult = AddRequestResult; | 24 using AddRequestResult = AddRequestResult; |
25 using GetRequestsResult = GetRequestsResult; | 25 using GetRequestsResult = GetRequestsResult; |
26 using UpdateRequestResult = UpdateRequestResult; | 26 using UpdateRequestResult = UpdateRequestResult; |
27 | 27 |
28 namespace { | 28 namespace { |
29 // Data for request 1. | 29 // Data for request 1. |
(...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
98 std::vector<std::unique_ptr<SavePageRequest>> requests); | 98 std::vector<std::unique_ptr<SavePageRequest>> requests); |
99 | 99 |
100 void UpdateRequestDone(UpdateRequestResult result); | 100 void UpdateRequestDone(UpdateRequestResult result); |
101 void UpdateRequestsDone(std::unique_ptr<UpdateRequestsResult> result); | 101 void UpdateRequestsDone(std::unique_ptr<UpdateRequestsResult> result); |
102 | 102 |
103 void ClearResults(); | 103 void ClearResults(); |
104 | 104 |
105 RequestQueue* queue() { return queue_.get(); } | 105 RequestQueue* queue() { return queue_.get(); } |
106 | 106 |
107 AddRequestResult last_add_result() const { return last_add_result_; } | 107 AddRequestResult last_add_result() const { return last_add_result_; } |
108 SavePageRequest* last_added_request() { | 108 SavePageRequest* last_added_request() { return last_added_request_.get(); } |
109 return last_added_request_.get(); | |
110 } | |
111 | 109 |
112 UpdateRequestResult last_update_result() const { return last_update_result_; } | 110 UpdateRequestResult last_update_result() const { return last_update_result_; } |
113 | 111 |
114 GetRequestsResult last_get_requests_result() const { | 112 GetRequestsResult last_get_requests_result() const { |
115 return last_get_requests_result_; | 113 return last_get_requests_result_; |
116 } | 114 } |
117 | 115 |
118 const std::vector<std::unique_ptr<SavePageRequest>>& last_requests() const { | 116 const std::vector<std::unique_ptr<SavePageRequest>>& last_requests() const { |
119 return last_requests_; | 117 return last_requests_; |
120 } | 118 } |
(...skipping 73 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
194 TEST_F(RequestQueueTest, GetRequestsEmpty) { | 192 TEST_F(RequestQueueTest, GetRequestsEmpty) { |
195 queue()->GetRequests( | 193 queue()->GetRequests( |
196 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); | 194 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); |
197 PumpLoop(); | 195 PumpLoop(); |
198 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); | 196 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); |
199 ASSERT_EQ(0ul, last_requests().size()); | 197 ASSERT_EQ(0ul, last_requests().size()); |
200 } | 198 } |
201 | 199 |
202 TEST_F(RequestQueueTest, AddRequest) { | 200 TEST_F(RequestQueueTest, AddRequest) { |
203 base::Time creation_time = base::Time::Now(); | 201 base::Time creation_time = base::Time::Now(); |
204 SavePageRequest request( | 202 SavePageRequest request(kRequestId, kUrl, kClientId, creation_time, |
205 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 203 kUserRequested); |
206 queue()->AddRequest(request, base::Bind(&RequestQueueTest::AddRequestDone, | 204 queue()->AddRequest(request, base::Bind(&RequestQueueTest::AddRequestDone, |
207 base::Unretained(this))); | 205 base::Unretained(this))); |
208 PumpLoop(); | 206 PumpLoop(); |
209 ASSERT_EQ(AddRequestResult::SUCCESS, last_add_result()); | 207 ASSERT_EQ(AddRequestResult::SUCCESS, last_add_result()); |
210 ASSERT_TRUE(last_added_request()); | 208 ASSERT_TRUE(last_added_request()); |
211 ASSERT_EQ(kRequestId, last_added_request()->request_id()); | 209 ASSERT_EQ(kRequestId, last_added_request()->request_id()); |
212 | 210 |
213 queue()->GetRequests( | 211 queue()->GetRequests( |
214 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); | 212 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); |
215 PumpLoop(); | 213 PumpLoop(); |
216 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); | 214 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); |
217 ASSERT_EQ(1ul, last_requests().size()); | 215 ASSERT_EQ(1ul, last_requests().size()); |
218 } | 216 } |
219 | 217 |
220 TEST_F(RequestQueueTest, RemoveRequest) { | 218 TEST_F(RequestQueueTest, RemoveRequest) { |
221 base::Time creation_time = base::Time::Now(); | 219 base::Time creation_time = base::Time::Now(); |
222 SavePageRequest request( | 220 SavePageRequest request(kRequestId, kUrl, kClientId, creation_time, |
223 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 221 kUserRequested); |
224 queue()->AddRequest(request, base::Bind(&RequestQueueTest::AddRequestDone, | 222 queue()->AddRequest(request, base::Bind(&RequestQueueTest::AddRequestDone, |
225 base::Unretained(this))); | 223 base::Unretained(this))); |
226 PumpLoop(); | 224 PumpLoop(); |
227 ASSERT_EQ(kRequestId, last_added_request()->request_id()); | 225 ASSERT_EQ(kRequestId, last_added_request()->request_id()); |
228 | 226 |
229 std::vector<int64_t> remove_requests{kRequestId}; | 227 std::vector<int64_t> remove_requests{kRequestId}; |
230 queue()->RemoveRequests(remove_requests, | 228 queue()->RemoveRequests(remove_requests, |
231 base::Bind(&RequestQueueTest::UpdateRequestsDone, | 229 base::Bind(&RequestQueueTest::UpdateRequestsDone, |
232 base::Unretained(this))); | 230 base::Unretained(this))); |
233 PumpLoop(); | 231 PumpLoop(); |
(...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
356 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); | 354 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); |
357 ASSERT_EQ(1ul, last_requests().size()); | 355 ASSERT_EQ(1ul, last_requests().size()); |
358 ASSERT_EQ(SavePageRequest::RequestState::AVAILABLE, | 356 ASSERT_EQ(SavePageRequest::RequestState::AVAILABLE, |
359 last_requests().at(0)->request_state()); | 357 last_requests().at(0)->request_state()); |
360 } | 358 } |
361 | 359 |
362 // A longer test populating the request queue with more than one item, properly | 360 // A longer test populating the request queue with more than one item, properly |
363 // listing multiple items and removing the right item. | 361 // listing multiple items and removing the right item. |
364 TEST_F(RequestQueueTest, MultipleRequestsAddGetRemove) { | 362 TEST_F(RequestQueueTest, MultipleRequestsAddGetRemove) { |
365 base::Time creation_time = base::Time::Now(); | 363 base::Time creation_time = base::Time::Now(); |
366 SavePageRequest request1( | 364 SavePageRequest request1(kRequestId, kUrl, kClientId, creation_time, |
367 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 365 kUserRequested); |
368 queue()->AddRequest(request1, base::Bind(&RequestQueueTest::AddRequestDone, | 366 queue()->AddRequest(request1, base::Bind(&RequestQueueTest::AddRequestDone, |
369 base::Unretained(this))); | 367 base::Unretained(this))); |
370 PumpLoop(); | 368 PumpLoop(); |
371 ASSERT_EQ(request1.request_id(), last_added_request()->request_id()); | 369 ASSERT_EQ(request1.request_id(), last_added_request()->request_id()); |
372 SavePageRequest request2( | 370 SavePageRequest request2(kRequestId2, kUrl2, kClientId2, creation_time, |
373 kRequestId2, kUrl2, kClientId2, creation_time, kUserRequested); | 371 kUserRequested); |
374 queue()->AddRequest(request2, base::Bind(&RequestQueueTest::AddRequestDone, | 372 queue()->AddRequest(request2, base::Bind(&RequestQueueTest::AddRequestDone, |
375 base::Unretained(this))); | 373 base::Unretained(this))); |
376 PumpLoop(); | 374 PumpLoop(); |
377 ASSERT_EQ(request2.request_id(), last_added_request()->request_id()); | 375 ASSERT_EQ(request2.request_id(), last_added_request()->request_id()); |
378 | 376 |
379 queue()->GetRequests( | 377 queue()->GetRequests( |
380 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); | 378 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); |
381 PumpLoop(); | 379 PumpLoop(); |
382 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); | 380 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); |
383 ASSERT_EQ(2ul, last_requests().size()); | 381 ASSERT_EQ(2ul, last_requests().size()); |
(...skipping 196 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
580 // Doing a get should show no entries left in the queue since the expired | 578 // Doing a get should show no entries left in the queue since the expired |
581 // request has been removed. | 579 // request has been removed. |
582 queue()->GetRequests( | 580 queue()->GetRequests( |
583 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); | 581 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); |
584 this->PumpLoop(); | 582 this->PumpLoop(); |
585 ASSERT_EQ(GetRequestsResult::SUCCESS, this->last_get_requests_result()); | 583 ASSERT_EQ(GetRequestsResult::SUCCESS, this->last_get_requests_result()); |
586 ASSERT_TRUE(this->last_requests().empty()); | 584 ASSERT_TRUE(this->last_requests().empty()); |
587 } | 585 } |
588 | 586 |
589 } // namespace offline_pages | 587 } // namespace offline_pages |
OLD | NEW |