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 0b4e691c97b8ea63961c5ce48c741295bc61244c..6cc39f27e3a1155b0942afa8e90bd97e3a90d094 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)); |
+ } |
Offliner::CancelCallback const cancel_callback() { |
return base::Bind(&BackgroundLoaderOfflinerTest::OnCancel, |
base::Unretained(this)); |
@@ -153,6 +157,7 @@ class BackgroundLoaderOfflinerTest : public testing::Test { |
private: |
void OnCompletion(const SavePageRequest& request, |
Offliner::RequestStatus status); |
+ void OnProgress(const SavePageRequest& request, int64_t bytes); |
void OnCancel(int64_t offline_id); |
content::TestBrowserThreadBundle thread_bundle_; |
TestingProfile profile_; |
@@ -188,6 +193,9 @@ void BackgroundLoaderOfflinerTest::OnCompletion( |
request_status_ = status; |
} |
+void BackgroundLoaderOfflinerTest::OnProgress(const SavePageRequest& request, |
+ int64_t bytes) {} |
+ |
void BackgroundLoaderOfflinerTest::OnCancel(int64_t offline_id) { |
DCHECK(!cancel_callback_called_); |
cancel_callback_called_ = true; |
@@ -201,7 +209,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:: |
@@ -221,7 +230,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:: |
@@ -238,7 +248,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()); |
@@ -249,7 +260,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()); |
@@ -261,7 +273,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(cancel_callback()); |
PumpLoop(); |
EXPECT_TRUE(cancel_callback_called()); |
@@ -272,7 +285,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(cancel_callback()); |
@@ -291,7 +305,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(); |
@@ -308,7 +323,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(); |
@@ -325,14 +341,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); |
@@ -344,7 +362,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); |
@@ -356,7 +375,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()); |
@@ -367,7 +387,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. |
@@ -390,7 +411,8 @@ TEST_F(BackgroundLoaderOfflinerTest, NoNextOnInternetDisconnected) { |
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. |
@@ -413,7 +435,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 |