Extend the ToRenderFrameHost magic to FrameTreeNode* and Shell*

content/browser/renderer_host/input/touch_selection_controller_client_aura_browsertest.cc

 // Copyright 2015 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/browser/renderer_host/input/touch_selection_controller_client_aura.h"
 
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
@@ skipping 117 matching lines @@
   void StartTestWithPage(const std::string& url) {
     ASSERT_TRUE(embedded_test_server()->Start());
     GURL test_url(embedded_test_server()->GetURL(url));
     NavigateToURL(shell(), test_url);
     aura::Window* content = shell()->web_contents()->GetContentNativeView();
     content->GetHost()->SetBounds(gfx::Rect(800, 600));
   }
 
   bool GetPointInsideText(gfx::PointF* point) {
     std::string str;
-    if (ExecuteScriptAndExtractString(shell()->web_contents()->GetMainFrame(),
-                                      "get_point_inside_text()", &str)) {
+    if (ExecuteScriptAndExtractString(shell(), "get_point_inside_text()",
+                                      &str)) {
       return JSONToPoint(str, point);
     }
     return false;
   }
 
   bool GetPointInsideTextfield(gfx::PointF* point) {
     std::string str;
-    if (ExecuteScriptAndExtractString(shell()->web_contents()->GetMainFrame(),
-                                      "get_point_inside_textfield()", &str)) {
+    if (ExecuteScriptAndExtractString(shell(), "get_point_inside_textfield()",
+                                      &str)) {
       return JSONToPoint(str, point);
     }
     return false;
   }
 
-  bool EmptyTextfield() {
-    return ExecuteScript(shell()->web_contents()->GetMainFrame(),
-                         "empty_textfield()");
-  }
+  bool EmptyTextfield() { return ExecuteScript(shell(), "empty_textfield()"); }
 
   RenderWidgetHostViewAura* GetRenderWidgetHostViewAura() {
     return static_cast<RenderWidgetHostViewAura*>(
         shell()->web_contents()->GetRenderWidgetHostView());
   }
 
   TestTouchSelectionControllerClientAura* selection_controller_client() {
     return selection_controller_client_;
   }
 
@@ skipping 509 matching lines @@
 
   // The handle should have moved to right.
   EXPECT_EQ(initial_handle_rect.y(), moved_handle_rect.y());
   EXPECT_LT(initial_handle_rect.x(), moved_handle_rect.x());
 
   EXPECT_EQ(ui::TouchSelectionController::INSERTION_ACTIVE,
             rwhva->selection_controller()->active_status());
 }
 
 }  // namespace content