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

chrome/browser/android/offline_pages/prerendering_loader_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 "chrome/browser/android/offline_pages/prerendering_loader.h"
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/run_loop.h"
@@ skipping 205 matching lines @@
   EXPECT_FALSE(loader()->IsLoaded());
   EXPECT_TRUE(loader()->LoadPage(
       gurl,
       base::Bind(&PrerenderingLoaderTest::OnLoadDone, base::Unretained(this))));
 
   PumpLoop();
   EXPECT_FALSE(loader()->IsIdle());
   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());
   EXPECT_TRUE(callback_called());
   EXPECT_EQ(Offliner::RequestStatus::LOADED, callback_load_status());
 
   // Consider Prerenderer stops (eg, times out) before Loader is done with it.
   test_adapter()->GetObserver()->OnPrerenderStop();
   PumpLoop();
   EXPECT_TRUE(loader()->IsIdle());
@@ skipping 145 matching lines @@
   test_adapter()->FailStart();
   GURL gurl("http://testit.sea");
   EXPECT_TRUE(loader()->IsIdle());
   EXPECT_FALSE(loader()->LoadPage(
       gurl,
       base::Bind(&PrerenderingLoaderTest::OnLoadDone, base::Unretained(this))));
   EXPECT_TRUE(loader()->IsIdle());
 }
 
 }  // namespace offline_pages