[Offline Pages] SnapshotController support for delay after onLoadCompleted and also parameterized c…

components/offline_pages/snapshot_controller.h

 // 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.
 
 #ifndef COMPONENTS_OFFLINE_PAGES_SNAPSHOT_CONTROLLER_H_
 #define COMPONENTS_OFFLINE_PAGES_SNAPSHOT_CONTROLLER_H_
 
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/single_thread_task_runner.h"
@@ skipping 30 matching lines @@
     // it is assumed that later snapshots are better then previous.
     virtual void StartSnapshot() = 0;
 
    protected:
     virtual ~Client() {}
   };
 
   SnapshotController(
       const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
       SnapshotController::Client* client);
+  SnapshotController(
+      const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
+      SnapshotController::Client* client,
+      size_t delay_after_document_available_ms,
+      size_t delay_after_document_on_load_completed_ms);
   virtual ~SnapshotController();
 
   // Resets the 'session', returning controller to initial state.
   void Reset();
 
   // Stops current session, no more Client::StartSnapshot calls will be
   // invoked from the SnapshotController until current session is Reset().
   // Called by Client, for example when it encounters an error loading the page.
   void Stop();
 
   // The way for Client to report that previously started snapshot is
   // now completed (so the next one can be started).
   void PendingSnapshotCompleted();
 
   // Invoked from WebContentObserver::DocumentAvailableInMainFrame
   void DocumentAvailableInMainFrame();
 
   // Invoked from WebContentObserver::DocumentOnLoadCompletedInMainFrame
   void DocumentOnLoadCompletedInMainFrame();
 
   size_t GetDelayAfterDocumentAvailableForTest();
+  size_t GetDelayAfterDocumentOnLoadCompletedForTest();
 
  private:
   void MaybeStartSnapshot();
+  void MaybeStartSnapshotThenStop();
 
   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
   // Client owns this class.
   SnapshotController::Client* client_;
   SnapshotController::State state_;
+  size_t delay_after_document_available_ms_;
+  size_t delay_after_document_on_load_completed_ms_;
 
   base::WeakPtrFactory<SnapshotController> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(SnapshotController);
 };
 
 }  // namespace offline_pages
 
 #endif  // COMPONENTS_OFFLINE_PAGES_SNAPSHOT_CONTROLLER_H_