Refactor WebCore::DragImage.

Source/WebKit/chromium/tests/DragImageTest.cpp

 /*
  * Copyright (C) 2010 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 14 matching lines @@
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 
 #include "core/platform/DragImage.h"
 
+#include "core/platform/graphics/Image.h"
+#include "core/platform/graphics/IntSize.h"
+#include "core/platform/graphics/skia/NativeImageSkia.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "wtf/OwnPtr.h"
+#include "wtf/PassOwnPtr.h"
+#include "wtf/PassRefPtr.h"
+#include "wtf/RefPtr.h"
+
 #include <gtest/gtest.h>
-#include "core/platform/graphics/Image.h"
-#include "core/platform/graphics/skia/NativeImageSkia.h"
-#include <wtf/PassOwnPtr.h>
 
 using namespace WebCore;
 
 namespace {
 
 class TestImage : public Image {
 public:
 
     static PassRefPtr<TestImage> create(const IntSize& size)
     {
@@ skipping 44 matching lines @@
 
 private:
 
     IntSize m_size;
 
     RefPtr<NativeImageSkia> m_nativeImage;
 };
 
 TEST(DragImageTest, NullHandling)
 {
-    EXPECT_FALSE(createDragImageFromImage(0));
+    EXPECT_FALSE(DragImage::create(0));
 
-    deleteDragImage(0);
-    EXPECT_TRUE(dragImageSize(0).isZero());
-    EXPECT_FALSE(scaleDragImage(0, FloatSize(0.5, 0.5)));
-    EXPECT_FALSE(dissolveDragImageToFraction(0, 0.5));
-    EXPECT_FALSE(createDragImageFromImage(0));
-    EXPECT_FALSE(createDragImageIconForCachedImage(0));
+    RefPtr<TestImage> nullTestImage(TestImage::create(IntSize()));
+    EXPECT_FALSE(DragImage::create(nullTestImage.get()));
 }
 
 TEST(DragImageTest, NonNullHandling)
 {
     RefPtr<TestImage> testImage(TestImage::create(IntSize(2, 2)));
-    DragImageRef dragImage = createDragImageFromImage(testImage.get());
+    OwnPtr<DragImage> dragImage = DragImage::create(testImage.get());
     ASSERT_TRUE(dragImage);
 
-    dragImage = scaleDragImage(dragImage, FloatSize(0.5, 0.5));
-    ASSERT_TRUE(dragImage);
-    IntSize size = dragImageSize(dragImage);
+    dragImage->scale(0.5, 0.5);
+    IntSize size = dragImage->size();
     EXPECT_EQ(1, size.width());
     EXPECT_EQ(1, size.height());
 
-    dragImage = dissolveDragImageToFraction(dragImage, 0.5);
-    ASSERT_TRUE(dragImage);
-
-    deleteDragImage(dragImage);
+    dragImage->dissolveToFraction(0.5);
 }
 
 TEST(DragImageTest, CreateDragImage)
 {
     {
         // Tests that the DrageImage implementation doesn't choke on null values
         // of nativeImageForCurrentFrame().
         RefPtr<TestImage> testImage(TestImage::create(IntSize()));
-        EXPECT_FALSE(createDragImageFromImage(testImage.get()));
+        EXPECT_FALSE(DragImage::create(testImage.get()));
     }
 
     {
         // Tests that the drag image is a deep copy.
         RefPtr<TestImage> testImage(TestImage::create(IntSize(1, 1)));
-        DragImageRef dragImage = createDragImageFromImage(testImage.get());
+        OwnPtr<DragImage> dragImage = DragImage::create(testImage.get());
         ASSERT_TRUE(dragImage);
-        SkAutoLockPixels lock1(*dragImage->bitmap), lock2(testImage->nativeImageForCurrentFrame()->bitmap());
-        EXPECT_NE(dragImage->bitmap->getPixels(), testImage->nativeImageForCurrentFrame()->bitmap().getPixels());
+        SkAutoLockPixels lock1(dragImage->bitmap()), lock2(testImage->nativeImageForCurrentFrame()->bitmap());
+        EXPECT_NE(dragImage->bitmap().getPixels(), testImage->nativeImageForCurrentFrame()->bitmap().getPixels());
     }
 }
 
 } // anonymous namespace