Implementing recent pages snapshot capture.

components/offline_pages/snapshot_controller_unittest.cc

Index: components/offline_pages/snapshot_controller_unittest.cc
diff --git a/components/offline_pages/snapshot_controller_unittest.cc b/components/offline_pages/snapshot_controller_unittest.cc
index 2a455cf6f82a85020add61c547aa8920e6308076..d85ac7aeb959a70d7dc726fe94780d5e101adb35 100644
--- a/components/offline_pages/snapshot_controller_unittest.cc
+++ b/components/offline_pages/snapshot_controller_unittest.cc
@@ -22,7 +22,6 @@ class 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
@@ -30,7 +29,7 @@ class SnapshotControllerTest
   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
@@ -65,9 +64,8 @@ void SnapshotControllerTest::TearDown() {
   controller_.reset();
 }
 
-bool SnapshotControllerTest::StartSnapshot() {
+void SnapshotControllerTest::StartSnapshot() {
   snapshot_count_++;
-  return snapshot_started_;
 }
 
 void SnapshotControllerTest::PumpLoop() {
@@ -144,16 +142,22 @@ TEST_F(SnapshotControllerTest, Stop) {
   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()));
-  // Should have one snapshot requested, but not reported started.
-  EXPECT_EQ(1, snapshot_count());
-  // Should start another snapshot since previous did not start
+  // 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()));
+  // No snapshot since session was reset.
   EXPECT_EQ(2, snapshot_count());
 }