| Index: components/offline_pages/background/request_coordinator_unittest.cc
|
| diff --git a/components/offline_pages/background/request_coordinator_unittest.cc b/components/offline_pages/background/request_coordinator_unittest.cc
|
| index 41393a47f6dd97cef81d1234cae0b53e2d9501b4..adec21875cf2c1ee9c45ea9ddfd01737b6a3a595 100644
|
| --- a/components/offline_pages/background/request_coordinator_unittest.cc
|
| +++ b/components/offline_pages/background/request_coordinator_unittest.cc
|
| @@ -214,6 +214,9 @@ class RequestCoordinatorTest
|
| void GetRequestsDone(RequestQueue::GetRequestsResult result,
|
| const std::vector<SavePageRequest>& requests);
|
|
|
| + // Callback for removing requests.
|
| + void RemoveRequestsDone(const std::vector<SavePageRequest>& requests);
|
| +
|
| // Callback for getting request statuses.
|
| void GetQueuedRequestsDone(const std::vector<SavePageRequest>& requests);
|
|
|
| @@ -304,6 +307,12 @@ void RequestCoordinatorTest::GetRequestsDone(
|
| last_requests_ = requests;
|
| }
|
|
|
| +void RequestCoordinatorTest::RemoveRequestsDone(
|
| + const std::vector<SavePageRequest>& requests) {
|
| + last_requests_ = requests;
|
| + waiter_.Signal();
|
| +}
|
| +
|
| void RequestCoordinatorTest::GetQueuedRequestsDone(
|
| const std::vector<SavePageRequest>& requests) {
|
| last_requests_ = requests;
|
| @@ -755,7 +764,7 @@ TEST_F(RequestCoordinatorTest, PauseAndResumeObserver) {
|
| EXPECT_EQ(SavePageRequest::RequestState::AVAILABLE, observer().state());
|
| }
|
|
|
| -TEST_F(RequestCoordinatorTest, ObserverdRemoveRequest) {
|
| +TEST_F(RequestCoordinatorTest, RemoveRequest) {
|
| // Add a request to the queue.
|
| offline_pages::SavePageRequest request1(kRequestId1, kUrl1, kClientId1,
|
| base::Time::Now(), kUserRequested);
|
| @@ -767,12 +776,19 @@ TEST_F(RequestCoordinatorTest, ObserverdRemoveRequest) {
|
| // Remove the request.
|
| std::vector<int64_t> request_ids;
|
| request_ids.push_back(kRequestId1);
|
| - coordinator()->RemoveRequests(request_ids);
|
| + coordinator()->RemoveRequests(
|
| + request_ids, base::Bind(&RequestCoordinatorTest::RemoveRequestsDone,
|
| + base::Unretained(this)));
|
| +
|
| + PumpLoop();
|
| + WaitForCallback();
|
| PumpLoop();
|
|
|
| EXPECT_TRUE(observer().completed_called());
|
| EXPECT_EQ(RequestCoordinator::SavePageStatus::REMOVED,
|
| observer().last_status());
|
| + EXPECT_EQ(1UL, last_requests().size());
|
| + EXPECT_EQ(kRequestId1, last_requests().at(0).request_id());
|
| }
|
|
|
| } // namespace offline_pages
|
|
|