In-place modification of drag related functionality present in LocalFrame

third_party/WebKit/Source/core/frame/LocalFrameTest.cpp

 // 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 "core/frame/LocalFrame.h"
 
 #include "core/editing/FrameSelection.h"
 #include "core/frame/LocalFrameView.h"
 #include "core/frame/PerformanceMonitor.h"
 #include "core/frame/VisualViewport.h"
@@ skipping 34 matching lines @@
   Persistent<Performance> performance_;
 };
 
 TEST_F(LocalFrameTest, nodeImage) {
   SetBodyContent(
       "<style>"
       "#sample { width: 100px; height: 100px; }"
       "</style>"
       "<div id=sample></div>");
   Element* sample = GetDocument().getElementById("sample");
-  const std::unique_ptr<DragImage> image = GetFrame().NodeImage(*sample);
+  const std::unique_ptr<DragImage> image =
+      GetFrame().NodeImage(GetFrame(), *sample);
   EXPECT_EQ(IntSize(100, 100), image->Size());
 }
 
 TEST_F(LocalFrameTest, nodeImageWithNestedElement) {
   SetBodyContent(
       "<style>"
       "div { -webkit-user-drag: element }"
       "span:-webkit-drag { color: #0F0 }"
       "</style>"
       "<div id=sample><span>Green when dragged</span></div>");
   Element* sample = GetDocument().getElementById("sample");
-  const std::unique_ptr<DragImage> image = GetFrame().NodeImage(*sample);
+  const std::unique_ptr<DragImage> image =
+      GetFrame().NodeImage(GetFrame(), *sample);
   EXPECT_EQ(
       Color(0, 255, 0),
       sample->firstChild()->GetLayoutObject()->ResolveColor(CSSPropertyColor))
       << "Descendants node should have :-webkit-drag.";
 }
 
 TEST_F(LocalFrameTest, nodeImageWithPsuedoClassWebKitDrag) {
   SetBodyContent(
       "<style>"
       "#sample { width: 100px; height: 100px; }"
       "#sample:-webkit-drag { width: 200px; height: 200px; }"
       "</style>"
       "<div id=sample></div>");
   Element* sample = GetDocument().getElementById("sample");
-  const std::unique_ptr<DragImage> image = GetFrame().NodeImage(*sample);
+  const std::unique_ptr<DragImage> image =
+      GetFrame().NodeImage(GetFrame(), *sample);
   EXPECT_EQ(IntSize(200, 200), image->Size())
       << ":-webkit-drag should affect dragged image.";
 }
 
 TEST_F(LocalFrameTest, nodeImageWithoutDraggedLayoutObject) {
   SetBodyContent(
       "<style>"
       "#sample { width: 100px; height: 100px; }"
       "#sample:-webkit-drag { display:none }"
       "</style>"
       "<div id=sample></div>");
   Element* sample = GetDocument().getElementById("sample");
-  const std::unique_ptr<DragImage> image = GetFrame().NodeImage(*sample);
+  const std::unique_ptr<DragImage> image =
+      GetFrame().NodeImage(GetFrame(), *sample);
   EXPECT_EQ(nullptr, image.get()) << ":-webkit-drag blows away layout object";
 }
 
 TEST_F(LocalFrameTest, nodeImageWithChangingLayoutObject) {
   SetBodyContent(
       "<style>"
       "#sample { color: blue; }"
       "#sample:-webkit-drag { display: inline-block; color: red; }"
       "</style>"
       "<span id=sample>foo</span>");
   Element* sample = GetDocument().getElementById("sample");
   UpdateAllLifecyclePhases();
   LayoutObject* before_layout_object = sample->GetLayoutObject();
-  const std::unique_ptr<DragImage> image = GetFrame().NodeImage(*sample);
+  const std::unique_ptr<DragImage> image =
+      GetFrame().NodeImage(GetFrame(), *sample);
 
   EXPECT_TRUE(sample->GetLayoutObject() != before_layout_object)
       << ":-webkit-drag causes sample to have different layout object.";
   EXPECT_EQ(Color(255, 0, 0),
             sample->GetLayoutObject()->ResolveColor(CSSPropertyColor))
       << "#sample has :-webkit-drag.";
 
   // Layout w/o :-webkit-drag
   UpdateAllLifecyclePhases();
 
   EXPECT_EQ(Color(0, 0, 255),
             sample->GetLayoutObject()->ResolveColor(CSSPropertyColor))
       << "#sample doesn't have :-webkit-drag.";
 }
 
 TEST_F(LocalFrameTest, dragImageForSelectionUsesPageScaleFactor) {
   SetBodyContent(
       "<div>Hello world! This tests that the bitmap for drag image is scaled "
       "by page scale factor</div>");
   GetFrame().GetPage()->GetVisualViewport().SetScale(1);
   GetFrame().Selection().SelectAll();
   UpdateAllLifecyclePhases();
   const std::unique_ptr<DragImage> image1(
-      GetFrame().DragImageForSelection(0.75f));
+      GetFrame().DragImageForSelection(GetFrame(), 0.75f));
   GetFrame().GetPage()->GetVisualViewport().SetScale(2);
   GetFrame().Selection().SelectAll();
   UpdateAllLifecyclePhases();
   const std::unique_ptr<DragImage> image2(
-      GetFrame().DragImageForSelection(0.75f));
+      GetFrame().DragImageForSelection(GetFrame(), 0.75f));
 
   EXPECT_GT(image1->Size().Width(), 0);
   EXPECT_GT(image1->Size().Height(), 0);
   EXPECT_EQ(image1->Size().Width() * 2, image2->Size().Width());
   EXPECT_EQ(image1->Size().Height() * 2, image2->Size().Height());
 }
 
 }  // namespace blink