Index: chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc |
diff --git a/chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc b/chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc |
index 4018ad481a6aa6e0fc26cff7abada67826c04dac..2efe125e4fbd5718abd4a73f062e6bf39f763906 100644 |
--- a/chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc |
+++ b/chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc |
@@ -126,10 +126,14 @@ class BackgroundLoaderOfflinerTest : public testing::Test { |
void SetUp() override; |
TestBackgroundLoaderOffliner* offliner() const { return offliner_.get(); } |
- Offliner::CompletionCallback const callback() { |
+ Offliner::CompletionCallback const completion_callback() { |
return base::Bind(&BackgroundLoaderOfflinerTest::OnCompletion, |
base::Unretained(this)); |
} |
+ Offliner::ProgressCallback const progress_callback() { |
+ return base::Bind(&BackgroundLoaderOfflinerTest::OnProgress, |
+ base::Unretained(this)); |
+ } |
Profile* profile() { return &profile_; } |
bool completion_callback_called() { return completion_callback_called_; } |
Offliner::RequestStatus request_status() { return request_status_; } |
@@ -148,6 +152,7 @@ class BackgroundLoaderOfflinerTest : public testing::Test { |
private: |
void OnCompletion(const SavePageRequest& request, |
Offliner::RequestStatus status); |
+ void OnProgress(const SavePageRequest& request, int64_t bytes); |
content::TestBrowserThreadBundle thread_bundle_; |
TestingProfile profile_; |
std::unique_ptr<TestBackgroundLoaderOffliner> offliner_; |
@@ -180,6 +185,9 @@ void BackgroundLoaderOfflinerTest::OnCompletion( |
request_status_ = status; |
} |
+void BackgroundLoaderOfflinerTest::OnProgress(const SavePageRequest& request, |
+ int64_t bytes) {} |
+ |
TEST_F(BackgroundLoaderOfflinerTest, |
LoadAndSaveBlockThirdPartyCookiesForCustomTabs) { |
base::Time creation_time = base::Time::Now(); |
@@ -188,7 +196,8 @@ TEST_F(BackgroundLoaderOfflinerTest, |
creation_time, kUserRequested); |
profile()->GetPrefs()->SetBoolean(prefs::kBlockThirdPartyCookies, true); |
- EXPECT_FALSE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_FALSE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
histograms().ExpectBucketCount( |
"OfflinePages.Background.CctApiDisableStatus", |
static_cast<int>(OfflinePagesCctApiPrerenderAllowedStatus:: |
@@ -208,7 +217,8 @@ TEST_F(BackgroundLoaderOfflinerTest, |
profile()->GetPrefs()->SetInteger( |
prefs::kNetworkPredictionOptions, |
chrome_browser_net::NETWORK_PREDICTION_NEVER); |
- EXPECT_FALSE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_FALSE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
histograms().ExpectBucketCount( |
"OfflinePages.Background.CctApiDisableStatus", |
static_cast<int>(OfflinePagesCctApiPrerenderAllowedStatus:: |
@@ -225,7 +235,8 @@ TEST_F(BackgroundLoaderOfflinerTest, LoadAndSaveStartsLoading) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
EXPECT_TRUE(offliner()->is_loading()); |
EXPECT_FALSE(SaveInProgress()); |
EXPECT_FALSE(completion_callback_called()); |
@@ -236,7 +247,8 @@ TEST_F(BackgroundLoaderOfflinerTest, CompleteLoadingInitiatesSave) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
CompleteLoading(); |
PumpLoop(); |
EXPECT_FALSE(completion_callback_called()); |
@@ -248,7 +260,8 @@ TEST_F(BackgroundLoaderOfflinerTest, CancelWhenLoading) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
offliner()->Cancel(); |
EXPECT_FALSE(offliner()->is_loading()); // Offliner reset. |
} |
@@ -257,7 +270,8 @@ TEST_F(BackgroundLoaderOfflinerTest, CancelWhenLoaded) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
CompleteLoading(); |
PumpLoop(); |
offliner()->Cancel(); |
@@ -274,7 +288,8 @@ TEST_F(BackgroundLoaderOfflinerTest, LoadedButSaveFails) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
CompleteLoading(); |
PumpLoop(); |
@@ -291,7 +306,8 @@ TEST_F(BackgroundLoaderOfflinerTest, LoadAndSaveSuccess) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
CompleteLoading(); |
PumpLoop(); |
@@ -308,14 +324,16 @@ TEST_F(BackgroundLoaderOfflinerTest, FailsOnInvalidURL) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kFileUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_FALSE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_FALSE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
} |
TEST_F(BackgroundLoaderOfflinerTest, ReturnsOnRenderCrash) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
offliner()->RenderProcessGone( |
base::TerminationStatus::TERMINATION_STATUS_PROCESS_CRASHED); |
@@ -327,7 +345,8 @@ TEST_F(BackgroundLoaderOfflinerTest, ReturnsOnRenderKilled) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
offliner()->RenderProcessGone( |
base::TerminationStatus::TERMINATION_STATUS_PROCESS_WAS_KILLED); |
@@ -339,7 +358,8 @@ TEST_F(BackgroundLoaderOfflinerTest, ReturnsOnWebContentsDestroyed) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
offliner()->WebContentsDestroyed(); |
EXPECT_TRUE(completion_callback_called()); |
@@ -350,7 +370,8 @@ TEST_F(BackgroundLoaderOfflinerTest, FailsOnErrorPage) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
// Create handle with net error code. |
// Called after calling LoadAndSave so we have web_contents to work with. |
@@ -373,7 +394,8 @@ TEST_F(BackgroundLoaderOfflinerTest, OnlySavesOnceOnMultipleLoads) { |
base::Time creation_time = base::Time::Now(); |
SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time, |
kUserRequested); |
- EXPECT_TRUE(offliner()->LoadAndSave(request, callback())); |
+ EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(), |
+ progress_callback())); |
// First load |
CompleteLoading(); |
// Second load |