In-place modification of drag related functionality present in LocalFrame

third_party/WebKit/Source/core/clipboard/DataTransferTest.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/clipboard/DataTransfer.h"
+
 #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"
-#include "core/html/HTMLElement.h"
 #include "core/layout/LayoutObject.h"
 #include "core/testing/DummyPageHolder.h"
 #include "core/timing/Performance.h"
 #include "platform/DragImage.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
-class LocalFrameTest : public ::testing::Test {
+class DataTransferTest : public ::testing::Test {
  protected:
-  LocalFrameTest() = default;
-  ~LocalFrameTest() override = default;
+  DataTransferTest() = default;
+  ~DataTransferTest() override = default;
 
   Document& GetDocument() const { return dummy_page_holder_->GetDocument(); }
   LocalFrame& GetFrame() const { return *GetDocument().GetFrame(); }
   Performance* GetPerformance() const { return performance_; }
 
   void SetBodyContent(const std::string& body_content) {
     GetDocument().body()->setInnerHTML(String::FromUTF8(body_content.c_str()));
     UpdateAllLifecyclePhases();
   }
 
   void UpdateAllLifecyclePhases() {
     GetDocument().View()->UpdateAllLifecyclePhases();
   }
 
  private:
   void SetUp() override {
     dummy_page_holder_ = DummyPageHolder::Create(IntSize(800, 600));
     performance_ = Performance::Create(&GetFrame());
   }
 
   std::unique_ptr<DummyPageHolder> dummy_page_holder_;
   Persistent<Performance> performance_;
 };
 
-TEST_F(LocalFrameTest, nodeImage) {
+TEST_F(DataTransferTest, 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 =
+      DataTransfer::NodeImage(GetFrame(), *sample);
   EXPECT_EQ(IntSize(100, 100), image->Size());
 }
 
-TEST_F(LocalFrameTest, nodeImageWithNestedElement) {
+TEST_F(DataTransferTest, 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 =
+      DataTransfer::NodeImage(GetFrame(), *sample);
   EXPECT_EQ(
       Color(0, 255, 0),
       sample->firstChild()->GetLayoutObject()->ResolveColor(CSSPropertyColor))
       << "Descendants node should have :-webkit-drag.";
 }
 
-TEST_F(LocalFrameTest, nodeImageWithPsuedoClassWebKitDrag) {
+TEST_F(DataTransferTest, 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 =
+      DataTransfer::NodeImage(GetFrame(), *sample);
   EXPECT_EQ(IntSize(200, 200), image->Size())
       << ":-webkit-drag should affect dragged image.";
 }
 
-TEST_F(LocalFrameTest, nodeImageWithoutDraggedLayoutObject) {
+TEST_F(DataTransferTest, 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 =
+      DataTransfer::NodeImage(GetFrame(), *sample);
   EXPECT_EQ(nullptr, image.get()) << ":-webkit-drag blows away layout object";
 }
 
-TEST_F(LocalFrameTest, nodeImageWithChangingLayoutObject) {
+TEST_F(DataTransferTest, 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 =
+      DataTransfer::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().GetPage()->GetVisualViewport().SetScale(2);
-  GetFrame().Selection().SelectAll();
-  UpdateAllLifecyclePhases();
-  const std::unique_ptr<DragImage> image2(
-      GetFrame().DragImageForSelection(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