| 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() {}
|
|
|
|
|