Move GetDocument method from WebFrame to WebLocalFrame.

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

 /*
  * Copyright (C) 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 148 matching lines @@
 
  private:
   bool printed_page_ = false;
 };
 
 void TestPlugin::PrintPage(int page_number, WebCanvas* canvas) {
   DCHECK(test_client_);
   test_client_->OnPrintPage();
 }
 
-WebPluginContainer* GetWebPluginContainer(WebView* web_view,
+WebPluginContainer* GetWebPluginContainer(WebViewBase* web_view,
                                           const WebString& id) {
-  WebElement element = web_view->MainFrame()->GetDocument().GetElementById(id);
+  WebElement element =
+      web_view->MainFrameImpl()->GetDocument().GetElementById(id);
   return element.PluginContainer();
 }
 
 }  // namespace
 
 TEST_F(WebPluginContainerTest, WindowToLocalPointTest) {
   RegisterMockedURL("plugin_container.html");
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebPluginContainer* plugin_container_one =
       GetWebPluginContainer(web_view, WebString::FromUTF8("translated-plugin"));
   DCHECK(plugin_container_one);
@@ skipping 18 matching lines @@
   ASSERT_EQ(10, point4.x);
   ASSERT_EQ(10, point4.y);
 }
 
 TEST_F(WebPluginContainerTest, PluginDocumentPluginIsFocused) {
   RegisterMockedURL("test.pdf", "application/pdf");
 
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "test.pdf", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->UpdateAllLifecyclePhases();
 
-  WebDocument document = web_view->MainFrame()->GetDocument();
+  WebDocument document = web_view->MainFrameImpl()->GetDocument();
   EXPECT_TRUE(document.IsPluginDocument());
   WebPluginContainer* plugin_container =
       GetWebPluginContainer(web_view, "plugin");
   EXPECT_EQ(document.FocusedElement(), plugin_container->GetElement());
 }
 
 TEST_F(WebPluginContainerTest, IFramePluginDocumentNotFocused) {
   RegisterMockedURL("test.pdf", "application/pdf");
   RegisterMockedURL("iframe_pdf.html", "text/html");
 
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "iframe_pdf.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->UpdateAllLifecyclePhases();
 
-  WebDocument document = web_view->MainFrame()->GetDocument();
-  WebFrame* iframe = web_view->MainFrame()->FirstChild();
+  WebDocument document = web_view->MainFrameImpl()->GetDocument();
+  WebLocalFrame* iframe =
+      web_view->MainFrame()->FirstChild()->ToWebLocalFrame();
   EXPECT_TRUE(iframe->GetDocument().IsPluginDocument());
   WebPluginContainer* plugin_container =
       iframe->GetDocument().GetElementById("plugin").PluginContainer();
   EXPECT_NE(document.FocusedElement(), plugin_container->GetElement());
   EXPECT_NE(iframe->GetDocument().FocusedElement(),
             plugin_container->GetElement());
 }
 
 TEST_F(WebPluginContainerTest, PrintOnePage) {
   RegisterMockedURL("test.pdf", "application/pdf");
@@ skipping 41 matching lines @@
   frame->PrintPagesForTesting(recorder.beginRecording(IntRect()), WebSize());
   frame->PrintEnd();
   DCHECK(plugin_web_frame_client.PrintedAtLeastOnePage());
 }
 
 TEST_F(WebPluginContainerTest, LocalToWindowPointTest) {
   RegisterMockedURL("plugin_container.html");
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebPluginContainer* plugin_container_one =
       GetWebPluginContainer(web_view, WebString::FromUTF8("translated-plugin"));
   DCHECK(plugin_container_one);
@@ skipping 17 matching lines @@
   ASSERT_EQ(-10, point4.x);
   ASSERT_EQ(10, point4.y);
 }
 
 // Verifies executing the command 'Copy' results in copying to the clipboard.
 TEST_F(WebPluginContainerTest, Copy) {
   RegisterMockedURL("plugin_container.html");
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
-  web_view->MainFrame()
+  web_view->MainFrameImpl()
       ->GetDocument()
       .Unwrap<Document>()
       ->body()
       ->getElementById("translated-plugin")
       ->focus();
   EXPECT_TRUE(web_view->MainFrame()->ToWebLocalFrame()->ExecuteCommand("Copy"));
   EXPECT_EQ(WebString("x"), Platform::Current()->Clipboard()->ReadPlainText(
                                 WebClipboard::Buffer()));
 }
 
 TEST_F(WebPluginContainerTest, CopyFromContextMenu) {
   RegisterMockedURL("plugin_container.html");
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   auto event = FrameTestHelpers::CreateMouseEvent(WebMouseEvent::kMouseDown,
                                                   WebMouseEvent::Button::kRight,
                                                   WebPoint(30, 30), 0);
@@ skipping 10 matching lines @@
   EXPECT_EQ(WebString(""), Platform::Current()->Clipboard()->ReadPlainText(
                                WebClipboard::Buffer()));
 
   // Now, let's try a more complex scenario:
   // 1) open the context menu. This will focus the plugin.
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   // 2) document blurs the plugin, because it can.
   web_view->ClearFocusedElement();
   // 3) Copy should still operate on the context node, even though the focus had
   //    shifted.
-  EXPECT_TRUE(web_view->MainFrame()->ToWebLocalFrame()->ExecuteCommand("Copy"));
+  EXPECT_TRUE(web_view->MainFrameImpl()->ExecuteCommand("Copy"));
   EXPECT_EQ(WebString("x"), Platform::Current()->Clipboard()->ReadPlainText(
                                 WebClipboard::Buffer()));
 }
 
 // Verifies |Ctrl-C| and |Ctrl-Insert| keyboard events, results in copying to
 // the clipboard.
 TEST_F(WebPluginContainerTest, CopyInsertKeyboardEventsTest) {
   RegisterMockedURL("plugin_container.html");
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("translated-plugin"));
   WebInputEvent::Modifiers modifier_key = static_cast<WebInputEvent::Modifiers>(
       WebInputEvent::kControlKey | WebInputEvent::kNumLockOn |
       WebInputEvent::kIsLeft);
 #if OS(MACOSX)
   modifier_key = static_cast<WebInputEvent::Modifiers>(
       WebInputEvent::kMetaKey | WebInputEvent::kNumLockOn |
       WebInputEvent::kIsLeft);
 #endif
   WebKeyboardEvent web_keyboard_event_c(WebInputEvent::kRawKeyDown,
@@ skipping 67 matching lines @@
   size_t coalesced_event_count_;
   WebInputEvent::Type last_event_type_;
   IntPoint last_event_location_;
 };
 
 TEST_F(WebPluginContainerTest, GestureLongPressReachesPlugin) {
   RegisterMockedURL("plugin_container.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("translated-plugin"));
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebGestureEvent event(WebInputEvent::kGestureLongPress,
                         WebInputEvent::kNoModifiers,
                         WebInputEvent::kTimeStampForTesting);
   event.source_device = kWebGestureDeviceTouchscreen;
@@ skipping 19 matching lines @@
 
   EXPECT_EQ(WebInputEvent::kGestureLongPress,
             test_plugin->GetLastInputEventType());
 }
 
 TEST_F(WebPluginContainerTest, MouseWheelEventTranslated) {
   RegisterMockedURL("plugin_container.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("translated-plugin"));
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebMouseWheelEvent event(WebInputEvent::kMouseWheel,
                            WebInputEvent::kNoModifiers,
                            WebInputEvent::kTimeStampForTesting);
 
   WebRect rect = plugin_container_one_element.BoundsInViewport();
   event.SetPositionInWidget(rect.x + rect.width / 2, rect.y + rect.height / 2);
 
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   RunPendingTasks();
 
   EXPECT_EQ(WebInputEvent::kMouseWheel, test_plugin->GetLastInputEventType());
   EXPECT_EQ(rect.width / 2, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 2, test_plugin->GetLastEventLocation().Y());
 }
 
 TEST_F(WebPluginContainerTest, TouchEventScrolled) {
   RegisterMockedURL("plugin_scroll.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_scroll.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
   web_view->SmoothScroll(0, 200, 0);
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("scrolled-plugin"));
   plugin_container_one_element.PluginContainer()->RequestTouchEventType(
       WebPluginContainer::kTouchEventRequestTypeRaw);
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebTouchEvent event(WebInputEvent::kTouchStart, WebInputEvent::kNoModifiers,
                       WebInputEvent::kTimeStampForTesting);
   event.touches_length = 1;
   WebRect rect = plugin_container_one_element.BoundsInViewport();
   event.touches[0].state = WebTouchPoint::kStatePressed;
   event.touches[0].SetPositionInWidget(rect.x + rect.width / 2,
                                        rect.y + rect.height / 2);
 
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   RunPendingTasks();
 
   EXPECT_EQ(WebInputEvent::kTouchStart, test_plugin->GetLastInputEventType());
   EXPECT_EQ(rect.width / 2, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 2, test_plugin->GetLastEventLocation().Y());
 }
 
 TEST_F(WebPluginContainerTest, TouchEventScrolledWithCoalescedTouches) {
   RegisterMockedURL("plugin_scroll.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_scroll.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
   web_view->SmoothScroll(0, 200, 0);
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("scrolled-plugin"));
   plugin_container_one_element.PluginContainer()->RequestTouchEventType(
       WebPluginContainer::kTouchEventRequestTypeRawLowLatency);
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebTouchEvent event(WebInputEvent::kTouchStart, WebInputEvent::kNoModifiers,
                       WebInputEvent::kTimeStampForTesting);
@@ skipping 25 matching lines @@
   EXPECT_EQ(WebInputEvent::kTouchStart, test_plugin->GetLastInputEventType());
   EXPECT_EQ(rect.width / 2, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 2, test_plugin->GetLastEventLocation().Y());
 }
 
 TEST_F(WebPluginContainerTest, MouseWheelEventScrolled) {
   RegisterMockedURL("plugin_scroll.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_scroll.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
   web_view->SmoothScroll(0, 200, 0);
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("scrolled-plugin"));
   plugin_container_one_element.PluginContainer()->RequestTouchEventType(
       WebPluginContainer::kTouchEventRequestTypeRaw);
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebMouseWheelEvent event(WebInputEvent::kMouseWheel,
                            WebInputEvent::kNoModifiers,
                            WebInputEvent::kTimeStampForTesting);
 
   WebRect rect = plugin_container_one_element.BoundsInViewport();
   event.SetPositionInWidget(rect.x + rect.width / 2, rect.y + rect.height / 2);
 
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   RunPendingTasks();
 
   EXPECT_EQ(WebInputEvent::kMouseWheel, test_plugin->GetLastInputEventType());
   EXPECT_EQ(rect.width / 2, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 2, test_plugin->GetLastEventLocation().Y());
 }
 
 TEST_F(WebPluginContainerTest, MouseEventScrolled) {
   RegisterMockedURL("plugin_scroll.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_scroll.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
   web_view->SmoothScroll(0, 200, 0);
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("scrolled-plugin"));
   plugin_container_one_element.PluginContainer()->RequestTouchEventType(
       WebPluginContainer::kTouchEventRequestTypeRaw);
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebMouseEvent event(WebInputEvent::kMouseMove, WebInputEvent::kNoModifiers,
                       WebInputEvent::kTimeStampForTesting);
 
   WebRect rect = plugin_container_one_element.BoundsInViewport();
   event.SetPositionInWidget(rect.x + rect.width / 2, rect.y + rect.height / 2);
 
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   RunPendingTasks();
 
   EXPECT_EQ(WebInputEvent::kMouseMove, test_plugin->GetLastInputEventType());
   EXPECT_EQ(rect.width / 2, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 2, test_plugin->GetLastEventLocation().Y());
 }
 
 TEST_F(WebPluginContainerTest, MouseEventZoomed) {
   RegisterMockedURL("plugin_scroll.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_scroll.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->SetPageScaleFactor(2);
   web_view->SmoothScroll(0, 300, 0);
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("scrolled-plugin"));
   plugin_container_one_element.PluginContainer()->RequestTouchEventType(
       WebPluginContainer::kTouchEventRequestTypeRaw);
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebMouseEvent event(WebInputEvent::kMouseMove, WebInputEvent::kNoModifiers,
                       WebInputEvent::kTimeStampForTesting);
 
   WebRect rect = plugin_container_one_element.BoundsInViewport();
   event.SetPositionInWidget(rect.x + rect.width / 2, rect.y + rect.height / 2);
 
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   RunPendingTasks();
 
   // rect.width/height divided by 4 because the rect is in viewport bounds and
   // there is a scale of 2 set.
   EXPECT_EQ(WebInputEvent::kMouseMove, test_plugin->GetLastInputEventType());
   EXPECT_EQ(rect.width / 4, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 4, test_plugin->GetLastEventLocation().Y());
 }
 
 TEST_F(WebPluginContainerTest, MouseWheelEventZoomed) {
   RegisterMockedURL("plugin_scroll.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_scroll.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->SetPageScaleFactor(2);
   web_view->SmoothScroll(0, 300, 0);
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("scrolled-plugin"));
   plugin_container_one_element.PluginContainer()->RequestTouchEventType(
       WebPluginContainer::kTouchEventRequestTypeRaw);
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebMouseWheelEvent event(WebInputEvent::kMouseWheel,
                            WebInputEvent::kNoModifiers,
@@ skipping 10 matching lines @@
   EXPECT_EQ(WebInputEvent::kMouseWheel, test_plugin->GetLastInputEventType());
   EXPECT_EQ(rect.width / 4, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 4, test_plugin->GetLastEventLocation().Y());
 }
 
 TEST_F(WebPluginContainerTest, TouchEventZoomed) {
   RegisterMockedURL("plugin_scroll.html");
   CustomPluginWebFrameClient<EventTestPlugin>
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_scroll.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->SetPageScaleFactor(2);
   web_view->SmoothScroll(0, 300, 0);
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("scrolled-plugin"));
   plugin_container_one_element.PluginContainer()->RequestTouchEventType(
       WebPluginContainer::kTouchEventRequestTypeRaw);
   WebPlugin* plugin = static_cast<WebPluginContainerBase*>(
                           plugin_container_one_element.PluginContainer())
                           ->Plugin();
   EventTestPlugin* test_plugin = static_cast<EventTestPlugin*>(plugin);
 
   WebTouchEvent event(WebInputEvent::kTouchStart, WebInputEvent::kNoModifiers,
                       WebInputEvent::kTimeStampForTesting);
@@ skipping 13 matching lines @@
   EXPECT_EQ(rect.width / 4, test_plugin->GetLastEventLocation().X());
   EXPECT_EQ(rect.height / 4, test_plugin->GetLastEventLocation().Y());
 }
 
 // Verify that isRectTopmost returns false when the document is detached.
 TEST_F(WebPluginContainerTest, IsRectTopmostTest) {
   RegisterMockedURL("plugin_container.html");
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebPluginContainerBase* plugin_container_impl =
       ToWebPluginContainerBase(GetWebPluginContainer(
           web_view, WebString::FromUTF8("translated-plugin")));
@@ skipping 25 matching lines @@
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
   WebView* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_containing_page.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
-  WebElement plugin_element =
-      web_view->MainFrame()->FirstChild()->GetDocument().GetElementById(
-          "translated-plugin");
+  WebElement plugin_element = web_view->MainFrame()
+                                  ->FirstChild()
+                                  ->ToWebLocalFrame()
+                                  ->GetDocument()
+                                  .GetElementById("translated-plugin");
   WebPluginContainerBase* plugin_container_impl =
       ToWebPluginContainerBase(plugin_element.PluginContainer());
 
   DCHECK(plugin_container_impl);
 
   IntRect window_rect, clip_rect, unobscured_rect;
   CalculateGeometry(plugin_container_impl, window_rect, clip_rect,
                     unobscured_rect);
   EXPECT_RECT_EQ(IntRect(20, 220, 40, 40), window_rect);
   EXPECT_RECT_EQ(IntRect(0, 0, 40, 40), clip_rect);
   EXPECT_RECT_EQ(IntRect(0, 0, 40, 40), unobscured_rect);
 
   // Cause the plugin's frame to be detached.
   web_view_helper.Reset();
 }
 
 TEST_F(WebPluginContainerTest, ClippedRectsForSubpixelPositionedPlugin) {
   RegisterMockedURL("plugin_container.html");
 
   TestPluginWebFrameClient
       plugin_web_frame_client;  // Must outlive webViewHelper.
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           "subpixel-positioned-plugin");
   WebPluginContainerBase* plugin_container_impl =
       ToWebPluginContainerBase(plugin_element.PluginContainer());
 
   DCHECK(plugin_container_impl);
 
   IntRect window_rect, clip_rect, unobscured_rect;
   CalculateGeometry(plugin_container_impl, window_rect, clip_rect,
                     unobscured_rect);
   EXPECT_RECT_EQ(IntRect(0, 0, 40, 40), window_rect);
@@ skipping 19 matching lines @@
       // In destroy, IsRectTopmost is no longer valid.
       EXPECT_FALSE(Container()->IsRectTopmost(topmost_rect));
       FakeWebPlugin::Destroy();
     }
   };
 
   RegisterMockedURL("plugin_container.html");
   // The client must outlive WebViewHelper.
   CustomPluginWebFrameClient<TopmostPlugin> plugin_web_frame_client;
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebPluginContainerBase* plugin_container_impl =
       ToWebPluginContainerBase(GetWebPluginContainer(
           web_view, WebString::FromUTF8("translated-plugin")));
@@ skipping 39 matching lines @@
   std::unique_ptr<WebLayer> layer_;
 };
 
 }  // namespace
 
 TEST_F(WebPluginContainerTest, CompositedPluginSPv2) {
   ScopedSlimmingPaintV2ForTest enable_s_pv2(true);
   RegisterMockedURL("plugin.html");
   CustomPluginWebFrameClient<CompositedPlugin> web_frame_client;
   FrameTestHelpers::WebViewHelper web_view_helper;
-  WebView* web_view = web_view_helper.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin.html", &web_frame_client);
   ASSERT_TRUE(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(800, 600));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebPluginContainerBase* container = static_cast<WebPluginContainerBase*>(
       GetWebPluginContainer(web_view, WebString::FromUTF8("plugin")));
   ASSERT_TRUE(container);
@@ skipping 30 matching lines @@
   FrameTestHelpers::WebViewHelper web_view_helper;
   WebViewBase* web_view = web_view_helper.InitializeAndLoad(
       base_url_ + "plugin_container.html", &plugin_web_frame_client);
   DCHECK(web_view);
   web_view->GetSettings()->SetPluginsEnabled(true);
   web_view->Resize(WebSize(300, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebElement plugin_container_one_element =
-      web_view->MainFrame()->GetDocument().GetElementById(
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
           WebString::FromUTF8("translated-plugin"));
   plugin_container_one_element.PluginContainer()->SetWantsWheelEvents(true);
 
   RunPendingTasks();
   EXPECT_TRUE(web_view->GetPage()->GetEventHandlerRegistry().HasEventHandlers(
       EventHandlerRegistry::kWheelEventBlocking));
 }
 
 }  // namespace blink