Move GetDocument method from WebFrame to WebLocalFrame.

content/shell/test_runner/pixel_dump.cc

 // Copyright 2016 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 "content/shell/test_runner/pixel_dump.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/bind.h"
@@ skipping 57 matching lines @@
   gfx::Point popup_position_;
 };
 
 void DrawSelectionRect(const PixelsDumpRequest& dump_request,
                        cc::PaintCanvas* canvas) {
   // See if we need to draw the selection bounds rect. Selection bounds
   // rect is the rect enclosing the (possibly transformed) selection.
   // The rect should be drawn after everything is laid out and painted.
   if (!dump_request.layout_test_runtime_flags.dump_selection_rect())
     return;
-  // If there is a selection rect - draw a red 1px border enclosing rect
+
+  // TODO(lukasza): https://crbug.com/667551: Support OOPIFs in pixel dumps.
   CHECK(dump_request.web_view->MainFrame()->IsWebLocalFrame())
       << "This function cannot be called if the main frame is not a "
          "local frame.";
+
+  // If there is a selection rect - draw a red 1px border enclosing rect
   blink::WebRect wr = dump_request.web_view->MainFrame()
                           ->ToWebLocalFrame()
                           ->GetSelectionBoundsRectForTesting();
   if (wr.IsEmpty())
     return;
   // Render a red rectangle bounding selection rect
   cc::PaintFlags flags;
   flags.setColor(0xFFFF0000);  // Fully opaque red
   flags.setStyle(cc::PaintFlags::kStroke_Style);
   flags.setAntiAlias(true);
   flags.setStrokeWidth(1.0f);
   SkIRect rect;  // Bounding rect
   rect.set(wr.x, wr.y, wr.x + wr.width, wr.y + wr.height);
   canvas->drawIRect(rect, flags);
 }
 
 void CapturePixelsForPrinting(std::unique_ptr<PixelsDumpRequest> dump_request) {
   dump_request->web_view->UpdateAllLifecyclePhases();
 
   blink::WebSize page_size_in_pixels = dump_request->web_view->Size();
 
+  // TODO(lukasza): https://crbug.com/667551: Support OOPIFs in pixel dumps.
   CHECK(dump_request->web_view->MainFrame()->IsWebLocalFrame())
       << "This function cannot be called if the main frame is not a "
          "local frame.";
   blink::WebLocalFrame* web_frame =
       dump_request->web_view->MainFrame()->ToWebLocalFrame();
 
   int page_count = web_frame->PrintBegin(page_size_in_pixels);
   int totalHeight = page_count * (page_size_in_pixels.height + 1) - 1;
 
   bool is_opaque = false;
@@ skipping 103 matching lines @@
     return;
   }
 
   blink::WebImage image = static_cast<blink::WebMockClipboard*>(
                               blink::Platform::Current()->Clipboard())
                               ->ReadRawImage(blink::WebClipboard::Buffer());
   callback.Run(image.GetSkBitmap());
 }
 
 }  // namespace test_runner