Move GetDocument method from WebFrame to WebLocalFrame.

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 1323 matching lines @@
   EXPECT_EQ(0, web_view->MainFrame()->GetScrollOffset().height);
 
   // Set up a composition from existing text that needs to be committed.
   Vector<CompositionUnderline> empty_underlines;
   WebLocalFrameBase* frame = web_view->MainFrameImpl();
   frame->GetFrame()->GetInputMethodController().SetCompositionFromExistingText(
       empty_underlines, 0, 3);
 
   // Scroll the input field out of the viewport.
   Element* element = static_cast<Element*>(
-      web_view->MainFrame()->GetDocument().GetElementById("btn"));
+      web_view->MainFrameImpl()->GetDocument().GetElementById("btn"));
   element->scrollIntoView();
   float offset_height = web_view->MainFrame()->GetScrollOffset().height;
   EXPECT_EQ(0, web_view->MainFrame()->GetScrollOffset().width);
   EXPECT_LT(0, offset_height);
 
   WebTextInputInfo info = frame->GetInputMethodController()->TextInputInfo();
   EXPECT_EQ("hello", std::string(info.value.Utf8().data()));
 
   // Verify that the input field is not scrolled back into the viewport.
   frame->FrameWidget()
@@ skipping 179 matching lines @@
 
 TEST_P(WebViewTest, SetCompositionFromExistingTextInRichText) {
   RegisterMockedHttpURLLoad("content_editable_rich_text.html");
   WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "content_editable_rich_text.html");
   web_view->SetInitialFocus(false);
   WebVector<WebCompositionUnderline> underlines(static_cast<size_t>(1));
   underlines[0] = WebCompositionUnderline(0, 4, 0, false, 0);
   WebLocalFrameBase* frame = web_view->MainFrameImpl();
   frame->SetEditableSelectionOffsets(1, 1);
-  WebDocument document = web_view->MainFrame()->GetDocument();
+  WebDocument document = web_view->MainFrameImpl()->GetDocument();
   EXPECT_FALSE(document.GetElementById("bold").IsNull());
   frame->SetCompositionFromExistingText(0, 4, underlines);
   EXPECT_FALSE(document.GetElementById("bold").IsNull());
 }
 
 TEST_P(WebViewTest, SetEditableSelectionOffsetsKeepsComposition) {
   RegisterMockedHttpURLLoad("input_field_populated.html");
   WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "input_field_populated.html");
   web_view->SetInitialFocus(false);
