Index: content/public/test/test_renderer_host.cc |
diff --git a/content/public/test/test_renderer_host.cc b/content/public/test/test_renderer_host.cc |
index d385871bbf9ea30a84975834abc2050cec70f82a..8765eef5cc25f117f53e8df564e92452e7402873 100644 |
--- a/content/public/test/test_renderer_host.cc |
+++ b/content/public/test/test_renderer_host.cc |
@@ -16,9 +16,9 @@ |
#include "content/public/browser/browser_thread.h" |
#include "content/public/browser/web_contents.h" |
#include "content/public/common/browser_side_navigation_policy.h" |
+#include "content/public/test/browser_side_navigation_test_utils.h" |
#include "content/public/test/mock_render_process_host.h" |
#include "content/public/test/test_browser_context.h" |
-#include "content/test/browser_side_navigation_test_utils.h" |
#include "content/test/content_browser_sanity_checker.h" |
#include "content/test/test_render_frame_host.h" |
#include "content/test/test_render_frame_host_factory.h" |
@@ -60,6 +60,10 @@ RenderFrameHost* RenderFrameHostTester::GetPendingForController( |
NavigationController* controller) { |
WebContentsImpl* web_contents = static_cast<WebContentsImpl*>( |
controller->GetWebContents()); |
+ if (IsBrowserSideNavigationEnabled()) { |
+ return web_contents->GetRenderManagerForTesting() |
+ ->speculative_render_frame_host_.get(); |
+ } |
return web_contents->GetRenderManagerForTesting()->pending_frame_host(); |
} |