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

chrome/browser/android/offline_pages/prerendering_loader_unittest.cc

Index: chrome/browser/android/offline_pages/prerendering_loader_unittest.cc
diff --git a/chrome/browser/android/offline_pages/prerendering_loader_unittest.cc b/chrome/browser/android/offline_pages/prerendering_loader_unittest.cc
index eb0bf2bc15da74dbbc0ec01053420f60ebaebb56..9a01ce8b02aeff3621a534395781562c0563fc52 100644
--- a/chrome/browser/android/offline_pages/prerendering_loader_unittest.cc
+++ b/chrome/browser/android/offline_pages/prerendering_loader_unittest.cc
@@ -223,6 +223,8 @@ TEST_F(PrerenderingLoaderTest, LoadPageLoadSucceededFromPrerenderStopLoading) {
   EXPECT_FALSE(loader()->IsLoaded());
 
   test_adapter()->GetObserver()->OnPrerenderStopLoading();
+  // Skip SnapshotController's wait time and emulate StartSnapshot call.
+  loader()->StartSnapshot();
   PumpLoop();
   EXPECT_FALSE(loader()->IsIdle());
   EXPECT_TRUE(loader()->IsLoaded());