Move GetDocument method from WebFrame to WebLocalFrame.

third_party/WebKit/Source/web/tests/WebDocumentTest.cpp

 // 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 "public/web/WebDocument.h"
 
+#include <string>
+
 #include "core/CSSPropertyNames.h"
 #include "core/HTMLNames.h"
 #include "core/dom/NodeComputedStyle.h"
 #include "core/dom/StyleEngine.h"
 #include "core/frame/FrameTestHelpers.h"
 #include "core/frame/LocalFrame.h"
+#include "core/frame/WebLocalFrameBase.h"
 #include "core/html/HTMLElement.h"
 #include "core/html/HTMLLinkElement.h"
 #include "core/page/Page.h"
 #include "core/style/ComputedStyle.h"
 #include "platform/RuntimeEnabledFeatures.h"
 #include "platform/graphics/Color.h"
 #include "platform/testing/URLTestHelpers.h"
 #include "platform/testing/UnitTestHelpers.h"
 #include "platform/weborigin/SchemeRegistry.h"
 #include "platform/weborigin/SecurityOrigin.h"
@@ skipping 27 matching lines @@
 void WebDocumentTest::LoadURL(const std::string& url) {
   web_view_helper_.InitializeAndLoad(url);
 }
 
 Document* WebDocumentTest::TopDocument() const {
   return ToLocalFrame(web_view_helper_.WebView()->GetPage()->MainFrame())
       ->GetDocument();
 }
 
 WebDocument WebDocumentTest::TopWebDocument() const {
-  return web_view_helper_.WebView()->MainFrame()->GetDocument();
+  return web_view_helper_.WebView()->MainFrameImpl()->GetDocument();
 }
 
 TEST_F(WebDocumentTest, InsertAndRemoveStyleSheet) {
   LoadURL("about:blank");
 
   WebDocument web_doc = TopWebDocument();
   Document* core_doc = TopDocument();
 
   unsigned start_count = core_doc->GetStyleEngine().StyleForElementCount();
 
@@ skipping 305 matching lines @@
 
   ASSERT_EQ(ToOriginA(g_nested_origin_a_in_origin_b),
             TopDocument()->FirstPartyForCookies());
   ASSERT_EQ(ToOriginA(g_nested_origin_a_in_origin_b),
             NestedDocument()->FirstPartyForCookies());
   ASSERT_EQ(ToOriginA(g_nested_origin_a_in_origin_b),
             NestedNestedDocument()->FirstPartyForCookies());
 }
 
 }  // namespace blink