Index: components/offline_pages/core/background/request_coordinator_unittest.cc |
diff --git a/components/offline_pages/core/background/request_coordinator_unittest.cc b/components/offline_pages/core/background/request_coordinator_unittest.cc |
index 5e1b1dca3cf3046a76d3378467824ff533015eee..3f02bafb99a3da42af854393d0c81e8746107004 100644 |
--- a/components/offline_pages/core/background/request_coordinator_unittest.cc |
+++ b/components/offline_pages/core/background/request_coordinator_unittest.cc |
@@ -118,6 +118,8 @@ class ObserverStub : public RequestCoordinator::Observer { |
class RequestCoordinatorTest : public testing::Test { |
public: |
+ using RequestCoordinatorState = RequestCoordinator::RequestCoordinatorState; |
+ |
RequestCoordinatorTest(); |
~RequestCoordinatorTest() override; |
@@ -129,9 +131,7 @@ class RequestCoordinatorTest : public testing::Test { |
return coordinator_taco_->request_coordinator(); |
} |
- bool is_busy() { return coordinator()->is_busy(); } |
- |
- bool is_starting() { return coordinator()->is_starting(); } |
+ RequestCoordinatorState state() { return coordinator()->state(); } |
// Test processing callback function. |
void ProcessingCallbackFunction(bool result) { |
@@ -492,7 +492,7 @@ TEST_F(RequestCoordinatorTest, StartScheduledProcessingWithRequestInProgress) { |
processing_callback())); |
PumpLoop(); |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
// Since the offliner is disabled, this callback should not be called. |
EXPECT_FALSE(processing_callback_called()); |
@@ -541,7 +541,7 @@ TEST_F(RequestCoordinatorTest, StartImmediateProcessingWithRequestInProgress) { |
EXPECT_TRUE(coordinator()->StartImmediateProcessing(processing_callback())); |
PumpLoop(); |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
// Since the offliner is disabled, this callback should not be called. |
EXPECT_FALSE(processing_callback_called()); |
@@ -696,13 +696,13 @@ TEST_F(RequestCoordinatorTest, OfflinerDoneRequestSucceededButLostNetwork) { |
EXPECT_TRUE(processing_callback_called()); |
// Verify not busy with 2nd request (since no connection). |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
// Now connect network and verify processing starts. |
SetNetworkConnected(true); |
CallConnectionTypeObserver(); |
PumpLoop(); |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
} |
TEST_F(RequestCoordinatorTest, OfflinerDoneRequestFailed) { |
@@ -728,7 +728,7 @@ TEST_F(RequestCoordinatorTest, OfflinerDoneRequestFailed) { |
EXPECT_FALSE(processing_callback_called()); |
// Busy processing 2nd request. |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
coordinator()->queue()->GetRequests(base::Bind( |
&RequestCoordinatorTest::GetRequestsDone, base::Unretained(this))); |
@@ -770,7 +770,7 @@ TEST_F(RequestCoordinatorTest, OfflinerDoneRequestFailedNoRetryFailure) { |
EXPECT_FALSE(processing_callback_called()); |
// Busy processing 2nd request. |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
coordinator()->queue()->GetRequests(base::Bind( |
&RequestCoordinatorTest::GetRequestsDone, base::Unretained(this))); |
@@ -807,7 +807,7 @@ TEST_F(RequestCoordinatorTest, OfflinerDoneRequestFailedNoNextFailure) { |
EXPECT_TRUE(processing_callback_called()); |
// Not busy for NO_NEXT failure. |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
coordinator()->queue()->GetRequests(base::Bind( |
&RequestCoordinatorTest::GetRequestsDone, base::Unretained(this))); |
@@ -871,13 +871,13 @@ TEST_F(RequestCoordinatorTest, OfflinerDonePrerenderingCancel) { |
TEST_F(RequestCoordinatorTest, RequestNotPickedDisabledItemsRemain) { |
coordinator()->StartScheduledProcessing(device_conditions(), |
processing_callback()); |
- EXPECT_TRUE(is_starting()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::PICKING); |
// Call RequestNotPicked, simulating a request on the disabled list. |
CallRequestNotPicked(false, true); |
PumpLoop(); |
- EXPECT_FALSE(is_starting()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::PICKING); |
// The scheduler should have been called to schedule the disabled task for |
// 5 minutes from now. |
@@ -892,14 +892,14 @@ TEST_F(RequestCoordinatorTest, RequestNotPickedDisabledItemsRemain) { |
TEST_F(RequestCoordinatorTest, RequestNotPickedNonUserRequestedItemsRemain) { |
coordinator()->StartScheduledProcessing(device_conditions(), |
processing_callback()); |
- EXPECT_TRUE(is_starting()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::PICKING); |
// Call RequestNotPicked, and make sure we pick schedule a task for non user |
// requested conditions, with no tasks on the disabled list. |
CallRequestNotPicked(true, false); |
PumpLoop(); |
- EXPECT_FALSE(is_starting()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::PICKING); |
EXPECT_TRUE(processing_callback_called()); |
// The scheduler should have been called to schedule the non-user requested |
@@ -962,7 +962,7 @@ TEST_F(RequestCoordinatorTest, StartScheduledProcessingWithLoadingDisabled) { |
PumpLoop(); |
EXPECT_TRUE(processing_callback_called()); |
- EXPECT_FALSE(is_starting()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::PICKING); |
EXPECT_EQ(Offliner::LOADING_NOT_ACCEPTED, last_offlining_status()); |
} |
@@ -978,7 +978,7 @@ TEST_F(RequestCoordinatorTest, |
EXPECT_TRUE(coordinator()->StartScheduledProcessing(device_conditions(), |
processing_callback())); |
- EXPECT_TRUE(is_starting()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::PICKING); |
// Now, quick, before it can do much (we haven't called PumpLoop), cancel it. |
coordinator()->StopProcessing(Offliner::REQUEST_COORDINATOR_CANCELED); |
@@ -987,7 +987,7 @@ TEST_F(RequestCoordinatorTest, |
PumpLoop(); |
EXPECT_TRUE(processing_callback_called()); |
- EXPECT_FALSE(is_starting()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::PICKING); |
// OfflinerDoneCallback will not end up getting called with status SAVED, |
// since we cancelled the event before it called offliner_->LoadAndSave(). |
@@ -1010,7 +1010,7 @@ TEST_F(RequestCoordinatorTest, |
EXPECT_TRUE(coordinator()->StartScheduledProcessing(device_conditions(), |
processing_callback())); |
- EXPECT_TRUE(is_starting()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::PICKING); |
// Let all the async parts of the start processing pipeline run to completion. |
PumpLoop(); |
@@ -1024,8 +1024,8 @@ TEST_F(RequestCoordinatorTest, |
EXPECT_FALSE(processing_callback_called()); |
// Coordinator should now be busy. |
- EXPECT_TRUE(is_busy()); |
- EXPECT_FALSE(is_starting()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::PICKING); |
// Now we cancel it while the prerenderer is busy. |
coordinator()->StopProcessing(Offliner::REQUEST_COORDINATOR_CANCELED); |
@@ -1038,7 +1038,7 @@ TEST_F(RequestCoordinatorTest, |
EXPECT_EQ(SavePageRequest::RequestState::AVAILABLE, observer().state()); |
observer().Clear(); |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
// OfflinerDoneCallback will not end up getting called with status SAVED, |
// since we cancelled the event before the LoadAndSave completed. |
@@ -1170,8 +1170,8 @@ TEST_F(RequestCoordinatorTest, |
WaitForCallback(); |
PumpLoop(); |
- EXPECT_FALSE(is_starting()); |
- EXPECT_FALSE(coordinator()->is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::PICKING); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
EXPECT_TRUE(OfflinerWasCanceled()); |
} |
@@ -1185,7 +1185,7 @@ TEST_F(RequestCoordinatorTest, |
PumpLoop(); |
// Verify that immediate start from adding the request did happen. |
- EXPECT_TRUE(coordinator()->is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
// Advance the mock clock 1 second before the watchdog timeout. |
AdvanceClockBy(base::TimeDelta::FromSeconds( |
@@ -1196,7 +1196,7 @@ TEST_F(RequestCoordinatorTest, |
PumpLoop(); |
// Verify still busy. |
- EXPECT_TRUE(coordinator()->is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
EXPECT_FALSE(OfflinerWasCanceled()); |
// Advance the mock clock past the watchdog timeout now. |
@@ -1260,7 +1260,7 @@ TEST_F(RequestCoordinatorTest, TryNextRequestWithNoNetwork) { |
EXPECT_TRUE(coordinator()->StartScheduledProcessing(device_conditions(), |
waiting_callback())); |
PumpLoop(); |
- EXPECT_TRUE(coordinator()->is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
// Now lose the network connection. |
SetNetworkConnected(false); |
@@ -1271,8 +1271,8 @@ TEST_F(RequestCoordinatorTest, TryNextRequestWithNoNetwork) { |
PumpLoop(); |
// Not starting nor busy with next request. |
- EXPECT_FALSE(coordinator()->is_starting()); |
- EXPECT_FALSE(coordinator()->is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::PICKING); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
// Get queued requests. |
coordinator()->queue()->GetRequests(base::Bind( |
@@ -1376,7 +1376,7 @@ TEST_F(RequestCoordinatorTest, |
EXPECT_NE(0, SavePageLater()); |
PumpLoop(); |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
} |
TEST_F(RequestCoordinatorTest, |
@@ -1390,7 +1390,7 @@ TEST_F(RequestCoordinatorTest, |
PumpLoop(); |
// Verify not immediately busy (since low-end device). |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
// Set feature flag to allow concurrent loads. |
base::test::ScopedFeatureList scoped_feature_list; |
@@ -1411,7 +1411,7 @@ TEST_F(RequestCoordinatorTest, |
PumpLoop(); |
// Verify immediate processing did start this time. |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
} |
TEST_F(RequestCoordinatorTest, SavePageDoesntStartProcessingWhenDisconnected) { |
@@ -1419,13 +1419,13 @@ TEST_F(RequestCoordinatorTest, SavePageDoesntStartProcessingWhenDisconnected) { |
EnableOfflinerCallback(false); |
EXPECT_NE(0, SavePageLater()); |
PumpLoop(); |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
// Now connect network and verify processing starts. |
SetNetworkConnected(true); |
CallConnectionTypeObserver(); |
PumpLoop(); |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
} |
TEST_F(RequestCoordinatorTest, |
@@ -1443,7 +1443,7 @@ TEST_F(RequestCoordinatorTest, |
EXPECT_NE(0, SavePageLater()); |
PumpLoop(); |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
} |
TEST_F(RequestCoordinatorTest, |
@@ -1458,7 +1458,7 @@ TEST_F(RequestCoordinatorTest, |
// Add a request to the queue. |
AddRequest1(); |
PumpLoop(); |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
// Pause the request. |
std::vector<int64_t> request_ids; |
@@ -1469,7 +1469,7 @@ TEST_F(RequestCoordinatorTest, |
// Resume the request while disconnected. |
coordinator()->ResumeRequests(request_ids); |
PumpLoop(); |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
EXPECT_EQ(1UL, prioritized_requests().size()); |
// Pause the request again. |
@@ -1482,11 +1482,11 @@ TEST_F(RequestCoordinatorTest, |
// Resume the request while connected. |
coordinator()->ResumeRequests(request_ids); |
- EXPECT_FALSE(is_busy()); |
+ EXPECT_FALSE(state() == RequestCoordinatorState::OFFLINING); |
PumpLoop(); |
EXPECT_EQ(1UL, prioritized_requests().size()); |
- EXPECT_TRUE(is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
} |
TEST_F(RequestCoordinatorTest, SnapshotOnLastTryForScheduledProcessing) { |
@@ -1552,7 +1552,7 @@ TEST_F(RequestCoordinatorTest, SnapshotOnLastTryForImmediateProcessing) { |
observer().Clear(); |
// Verify that the request is being processed. |
- EXPECT_TRUE(coordinator()->is_busy()); |
+ EXPECT_TRUE(state() == RequestCoordinatorState::OFFLINING); |
// Advance the mock clock 1 second more than the watchdog timeout. |
AdvanceClockBy(base::TimeDelta::FromSeconds( |