Use WebViewHelper::LocalMainFrame where possible.

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

 /*
  * Copyright (C) 2011, 2012 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 475 matching lines @@
   // Set canvas background to red with alpha.
   SkBitmap bitmap;
   bitmap.allocN32Pixels(kWidth, kHeight);
   SkCanvas canvas(bitmap);
   canvas.clear(kAlphaRed);
 
   PaintRecordBuilder builder(FloatRect(0, 0, kWidth, kHeight));
 
   // Paint the root of the main frame in the way that CompositedLayerMapping
   // would.
-  LocalFrameView* view =
-      web_view_helper_.WebView()->MainFrameImpl()->GetFrameView();
+  LocalFrameView* view = web_view_helper_.LocalMainFrame()->GetFrameView();
   PaintLayer* root_layer = view->GetLayoutViewItem().Layer();
   LayoutRect paint_rect(0, 0, kWidth, kHeight);
   PaintLayerPaintingInfo painting_info(root_layer, paint_rect,
                                        kGlobalPaintNormalPhase, LayoutSize());
   PaintLayerPainter(*root_layer)
       .PaintLayerContents(builder.Context(), painting_info,
                           kPaintLayerPaintingCompositingAllPhases);
   builder.EndRecording()->Playback(&canvas);
 
   // The result should be a blend of red and green.
@@ skipping 1496 matching lines @@
 
   web_view_helper_.WebView()->HandleInputEvent(WebCoalescedInputEvent(event));
   RunPendingTasks();
   return true;
 }
 
 bool WebViewTest::TapElementById(WebInputEvent::Type type,
                                  const WebString& id) {
   DCHECK(web_view_helper_.WebView());
   Element* element = static_cast<Element*>(
-      web_view_helper_.WebView()->MainFrameImpl()->GetDocument().GetElementById(
-          id));
+      web_view_helper_.LocalMainFrame()->GetDocument().GetElementById(id));
   return TapElement(type, element);
 }
 
 IntSize WebViewTest::PrintICBSizeFromPageSize(const FloatSize& page_size) {
   // The expected layout size comes from the calculation done in
   // ResizePageRectsKeepingRatio() which is used from PrintContext::begin() to
   // scale the page size.
   const float ratio = page_size.Height() / (float)page_size.Width();
   const int icb_width =
       floor(page_size.Width() * PrintContext::kPrintingMinimumShrinkFactor);
@@ skipping 1159 matching lines @@
                                      &frame_client);
   web_view_helper_.Reset();
   EXPECT_EQ(0, frame_client.Count());
 }
 
 TEST_P(WebViewTest, AddFrameInCloseURLUnload) {
   CreateChildCounterFrameClient frame_client;
   RegisterMockedHttpURLLoad("add_frame_in_unload.html");
   web_view_helper_.InitializeAndLoad(base_url_ + "add_frame_in_unload.html",
                                      &frame_client);
-  web_view_helper_.WebView()->MainFrameImpl()->DispatchUnloadEvent();
+  web_view_helper_.LocalMainFrame()->DispatchUnloadEvent();
   EXPECT_EQ(0, frame_client.Count());
   web_view_helper_.Reset();
 }
 
 TEST_P(WebViewTest, AddFrameInNavigateUnload) {
   CreateChildCounterFrameClient frame_client;
   RegisterMockedHttpURLLoad("add_frame_in_unload.html");
   web_view_helper_.InitializeAndLoad(base_url_ + "add_frame_in_unload.html",
                                      &frame_client);
   FrameTestHelpers::LoadFrame(web_view_helper_.WebView()->MainFrameImpl(),
@@ skipping 343 matching lines @@
 }
 
 TEST_P(WebViewTest, AutoResizeSubtreeLayout) {
   std::string url = RegisterMockedHttpURLLoad("subtree-layout.html");
   WebViewBase* web_view = web_view_helper_.Initialize();
 
   web_view->EnableAutoResizeMode(WebSize(200, 200), WebSize(200, 200));
   LoadFrame(web_view->MainFrameImpl(), url);
 
   LocalFrameView* frame_view =
-      web_view_helper_.WebView()->MainFrameImpl()->GetFrameView();
+      web_view_helper_.LocalMainFrame()->GetFrameView();
 
   // Auto-resizing used to DCHECK(needsLayout()) in LayoutBlockFlow::layout.
   // This EXPECT is merely a dummy. The real test is that we don't trigger
   // asserts in debug builds.
   EXPECT_FALSE(frame_view->NeedsLayout());
 };
 
 TEST_P(WebViewTest, PreferredSize) {
   std::string url = base_url_ + "specify_size.html?100px:100px";
   URLTestHelpers::RegisterMockedURLLoad(
@@ skipping 383 matching lines @@
 }
 
 // Make sure the SubframeBeforeUnloadUseCounter is only incremented on subframe
 // unloads. crbug.com/635029.
 TEST_P(WebViewTest, SubframeBeforeUnloadUseCounter) {
   RegisterMockedHttpURLLoad("visible_iframe.html");
   RegisterMockedHttpURLLoad("single_iframe.html");
   WebViewBase* web_view =
       web_view_helper_.InitializeAndLoad(base_url_ + "single_iframe.html");
 
-  WebLocalFrame* frame = web_view_helper_.WebView()->MainFrameImpl();
+  WebLocalFrame* frame = web_view_helper_.LocalMainFrame();
   Document* document =
       ToLocalFrame(web_view_helper_.WebView()->GetPage()->MainFrame())
           ->GetDocument();
 
   // Add a beforeunload handler in the main frame. Make sure firing
   // beforeunload doesn't increment the subframe use counter.
   {
     frame->ExecuteScript(
         WebScriptSource("addEventListener('beforeunload', function() {});"));
     web_view->MainFrame()->ToWebLocalFrame()->DispatchBeforeUnloadEvent(false);
@@ skipping 364 matching lines @@
   EXPECT_FALSE(frame_view->VisualViewportSuppliesScrollbars());
   if (RuntimeEnabledFeatures::RootLayerScrollingEnabled()) {
     EXPECT_NE(nullptr,
               frame_view->LayoutViewportScrollableArea()->VerticalScrollbar());
   } else {
     EXPECT_NE(nullptr, frame_view->VerticalScrollbar());
   }
 }
 
 }  // namespace blink