@@ skipping 400 matching lines @@
   const std::string foo_url = base_url_ + "foo.html";
   const std::string bar_url = base_url_ + "bar.html";
 
   WebViewBase* web_view = web_view_helper_.InitializeAndLoad(foo_url);
 
   ASSERT_TRUE(web_view);
 
   // Drag and drop barUrl and verify that we've navigated to it.
   DragAndDropURL(web_view, bar_url);
   EXPECT_EQ(bar_url,
-            web_view->MainFrame()->GetDocument().Url().GetString().Utf8());
+            web_view->MainFrameImpl()->GetDocument().Url().GetString().Utf8());
 
   // Drag and drop fooUrl and verify that we've navigated back to it.
   DragAndDropURL(web_view, foo_url);
   EXPECT_EQ(foo_url,
-            web_view->MainFrame()->GetDocument().Url().GetString().Utf8());
+            web_view->MainFrameImpl()->GetDocument().Url().GetString().Utf8());
 
   // Disable navigation on drag-and-drop.
   web_view->SettingsImpl()->SetNavigateOnDragDrop(false);
 
   // Attempt to drag and drop to barUrl and verify that no navigation has
   // occurred.
   DragAndDropURL(web_view, bar_url);
   EXPECT_EQ(foo_url,
-            web_view->MainFrame()->GetDocument().Url().GetString().Utf8());
+            web_view->MainFrameImpl()->GetDocument().Url().GetString().Utf8());
 }
 
 bool WebViewTest::TapElement(WebInputEvent::Type type, Element* element) {
   if (!element || !element->GetLayoutObject())
     return false;
 
   DCHECK(web_view_helper_.WebView());
   element->scrollIntoViewIfNeeded();
 
   // TODO(bokan): Technically incorrect, event positions should be in viewport
@@ skipping 15 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()->MainFrame()->GetDocument().GetElementById(
+      web_view_helper_.WebView()->MainFrameImpl()->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 =
@@ skipping 110 matching lines @@
   WebGestureEvent event(WebInputEvent::kGestureLongPress,
                         WebInputEvent::kNoModifiers,
                         WebInputEvent::kTimeStampForTesting);
   event.source_device = kWebGestureDeviceTouchscreen;
   event.x = 10;
   event.y = 10;
 
   EXPECT_NE(WebInputEventResult::kHandledSystem,
             web_view->HandleInputEvent(WebCoalescedInputEvent(event)));
 
-  HTMLElement* element =
-      ToHTMLElement(web_view->MainFrame()->GetDocument().GetElementById("obj"));
+  HTMLElement* element = ToHTMLElement(
+      web_view->MainFrameImpl()->GetDocument().GetElementById("obj"));
   EXPECT_FALSE(element->CanStartSelection());
 }
 
 TEST_P(WebViewTest, LongPressObjectFallback) {
   RegisterMockedHttpURLLoad("long_press_object_fallback.html");
 
   WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "long_press_object_fallback.html", true);
   web_view->SettingsImpl()->SetAlwaysShowContextMenuOnTouch(true);
   web_view->Resize(WebSize(500, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebGestureEvent event(WebInputEvent::kGestureLongPress,
                         WebInputEvent::kNoModifiers,
                         WebInputEvent::kTimeStampForTesting);
   event.source_device = kWebGestureDeviceTouchscreen;
   event.x = 10;
   event.y = 10;
 
   EXPECT_EQ(WebInputEventResult::kHandledSystem,
             web_view->HandleInputEvent(WebCoalescedInputEvent(event)));
 
-  HTMLElement* element =
-      ToHTMLElement(web_view->MainFrame()->GetDocument().GetElementById("obj"));
+  HTMLElement* element = ToHTMLElement(
+      web_view->MainFrameImpl()->GetDocument().GetElementById("obj"));
   EXPECT_TRUE(element->CanStartSelection());
 }
 
 TEST_P(WebViewTest, LongPressImage) {
   RegisterMockedHttpURLLoad("long_press_image.html");
 
   WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "long_press_image.html", true);
   web_view->SettingsImpl()->SetAlwaysShowContextMenuOnTouch(false);
   web_view->Resize(WebSize(500, 300));
@@ skipping 65 matching lines @@
   web_view->SettingsImpl()->SetTouchDragDropEnabled(true);
   web_view->Resize(WebSize(500, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   WebString anchor_tag_id = WebString::FromUTF8("anchorTag");
   WebString image_tag_id = WebString::FromUTF8("imageTag");
 
   EXPECT_TRUE(TapElementById(WebInputEvent::kGestureLongPress, anchor_tag_id));
   EXPECT_STREQ("anchor contextmenu",
-               web_view->MainFrame()->GetDocument().Title().Utf8().data());
+               web_view->MainFrameImpl()->GetDocument().Title().Utf8().data());
 
   EXPECT_TRUE(TapElementById(WebInputEvent::kGestureLongPress, image_tag_id));
   EXPECT_STREQ("image contextmenu",
-               web_view->MainFrame()->GetDocument().Title().Utf8().data());
+               web_view->MainFrameImpl()->GetDocument().Title().Utf8().data());
 }
 
 TEST_P(WebViewTest, LongPressEmptyEditableSelection) {
   RegisterMockedHttpURLLoad("long_press_empty_editable_selection.html");
 
   WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "long_press_empty_editable_selection.html", true);
   web_view->SettingsImpl()->SetAlwaysShowContextMenuOnTouch(false);
   web_view->Resize(WebSize(500, 300));
   web_view->UpdateAllLifecyclePhases();
@@ skipping 110 matching lines @@
                         WebInputEvent::kTimeStampForTesting);
   event.source_device = kWebGestureDeviceTouchscreen;
   event.x = 100;
   event.y = 25;
   event.data.tap.tap_count = 2;
 
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   EXPECT_TRUE(frame->SelectionAsText().IsEmpty());
 
   HTMLTextAreaElement* text_area_element = toHTMLTextAreaElement(
-      web_view->MainFrame()->GetDocument().GetElementById(blanklinestextbox));
+      web_view->MainFrameImpl()->GetDocument().GetElementById(
+          blanklinestextbox));
   text_area_element->setValue("hello");
 
   // Long-press past last word of textbox.
   EXPECT_TRUE(
       TapElementById(WebInputEvent::kGestureLongPress, blanklinestextbox));
   EXPECT_TRUE(frame->SelectionAsText().IsEmpty());
 
   // Double-tap past last word of textbox.
   web_view->HandleInputEvent(WebCoalescedInputEvent(event));
   EXPECT_TRUE(frame->SelectionAsText().IsEmpty());
