Introduce the ThrottleManager

content/public/test/navigation_simulator.cc

 // Copyright 2017 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/public/test/navigation_simulator.h"
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "content/browser/frame_host/navigation_handle_impl.h"
@@ skipping 334 matching lines @@
     render_frame_host_->OnMessageReceived(
         FrameHostMsg_DidFailProvisionalLoadWithError(
             render_frame_host_->GetRoutingID(), error_params));
     if (!should_result_in_error_page) {
       render_frame_host_->OnMessageReceived(
           FrameHostMsg_DidStopLoading(render_frame_host_->GetRoutingID()));
     }
   }
 
   if (IsBrowserSideNavigationEnabled()) {
-    CHECK_EQ(1, num_ready_to_commit_called_);
     if (should_result_in_error_page) {
+      CHECK_EQ(1, num_ready_to_commit_called_);
       // Update the RenderFrameHost now that we know which RenderFrameHost will
       // commit the error page.
       render_frame_host_ =
           static_cast<TestRenderFrameHost*>(handle_->GetRenderFrameHost());
     }
   }
 
   if (should_result_in_error_page)
     CHECK_EQ(0, num_did_finish_navigation_called_);
   else
@@ skipping 184 matching lines @@
     throttle_checks_wait_closure_.Run();
 }
 
 void NavigationSimulator::PrepareCompleteCallbackOnHandle() {
   last_throttle_check_result_.reset();
   handle_->set_complete_callback_for_testing(
       base::Bind(&NavigationSimulator::OnThrottleChecksComplete,
                  weak_factory_.GetWeakPtr()));
 }
 
+RenderFrameHost* NavigationSimulator::GetFinalRenderFrameHost() {
+  CHECK_EQ(state_, FINISHED);
+  return render_frame_host_;
+}
+
 }  // namespace content