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

components/offline_pages/core/snapshot_controller_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 "components/offline_pages/core/snapshot_controller.h"
 
 #include "base/bind.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/test/test_mock_time_task_runner.h"
@@ skipping 35 matching lines @@
 };
 
 SnapshotControllerTest::SnapshotControllerTest()
     : task_runner_(new base::TestMockTimeTaskRunner),
       snapshot_started_(true),
       snapshot_count_(0) {}
 
 SnapshotControllerTest::~SnapshotControllerTest() {}
 
 void SnapshotControllerTest::SetUp() {
-  controller_.reset(new SnapshotController(task_runner_, this));
+  controller_ =
+      SnapshotController::CreateForForegroundOfflining(task_runner_, this);
   snapshot_started_ = true;
 }
 
 void SnapshotControllerTest::TearDown() {
   controller_.reset();
 }
 
 void SnapshotControllerTest::StartSnapshot() {
   snapshot_count_++;
 }
@@ skipping 114 matching lines @@
   controller()->PendingSnapshotCompleted();
   // Next snapshot should be initiated when new document is loaded.
   controller()->DocumentAvailableInMainFrame();
   FastForwardBy(base::TimeDelta::FromMilliseconds(
       controller()->GetDelayAfterDocumentAvailableForTest()));
   // No snapshot since session was reset.
   EXPECT_EQ(2, snapshot_count());
 }
 
 }  // namespace offline_pages