Index: content/test/test_render_frame_host.cc |
diff --git a/content/test/test_render_frame_host.cc b/content/test/test_render_frame_host.cc |
index 57f8c0fe5122ac564bb1d3b413380759177eaa69..e9a3bc8d3e30be01812a0f78357fe520491675ea 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() {} |