@@ skipping 338 matching lines @@
 
 TEST_P(WebViewTest, FinishComposingTextDoesntTriggerAutofillTextChange) {
   RegisterMockedHttpURLLoad("input_field_populated.html");
   MockAutofillClient client;
   WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "input_field_populated.html");
   WebLocalFrameBase* frame = web_view->MainFrameImpl();
   frame->SetAutofillClient(&client);
   web_view->SetInitialFocus(false);
 
-  WebDocument document = web_view->MainFrame()->GetDocument();
+  WebDocument document = web_view->MainFrameImpl()->GetDocument();
   HTMLFormControlElement* form =
       ToHTMLFormControlElement(document.GetElementById("sample"));
 
   WebInputMethodController* active_input_method_controller =
       frame->FrameWidget()->GetActiveWebInputMethodController();
   // Set up a composition that needs to be committed.
   std::string composition_text("testingtext");
 
   WebVector<WebCompositionUnderline> empty_underlines;
   active_input_method_controller->SetComposition(
@@ skipping 34 matching lines @@
   frame->SetCompositionFromExistingText(8, 12, empty_underlines);
 
   WebTextInputInfo info = frame->GetInputMethodController()->TextInputInfo();
   EXPECT_EQ("0123456789abcdefghijklmnopqrstuvwxyz",
             std::string(info.value.Utf8().data()));
   EXPECT_EQ(8, info.composition_start);
   EXPECT_EQ(12, info.composition_end);
 
   EXPECT_EQ(0, client.TextChanges());
 
-  WebDocument document = web_view->MainFrame()->GetDocument();
+  WebDocument document = web_view->MainFrameImpl()->GetDocument();
   EXPECT_EQ(WebString::FromUTF8("none"),
             document.GetElementById("inputEvent").FirstChild().NodeValue());
 
   frame->SetAutofillClient(0);
 }
 
 class ViewCreatingWebViewClient : public FrameTestHelpers::TestWebViewClient {
  public:
   ViewCreatingWebViewClient() : did_focus_called_(false) {}
 
@@ skipping 66 matching lines @@
                    ->MainFrame())
       ->Loader()
       .Load(request_with_target_start);
   EXPECT_TRUE(client.DidFocusCalled());
 
   web_view_helper.Reset();  // Remove dependency on locally scoped client.
 }
 
 TEST_P(WebViewTest, DispatchesFocusOutFocusInOnViewToggleFocus) {
   RegisterMockedHttpURLLoad("focusout_focusin_events.html");
-  WebView* web_view = web_view_helper_.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "focusout_focusin_events.html", true, 0);
 
   web_view->SetFocus(true);
   web_view->SetFocus(false);
   web_view->SetFocus(true);
 
   WebElement element =
-      web_view->MainFrame()->GetDocument().GetElementById("message");
+      web_view->MainFrameImpl()->GetDocument().GetElementById("message");
   EXPECT_STREQ("focusoutfocusin", element.TextContent().Utf8().data());
 }
 
 TEST_P(WebViewTest, DispatchesDomFocusOutDomFocusInOnViewToggleFocus) {
   RegisterMockedHttpURLLoad("domfocusout_domfocusin_events.html");
-  WebView* web_view = web_view_helper_.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "domfocusout_domfocusin_events.html", true, 0);
 
   web_view->SetFocus(true);
   web_view->SetFocus(false);
   web_view->SetFocus(true);
 
   WebElement element =
