Index: content/public/test/navigation_simulator.cc |
diff --git a/content/public/test/navigation_simulator.cc b/content/public/test/navigation_simulator.cc |
index ffd2485f1f77af393490e4717780544621a65769..99087f171f751813ad9df018db555e9e0d967c5c 100644 |
--- a/content/public/test/navigation_simulator.cc |
+++ b/content/public/test/navigation_simulator.cc |
@@ -352,8 +352,8 @@ void NavigationSimulator::Fail(int error_code) { |
} |
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_ = |
@@ -558,4 +558,9 @@ void NavigationSimulator::PrepareCompleteCallbackOnHandle() { |
weak_factory_.GetWeakPtr())); |
} |
+RenderFrameHost* NavigationSimulator::GetFinalRenderFrameHost() { |
+ CHECK_EQ(state_, FINISHED); |
+ return render_frame_host_; |
+} |
+ |
} // namespace content |