[Offline pages]: Move logic for whether to consider the DocumentAvailableInMainFrame signal to the …

chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc

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 d9078c44a7ba5dcf554c2d39ba8ee7d5c9fbac0f..77dd44d2ba93fffe27d78a92e88418dc30c0adeb 100644
--- a/chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc
+++ b/chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc
@@ -150,6 +150,15 @@ class BackgroundLoaderOfflinerTest : public testing::Test {
   void PumpLoop() { base::RunLoop().RunUntilIdle(); }
 
   void CompleteLoading() {
+    // Reset snapshot controller.
+    std::unique_ptr<SnapshotController> snapshot_controller(
+        new SnapshotController(base::ThreadTaskRunnerHandle::Get(),
+                               offliner_.get(),
+                               0L /* DelayAfterDocumentAvailable */,
+                               0L /* DelayAfterDocumentOnLoad */,
+                               false /* DocumentAvailableTriggersSnapshot */));
+    offliner_->SetSnapshotControllerForTest(std::move(snapshot_controller));
+    // Call complete loading.
     offliner()->DocumentOnLoadCompletedInMainFrame();
     PumpLoop();
   }
@@ -184,7 +193,6 @@ BackgroundLoaderOfflinerTest::~BackgroundLoaderOfflinerTest() {}
 void BackgroundLoaderOfflinerTest::SetUp() {
   model_ = new MockOfflinePageModel();
   offliner_.reset(new TestBackgroundLoaderOffliner(profile(), nullptr, model_));
-  offliner_->SetPageDelayForTest(0L);
 }
 
 void BackgroundLoaderOfflinerTest::OnCompletion(