-      web_view->MainFrame()->GetDocument().GetElementById("message");
+      web_view->MainFrameImpl()->GetDocument().GetElementById("message");
   EXPECT_STREQ("DOMFocusOutDOMFocusIn", element.TextContent().Utf8().data());
 }
 
 static void OpenDateTimeChooser(WebView* web_view,
                                 HTMLInputElement* input_element) {
   input_element->focus();
 
   WebKeyboardEvent key_event(WebInputEvent::kRawKeyDown,
                              WebInputEvent::kNoModifiers,
                              WebInputEvent::kTimeStampForTesting);
@@ skipping 86 matching lines @@
 
   // Clear the WebViewClient from the webViewHelper to avoid use-after-free in
   // the WebViewHelper destructor.
   web_view_helper_.Reset();
   RuntimeEnabledFeatures::SetInputMultipleFieldsUIEnabled(
       original_multiple_fields_flag);
 }
 
 TEST_P(WebViewTest, DispatchesFocusBlurOnViewToggle) {
   RegisterMockedHttpURLLoad("focus_blur_events.html");
-  WebView* web_view = web_view_helper_.InitializeAndLoad(
+  WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
       base_url_ + "focus_blur_events.html", true, 0);
 
   web_view->SetFocus(true);
   web_view->SetFocus(false);
   web_view->SetFocus(true);
 
   WebElement element =
-      web_view->MainFrame()->GetDocument().GetElementById("message");
+      web_view->MainFrameImpl()->GetDocument().GetElementById("message");
   // Expect not to see duplication of events.
   EXPECT_STREQ("blurfocus", element.TextContent().Utf8().data());
 }
 
 TEST_P(WebViewTest, SmartClipData) {
   static const char kExpectedClipText[] = "\nPrice 10,000,000won";
   static const char kExpectedClipHtml[] =
       "<div id=\"div4\" style=\"padding: 10px; margin: 10px; border: 2px "
       "solid skyblue; float: left; width: 190px; height: 30px; "
       "color: rgb(0, 0, 0); font-family: myahem; font-size: 8px; font-style: "
@@ skipping 565 matching lines @@
   web_view->SetZoomLevel(WebView::ZoomFactorToZoomLevel(1));
   size = web_view->ContentsPreferredMinimumSize();
   EXPECT_EQ(2, size.width);
   EXPECT_EQ(2, size.height);
 }
 
 TEST_P(WebViewTest, PreferredSizeDirtyLayout) {
   std::string url = base_url_ + "specify_size.html?100px:100px";
   URLTestHelpers::RegisterMockedURLLoad(
       ToKURL(url), testing::WebTestDataPath("specify_size.html"));
-  WebView* web_view = web_view_helper_.InitializeAndLoad(url, true);
+  WebViewBase* web_view = web_view_helper_.InitializeAndLoad(url, true);
   WebElement document_element =
-      web_view->MainFrame()->GetDocument().DocumentElement();
+      web_view->MainFrameImpl()->GetDocument().DocumentElement();
 
   WebSize size = web_view->ContentsPreferredMinimumSize();
   EXPECT_EQ(100, size.width);
   EXPECT_EQ(100, size.height);
 
   document_element.SetAttribute("style", "display: none");
 
   size = web_view->ContentsPreferredMinimumSize();
   EXPECT_EQ(0, size.width);
   EXPECT_EQ(0, size.height);
@@ skipping 27 matching lines @@
   IntPoint tapped_position_;
   WebNode tapped_node_;
   bool page_changed_ = false;
 };
 
 TEST_P(WebViewTest, ShowUnhandledTapUIIfNeeded) {
   std::string test_file = "show_unhandled_tap.html";
   RegisterMockedHttpURLLoad("Ahem.ttf");
   RegisterMockedHttpURLLoad(test_file);
   UnhandledTapWebViewClient client;
-  WebView* web_view = web_view_helper_.InitializeAndLoad(base_url_ + test_file,
-                                                         true, 0, &client);
+  WebViewBase* web_view = web_view_helper_.InitializeAndLoad(
+      base_url_ + test_file, true, 0, &client);
   web_view->Resize(WebSize(500, 300));
   web_view->UpdateAllLifecyclePhases();
   RunPendingTasks();
 
   // Scroll the bottom into view so we can distinguish window coordinates from
   // document coordinates.
   EXPECT_TRUE(TapElementById(WebInputEvent::kGestureTap,
                              WebString::FromUTF8("bottom")));
   EXPECT_TRUE(client.GetWasCalled());
   EXPECT_EQ(64, client.GetTappedXPos());
   EXPECT_EQ(278, client.GetTappedYPos());
   EXPECT_FALSE(client.IsTappedNodeNull());
   EXPECT_TRUE(client.GetWebNode().IsTextNode());
 
   // Test basic tap handling and notification.
   client.Reset();
   EXPECT_TRUE(TapElementById(WebInputEvent::kGestureTap,
                              WebString::FromUTF8("target")));
   EXPECT_TRUE(client.GetWasCalled());
   EXPECT_EQ(144, client.GetTappedXPos());
   EXPECT_EQ(82, client.GetTappedYPos());
   EXPECT_FALSE(client.IsTappedNodeNull());
   EXPECT_TRUE(client.GetWebNode().IsTextNode());
   // Make sure the returned text node has the parent element that was our
   // target.
-  EXPECT_EQ(web_view->MainFrame()->GetDocument().GetElementById("target"),
+  EXPECT_EQ(web_view->MainFrameImpl()->GetDocument().GetElementById("target"),
             client.GetWebNode().ParentNode());
 
   // Test correct conversion of coordinates to viewport space under pinch-zoom.
   web_view->SetPageScaleFactor(2);
   web_view->SetVisualViewportOffset(WebFloatPoint(50, 20));
   client.Reset();
   EXPECT_TRUE(TapElementById(WebInputEvent::kGestureTap,
                              WebString::FromUTF8("target")));
   EXPECT_TRUE(client.GetWasCalled());
   EXPECT_EQ(188, client.GetTappedXPos());
@@ skipping 641 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