Move GetDocument method from WebFrame to WebLocalFrame.

extensions/renderer/api/automation/automation_api_helper.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 "extensions/renderer/api/automation/automation_api_helper.h"
 
 #include "content/public/renderer/render_view.h"
 #include "extensions/common/extension_messages.h"
 #include "third_party/WebKit/public/web/WebAXObject.h"
 #include "third_party/WebKit/public/web/WebDocument.h"
 #include "third_party/WebKit/public/web/WebElement.h"
 #include "third_party/WebKit/public/web/WebExceptionCode.h"
-#include "third_party/WebKit/public/web/WebFrame.h"
+#include "third_party/WebKit/public/web/WebLocalFrame.h"
 #include "third_party/WebKit/public/web/WebNode.h"
 #include "third_party/WebKit/public/web/WebView.h"
 
 namespace extensions {
 
 AutomationApiHelper::AutomationApiHelper(content::RenderView* render_view)
     : content::RenderViewObserver(render_view) {
 }
 
 AutomationApiHelper::~AutomationApiHelper() {
@@ skipping 16 matching lines @@
                                           int acc_obj_id,
                                           const base::string16& selector) {
   ExtensionHostMsg_AutomationQuerySelector_Error error;
   if (!render_view() || !render_view()->GetWebView() ||
       !render_view()->GetWebView()->MainFrame()) {
     error.value = ExtensionHostMsg_AutomationQuerySelector_Error::kNoMainFrame;
     Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
         routing_id(), request_id, error, 0));
     return;
   }
-  blink::WebDocument document =
-      render_view()->GetWebView()->MainFrame()->GetDocument();
+
+  // ExtensionMsg_AutomationQuerySelector should only be sent to an active view.
+  DCHECK(render_view()->GetWebView()->MainFrame()->IsWebLocalFrame());
+
+  blink::WebDocument document = render_view()
+                                    ->GetWebView()
+                                    ->MainFrame()
+                                    ->ToWebLocalFrame()
+                                    ->GetDocument();
   if (document.IsNull()) {
     error.value = ExtensionHostMsg_AutomationQuerySelector_Error::kNoDocument;
     Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
         routing_id(), request_id, error, 0));
     return;
   }
   blink::WebNode start_node = document;
   if (acc_obj_id > 0) {
     blink::WebAXObject start_acc_obj =
         blink::WebAXObject::FromWebDocumentByID(document, acc_obj_id);
@@ skipping 21 matching lines @@
         result_acc_obj = result_acc_obj.ParentObject();
 
       result_acc_obj_id = result_acc_obj.AxID();
     }
   }
   Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
       routing_id(), request_id, error, result_acc_obj_id));
 }
 
 }  // namespace extensions