diff options
Diffstat (limited to 'content/test/test_render_frame_host.cc')
-rw-r--r-- | content/test/test_render_frame_host.cc | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/content/test/test_render_frame_host.cc b/content/test/test_render_frame_host.cc index 57f8c0f..e9a3bc8 100644 --- a/content/test/test_render_frame_host.cc +++ b/content/test/test_render_frame_host.cc @@ -4,6 +4,9 @@ #include "content/test/test_render_frame_host.h" +#include "content/browser/frame_host/frame_tree.h" +#include "content/test/test_render_view_host.h" + namespace content { TestRenderFrameHost::TestRenderFrameHost(RenderViewHostImpl* render_view_host, @@ -17,7 +20,13 @@ TestRenderFrameHost::TestRenderFrameHost(RenderViewHostImpl* render_view_host, frame_tree, frame_tree_node, routing_id, - is_swapped_out) {} + is_swapped_out) { + // Allow TestRenderViewHosts to easily access their main frame RFH. + if (frame_tree_node == frame_tree->root()) { + static_cast<TestRenderViewHost*>(render_view_host)-> + set_main_render_frame_host(this); + } +} TestRenderFrameHost::~TestRenderFrameHost() {} |