Fix Android unit_tests failures with PlzNavigate.

content/test/test_render_frame_host.cc

 // Copyright 2013 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_render_frame_host.h"
 
 #include "base/guid.h"
+#include "base/run_loop.h"
 #include "content/browser/frame_host/frame_tree.h"
 #include "content/browser/frame_host/navigation_handle_impl.h"
 #include "content/browser/frame_host/navigation_request.h"
 #include "content/browser/frame_host/navigator.h"
 #include "content/browser/frame_host/navigator_impl.h"
 #include "content/browser/frame_host/render_frame_host_delegate.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/common/frame_messages.h"
 #include "content/common/frame_owner_properties.h"
 #include "content/public/browser/navigation_throttle.h"
@@ skipping 92 matching lines @@
   }
 
   OnDidStartLoading(true);
   OnDidStartProvisionalLoad(url, base::TimeTicks::Now());
   SimulateWillStartRequest(ui::PAGE_TRANSITION_LINK);
 }
 
 void TestRenderFrameHost::SimulateRedirect(const GURL& new_url) {
   if (IsBrowserSideNavigationEnabled()) {
     NavigationRequest* request = frame_tree_node_->navigation_request();
+    if (!request->loader_for_testing()) {
+      base::RunLoop loop;
+      request->set_on_start_checks_complete_closure_for_testing(
+          loop.QuitClosure());
+      loop.Run();
+    }
     TestNavigationURLLoader* url_loader =
         static_cast<TestNavigationURLLoader*>(request->loader_for_testing());
     CHECK(url_loader);
     url_loader->SimulateServerRedirect(new_url);
     return;
   }
 
   navigation_handle()->CallWillRedirectRequestForTesting(new_url, false, GURL(),
                                                          false);
 }
@@ skipping 46 matching lines @@
                                                   int error_code) {
   if (IsBrowserSideNavigationEnabled()) {
     NavigationRequest* request = frame_tree_node_->navigation_request();
     CHECK(request);
     // Simulate a beforeUnload ACK from the renderer if the browser is waiting
     // for it. If it runs it will update the request state.
     if (request->state() == NavigationRequest::WAITING_FOR_RENDERER_RESPONSE) {
       static_cast<TestRenderFrameHost*>(frame_tree_node()->current_frame_host())
           ->SendBeforeUnloadACK(true);
     }
+    if (!request->loader_for_testing()) {
+      base::RunLoop loop;
+      request->set_on_start_checks_complete_closure_for_testing(
+          loop.QuitClosure());
+      loop.Run();
+    }
     TestNavigationURLLoader* url_loader =
         static_cast<TestNavigationURLLoader*>(request->loader_for_testing());
     CHECK(url_loader);
     url_loader->SimulateError(error_code);
     return;
   }
 
   FrameHostMsg_DidFailProvisionalLoadWithError_Params error_params;
   error_params.error_code = error_code;
   error_params.url = url;
@@ skipping 244 matching lines @@
   if (request->state() == NavigationRequest::WAITING_FOR_RENDERER_RESPONSE) {
     static_cast<TestRenderFrameHost*>(frame_tree_node()->current_frame_host())
         ->SendBeforeUnloadACK(true);
   }
 
   if (!have_to_make_network_request)
     return;  // |request| is destructed by now.
 
   CHECK(request->state() == NavigationRequest::STARTED);
 
+  if (!request->loader_for_testing()) {
+    base::RunLoop loop;
+    request->set_on_start_checks_complete_closure_for_testing(
+        loop.QuitClosure());
+    loop.Run();
+  }
+
   TestNavigationURLLoader* url_loader =
       static_cast<TestNavigationURLLoader*>(request->loader_for_testing());
   CHECK(url_loader);
 
   // If a non-empty |redirect_url| was provided, simulate a server redirect.
   if (!redirect_url.is_empty())
     url_loader->SimulateServerRedirect(redirect_url);
 
   // Simulate the network stack commit.
   scoped_refptr<ResourceResponse> response(new ResourceResponse);
@@ skipping 22 matching lines @@
   // PlzNavigate: NavigationHandle::WillStartRequest has already been called at
   // this point.
   if (!navigation_handle() || IsBrowserSideNavigationEnabled())
     return;
   navigation_handle()->CallWillStartRequestForTesting(
       false /* is_post */, Referrer(GURL(), blink::WebReferrerPolicyDefault),
       true /* user_gesture */, transition, false /* is_external_protocol */);
 }
 
 }  // namespace content