| Index: components/offline_pages/background/request_queue_unittest.cc
|
| diff --git a/components/offline_pages/background/request_queue_unittest.cc b/components/offline_pages/background/request_queue_unittest.cc
|
| index 967257c6d83c047573e7cac7a124d32bf3fc7125..ffc3857f60615e42caada2651d0e1f37b5995d3a 100644
|
| --- a/components/offline_pages/background/request_queue_unittest.cc
|
| +++ b/components/offline_pages/background/request_queue_unittest.cc
|
| @@ -175,6 +175,39 @@ TEST_F(RequestQueueTest, RemoveRequest) {
|
| ASSERT_EQ(0ul, last_requests().size());
|
| }
|
|
|
| +TEST_F(RequestQueueTest, RemoveRequestByIds) {
|
| + base::Time creation_time = base::Time::Now();
|
| + SavePageRequest request(kRequestId, kUrl, kClientId, creation_time,
|
| + kUserRequested);
|
| + queue()->AddRequest(request, base::Bind(&RequestQueueTest::AddRequestDone,
|
| + base::Unretained(this)));
|
| + PumpLoop();
|
| +
|
| + SavePageRequest request2(kRequestId2, kUrl2, kClientId2, creation_time,
|
| + kUserRequested);
|
| + queue()->AddRequest(request2, base::Bind(&RequestQueueTest::AddRequestDone,
|
| + base::Unretained(this)));
|
| + PumpLoop();
|
| +
|
| + queue()->GetRequests(
|
| + base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this)));
|
| + PumpLoop();
|
| + ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result());
|
| + ASSERT_EQ(2ul, last_requests().size());
|
| +
|
| + queue()->RemoveRequestsById(
|
| + std::vector<int64_t>({kRequestId, kRequestId2}),
|
| + base::Bind(&RequestQueueTest::RemoveRequestDone, base::Unretained(this)));
|
| + PumpLoop();
|
| + ASSERT_EQ(UpdateRequestResult::SUCCESS, last_remove_result());
|
| +
|
| + queue()->GetRequests(
|
| + base::Bind(&RequestQueueTest::GetRequestsDone, base::Unretained(this)));
|
| + PumpLoop();
|
| + ASSERT_EQ(GetRequestsResult::SUCCESS, last_get_requests_result());
|
| + ASSERT_EQ(0ul, last_requests().size());
|
| +}
|
| +
|
| TEST_F(RequestQueueTest, RemoveRequestByClientId) {
|
| base::Time creation_time = base::Time::Now();
|
| // Put two requests on the queue with different client IDs.
|
|
|