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

chrome/browser/android/offline_pages/background_loader_offliner_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/android/offline_pages/background_loader_offliner.h"
 
 #include "base/bind.h"
 #include "base/run_loop.h"
 #include "base/test/histogram_tester.h"
 #include "base/test/scoped_mock_time_message_loop_task_runner.h"
@@ skipping 171 matching lines @@
       progress_(0LL),
       request_status_(Offliner::RequestStatus::UNKNOWN) {}
 
 BackgroundLoaderOfflinerTest::~BackgroundLoaderOfflinerTest() {}
 
 void BackgroundLoaderOfflinerTest::SetUp() {
   model_ = new MockOfflinePageModel();
   policy_.reset(new OfflinerPolicy());
   offliner_.reset(
       new TestBackgroundLoaderOffliner(profile(), policy_.get(), model_));
-  offliner_->SetPageDelayForTest(0L);
+  offliner_->SetSnapshotControllerForTest(new SnapshotController(
+      base::ThreadTaskRunnerHandle::Get(), offliner_.get(),
+      0L /* DelayAfterDocumentAvailable */, 0L /* DelayAfterDocumentOnLoad */,
+      false /* DocumentAvailableTriggersSnapshot */));
 }
 
 void BackgroundLoaderOfflinerTest::OnCompletion(
     const SavePageRequest& request,
     Offliner::RequestStatus status) {
   DCHECK(!completion_callback_called_);  // Expect 1 callback per request.
   completion_callback_called_ = true;
   request_status_ = status;
 }
 
@@ skipping 351 matching lines @@
   EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(),
                                       progress_callback()));
   // Sets lowbar.
   offliner()->DocumentAvailableInMainFrame();
   // Timeout
   EXPECT_FALSE(offliner()->HandleTimeout(request));
   EXPECT_FALSE(SaveInProgress());
 }
 
 }  // namespace offline_pages