Implementing recent pages snapshot capture.

components/offline_pages/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/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"
 #include "base/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace offline_pages {
 
 class SnapshotControllerTest
     : public testing::Test,
       public SnapshotController::Client {
  public:
   SnapshotControllerTest();
   ~SnapshotControllerTest() override;
 
   SnapshotController* controller() { return controller_.get(); }
-  void set_snapshot_started(bool started) { snapshot_started_ = started; }
   int snapshot_count() { return snapshot_count_; }
 
   // testing::Test
   void SetUp() override;
   void TearDown() override;
 
   // SnapshotController::Client
-  bool StartSnapshot() override;
+  void StartSnapshot() override;
 
   // Utility methods.
   // Runs until all of the tasks that are not delayed are gone from the task
   // queue.
   void PumpLoop();
   // Fast-forwards virtual time by |delta|, causing tasks with a remaining
   // delay less than or equal to |delta| to be executed.
   void FastForwardBy(base::TimeDelta delta);
 
  private:
@@ skipping 14 matching lines @@
 
 void SnapshotControllerTest::SetUp() {
   controller_.reset(new SnapshotController(task_runner_, this));
   snapshot_started_ = true;
 }
 
 void SnapshotControllerTest::TearDown() {
   controller_.reset();
 }
 
-bool SnapshotControllerTest::StartSnapshot() {
+void SnapshotControllerTest::StartSnapshot() {
   snapshot_count_++;
-  return snapshot_started_;
 }
 
 void SnapshotControllerTest::PumpLoop() {
   task_runner_->RunUntilIdle();
 }
 
 void SnapshotControllerTest::FastForwardBy(base::TimeDelta delta) {
   task_runner_->FastForwardBy(delta);
 }
 
@@ skipping 56 matching lines @@
   controller()->Stop();
   FastForwardBy(base::TimeDelta::FromMilliseconds(
       controller()->GetDelayAfterDocumentAvailableForTest()));
   // Should not start snapshots
   EXPECT_EQ(0, snapshot_count());
   // Also should not start snapshot.
   controller()->DocumentOnLoadCompletedInMainFrame();
   EXPECT_EQ(0, snapshot_count());
 }
 
-TEST_F(SnapshotControllerTest, ClientDidntStartSnapshot) {
-  // This will tell that Client did not start the snapshot
-  set_snapshot_started(false);
+TEST_F(SnapshotControllerTest, ClientReset) {
+  controller()->DocumentAvailableInMainFrame();
+
+  controller()->Reset();
+  FastForwardBy(base::TimeDelta::FromMilliseconds(
+      controller()->GetDelayAfterDocumentAvailableForTest()));
+  // No snapshot since session was reset.
+  EXPECT_EQ(0, snapshot_count());
+  controller()->DocumentOnLoadCompletedInMainFrame();
+  EXPECT_EQ(1, snapshot_count());
+
+  controller()->Reset();
   controller()->DocumentAvailableInMainFrame();
   FastForwardBy(base::TimeDelta::FromMilliseconds(
       controller()->GetDelayAfterDocumentAvailableForTest()));
-  // Should have one snapshot requested, but not reported started.
-  EXPECT_EQ(1, snapshot_count());
-  // Should start another snapshot since previous did not start
-  controller()->DocumentOnLoadCompletedInMainFrame();
+  // No snapshot since session was reset.
   EXPECT_EQ(2, snapshot_count());
 }
 
 }  // namespace offline_pages