Use WebViewHelper::LocalMainFrame where possible.

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"
@@ skipping 42 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()->MainFrameImpl()->GetDocument();
+  return web_view_helper_.LocalMainFrame()->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