PlzNavigate: Speculatively spawns a renderer process for navigations.

content/test/test_web_contents.cc

 // Copyright (c) 2012 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 "content/test/test_web_contents.h"
 
 #include <utility>
 
 #include "base/command_line.h"
 #include "content/browser/browser_url_handler_impl.h"
@@ skipping 136 matching lines @@
   TestRenderFrameHost* old_rfh = GetMainFrame();
   const NavigationEntry* entry = GetController().GetPendingEntry();
   DCHECK(entry);
 
   // Simulate the BeforeUnload ACK if necessary.
   // PlzNavigate: the pending RFH is not created before the navigation commit,
   // so it is necessary to simulate the IO thread response here to commit in the
   // proper renderer.
   old_rfh->PrepareForCommit(entry->GetURL());
 
-  TestRenderFrameHost* rfh = GetPendingMainFrame();
+  TestRenderFrameHost* rfh = GetFutureMainFrame();
+
   if (!rfh)
     rfh = old_rfh;
 
   int page_id = entry->GetPageID();
   if (page_id == -1) {
     // It's a new navigation, assign a never-seen page id to it.
     page_id = GetMaxPageIDForSiteInstance(rfh->GetSiteInstance()) + 1;
   }
 
   rfh->SendNavigate(page_id, entry->GetURL());
@@ skipping 52 matching lines @@
 
 void TestWebContents::TestDidFailLoadWithError(
     const GURL& url,
     int error_code,
     const base::string16& error_description) {
   FrameHostMsg_DidFailLoadWithError msg(
       0, url, error_code, error_description);
   frame_tree_.root()->current_frame_host()->OnMessageReceived(msg);
 }
 
+TestRenderFrameHost* TestWebContents::GetFutureMainFrame() const {
+  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
+          switches::kEnableBrowserSideNavigation)) {
+    return static_cast<TestRenderFrameHost*>(
+        GetRenderManager()->speculative_render_frame_host_.get());
+  }
+  return GetPendingMainFrame();
+}
+
 void TestWebContents::CreateNewWindow(
     int render_process_id,
     int route_id,
     int main_frame_route_id,
     const ViewHostMsg_CreateWindow_Params& params,
     SessionStorageNamespace* session_storage_namespace) {
 }
 
 void TestWebContents::CreateNewWidget(int render_process_id,
                                       int route_id,
@@ skipping 11 matching lines @@
 }
 
 void TestWebContents::ShowCreatedWidget(int route_id,
                                         const gfx::Rect& initial_pos) {
 }
 
 void TestWebContents::ShowCreatedFullscreenWidget(int route_id) {
 }
 
 }  // namespace content