Extend the ToRenderFrameHost magic to FrameTreeNode* and Shell*

content/test/content_browser_test_utils_internal.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/test/content_browser_test_utils_internal.h"
 
 #include <stddef.h>
 
 #include <algorithm>
 #include <map>
 #include <set>
 #include <vector>
 
 #include "base/strings/stringprintf.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "cc/surfaces/surface.h"
 #include "cc/surfaces/surface_manager.h"
 #include "content/browser/compositor/surface_utils.h"
 #include "content/browser/frame_host/cross_process_frame_connector.h"
 #include "content/browser/frame_host/frame_tree_node.h"
 #include "content/browser/frame_host/navigator.h"
 #include "content/browser/frame_host/render_frame_proxy_host.h"
 #include "content/browser/frame_host/render_widget_host_view_child_frame.h"
 #include "content/browser/renderer_host/delegated_frame_host.h"
 #include "content/browser/renderer_host/render_widget_host_view_base.h"
 #include "content/public/browser/resource_dispatcher_host.h"
 #include "content/public/browser/resource_throttle.h"
+#include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/content_browser_test_utils.h"
 #include "content/shell/browser/shell.h"
 #include "content/shell/browser/shell_javascript_dialog_manager.h"
 #include "content/test/test_frame_navigation_observer.h"
 #include "net/url_request/url_request.h"
 
 namespace content {
 
 namespace {
@@ skipping 29 matching lines @@
   observer.Wait();
 }
 
 void SetShouldProceedOnBeforeUnload(Shell* shell, bool proceed) {
   ShellJavaScriptDialogManager* manager =
       static_cast<ShellJavaScriptDialogManager*>(
           shell->GetJavaScriptDialogManager(shell->web_contents()));
   manager->set_should_proceed_on_beforeunload(proceed);
 }
 
+RenderFrameHost* ConvertToRenderFrameHost(FrameTreeNode* frame_tree_node) {
+  return frame_tree_node->current_frame_host();
+}
+
 FrameTreeVisualizer::FrameTreeVisualizer() {
 }
 
 FrameTreeVisualizer::~FrameTreeVisualizer() {
 }
 
 std::string FrameTreeVisualizer::DepictFrameTree(FrameTreeNode* root) {
   // Tracks the sites actually used in this depiction.
   std::map<std::string, SiteInstance*> legend;
 
@@ skipping 325 matching lines @@
     loop_runner_->Quit();
 }
 
 void TestNavigationManager::OnWillStartRequest() {
   navigation_paused_ = true;
   if (loop_runner_)
     loop_runner_->Quit();
 }
 
 }  // namespace content