Move GetDocument method from WebFrame to WebLocalFrame.

content/shell/test_runner/accessibility_controller.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/shell/test_runner/accessibility_controller.h"
 
+#include <string>
+
 #include "base/macros.h"
 #include "content/shell/test_runner/web_view_test_proxy.h"
 #include "gin/handle.h"
 #include "gin/object_template_builder.h"
 #include "gin/wrappable.h"
 #include "third_party/WebKit/public/web/WebDocument.h"
 #include "third_party/WebKit/public/web/WebElement.h"
 #include "third_party/WebKit/public/web/WebFrame.h"
 #include "third_party/WebKit/public/web/WebKit.h"
 #include "third_party/WebKit/public/web/WebLocalFrame.h"
@@ skipping 192 matching lines @@
 
 void AccessibilityController::UnsetNotificationListener() {
   notification_callback_.Reset();
 }
 
 v8::Local<v8::Object> AccessibilityController::FocusedElement() {
   blink::WebFrame* frame = web_view()->MainFrame();
   if (!frame)
     return v8::Local<v8::Object>();
 
+  CHECK(frame->IsWebLocalFrame())
+      << "This function cannot be called if the main frame is not a "
+         "local frame.";
   blink::WebAXObject focused_element =
-      blink::WebAXObject::FromWebDocumentFocused(frame->GetDocument());
+      blink::WebAXObject::FromWebDocumentFocused(
+          frame->ToWebLocalFrame()->GetDocument());
   if (focused_element.IsNull())
-    focused_element = blink::WebAXObject::FromWebView(*web_view());
+    focused_element = GetAccessibilityObjectForMainFrame();
   return elements_.GetOrCreate(focused_element);
 }
 
 v8::Local<v8::Object> AccessibilityController::RootElement() {
-  blink::WebAXObject root_element =
-      blink::WebAXObject::FromWebView(*web_view());
-  return elements_.GetOrCreate(root_element);
+  return elements_.GetOrCreate(GetAccessibilityObjectForMainFrame());
 }
 
 v8::Local<v8::Object> AccessibilityController::AccessibleElementById(
     const std::string& id) {
-  blink::WebAXObject root_element =
-      blink::WebAXObject::FromWebView(*web_view());
+  blink::WebAXObject root_element = GetAccessibilityObjectForMainFrame();
 
   if (!root_element.UpdateLayoutAndCheckValidity())
     return v8::Local<v8::Object>();
 
   return FindAccessibleElementByIdRecursive(
       root_element, blink::WebString::FromUTF8(id.c_str()));
 }
 
 v8::Local<v8::Object>
 AccessibilityController::FindAccessibleElementByIdRecursive(
@@ skipping 18 matching lines @@
       return result;
   }
 
   return v8::Local<v8::Object>();
 }
 
 blink::WebView* AccessibilityController::web_view() {
   return web_view_test_proxy_base_->web_view();
 }
 
+blink::WebAXObject
+AccessibilityController::GetAccessibilityObjectForMainFrame() {
+  blink::WebFrame* frame = web_view()->MainFrame();
+  CHECK(frame && frame->IsWebLocalFrame())
+      << "This function cannot be called if the main frame is not a "
+         "local frame.";
+  return blink::WebAXObject::FromWebDocument(
+      web_view()->MainFrame()->ToWebLocalFrame()->GetDocument());
+}
+
 }  // namespace test_runner