| 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/background/request_queue.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/test/test_simple_task_runner.h" | 10 #include "base/test/test_simple_task_runner.h" |
| (...skipping 241 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 252 TEST_F(RequestQueueTest, UpdateRequest) { | 252 TEST_F(RequestQueueTest, UpdateRequest) { |
| 253 // First add a request. Retry count will be set to 0. | 253 // First add a request. Retry count will be set to 0. |
| 254 base::Time creation_time = base::Time::Now(); | 254 base::Time creation_time = base::Time::Now(); |
| 255 SavePageRequest request( | 255 SavePageRequest request( |
| 256 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 256 kRequestId, kUrl, kClientId, creation_time, kUserRequested); |
| 257 queue()->AddRequest(request, base::Bind(&RequestQueueTest::AddRequestDone, | 257 queue()->AddRequest(request, base::Bind(&RequestQueueTest::AddRequestDone, |
| 258 base::Unretained(this))); | 258 base::Unretained(this))); |
| 259 PumpLoop(); | 259 PumpLoop(); |
| 260 | 260 |
| 261 // Update the request, ensure it succeeded. | 261 // Update the request, ensure it succeeded. |
| 262 request.set_attempt_count(kRetryCount); | 262 request.set_completed_attempt_count(kRetryCount); |
| 263 queue()->UpdateRequest( | 263 queue()->UpdateRequest( |
| 264 request, | 264 request, |
| 265 base::Bind(&RequestQueueTest::UpdateRequestDone, base::Unretained(this))); | 265 base::Bind(&RequestQueueTest::UpdateRequestDone, base::Unretained(this))); |
| 266 PumpLoop(); | 266 PumpLoop(); |
| 267 ASSERT_EQ(UpdateRequestResult::SUCCESS, last_update_result()); | 267 ASSERT_EQ(UpdateRequestResult::SUCCESS, last_update_result()); |
| 268 | 268 |
| 269 // Get the request, and verify the update took effect. | 269 // Get the request, and verify the update took effect. |
| 270 queue()->GetRequests( | 270 queue()->GetRequests( |
| 271 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); | 271 base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this))); |
| 272 PumpLoop(); | 272 PumpLoop(); |
| 273 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); | 273 ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result()); |
| 274 ASSERT_EQ(1ul, last_requests().size()); | 274 ASSERT_EQ(1ul, last_requests().size()); |
| 275 ASSERT_EQ(kRetryCount, last_requests().front().attempt_count()); | 275 ASSERT_EQ(kRetryCount, last_requests().front().completed_attempt_count()); |
| 276 } | 276 } |
| 277 | 277 |
| 278 TEST_F(RequestQueueTest, UpdateRequestNotPresent) { | 278 TEST_F(RequestQueueTest, UpdateRequestNotPresent) { |
| 279 // First add a request. Retry count will be set to 0. | 279 // First add a request. Retry count will be set to 0. |
| 280 base::Time creation_time = base::Time::Now(); | 280 base::Time creation_time = base::Time::Now(); |
| 281 SavePageRequest request1( | 281 SavePageRequest request1( |
| 282 kRequestId, kUrl, kClientId, creation_time, kUserRequested); | 282 kRequestId, kUrl, kClientId, creation_time, kUserRequested); |
| 283 SavePageRequest request2( | 283 SavePageRequest request2( |
| 284 kRequestId2, kUrl2, kClientId2, creation_time, kUserRequested); | 284 kRequestId2, kUrl2, kClientId2, creation_time, kUserRequested); |
| 285 queue()->AddRequest(request2, base::Bind(&RequestQueueTest::AddRequestDone, | 285 queue()->AddRequest(request2, base::Bind(&RequestQueueTest::AddRequestDone, |
| 286 base::Unretained(this))); | 286 base::Unretained(this))); |
| 287 PumpLoop(); | 287 PumpLoop(); |
| 288 | 288 |
| 289 // Try to update request1 when only request2 is in the queue. | 289 // Try to update request1 when only request2 is in the queue. |
| 290 queue()->UpdateRequest( | 290 queue()->UpdateRequest( |
| 291 request1, | 291 request1, |
| 292 base::Bind(&RequestQueueTest::UpdateRequestDone, base::Unretained(this))); | 292 base::Bind(&RequestQueueTest::UpdateRequestDone, base::Unretained(this))); |
| 293 PumpLoop(); | 293 PumpLoop(); |
| 294 ASSERT_EQ(UpdateRequestResult::REQUEST_DOES_NOT_EXIST, last_update_result()); | 294 ASSERT_EQ(UpdateRequestResult::REQUEST_DOES_NOT_EXIST, last_update_result()); |
| 295 } | 295 } |
| 296 | 296 |
| 297 } // namespace offline_pages | 297 } // namespace offline_pages |
| OLD | NEW |