diff -r 000000000000 -r 4f2f89ce4247 WebKit/chromium/tests/DragImageTest.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/WebKit/chromium/tests/DragImageTest.cpp	Fri Sep 17 09:02:29 2010 +0300
@@ -0,0 +1,151 @@
+/*
+ * 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
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ *     * Neither the name of Google Inc. nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * 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 <gtest/gtest.h>
+
+#include "DragImage.h"
+#include "Image.h"
+#include "NativeImageSkia.h"
+
+using namespace WebCore;
+
+namespace {
+
+class TestImage : public Image {
+public:
+
+    static PassRefPtr<TestImage> create(const IntSize& size)
+    {
+        return adoptRef(new TestImage(size));
+    }
+
+    explicit TestImage(const IntSize& size)
+        : Image(0)
+        , m_size(size)
+    {
+        m_nativeImage = new NativeImageSkia();
+        m_nativeImage->setConfig(SkBitmap::kARGB_8888_Config,
+                                 size.width(), size.height(), 0);
+        m_nativeImage->allocPixels();
+    }
+
+    virtual ~TestImage()
+    {
+        delete m_nativeImage;
+    }
+
+    virtual IntSize size() const
+    {
+        return m_size;
+    }
+
+    virtual NativeImagePtr nativeImageForCurrentFrame()
+    {
+        if (m_size.isZero())
+            return 0;
+
+        return m_nativeImage;
+    }
+
+    // Stub implementations of pure virtual Image functions.
+    virtual void destroyDecodedData(bool)
+    {
+    }
+
+    virtual unsigned int decodedSize() const
+    {
+        return 0u;
+    }
+
+    virtual void draw(WebCore::GraphicsContext*, const WebCore::FloatRect&,
+                      const WebCore::FloatRect&, WebCore::ColorSpace,
+                      WebCore::CompositeOperator)
+    {
+    }
+
+private:
+
+    IntSize m_size;
+
+    NativeImagePtr m_nativeImage;
+};
+
+TEST(DragImageTest, NullHandling)
+{
+    EXPECT_FALSE(createDragImageFromImage(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));
+}
+
+TEST(DragImageTest, NonNullHandling)
+{
+    RefPtr<TestImage> testImage(TestImage::create(IntSize(2, 2)));
+    DragImageRef dragImage = createDragImageFromImage(testImage.get());
+    ASSERT_TRUE(dragImage);
+
+    dragImage = scaleDragImage(dragImage, FloatSize(0.5, 0.5));
+    ASSERT_TRUE(dragImage);
+    IntSize size = dragImageSize(dragImage);
+    EXPECT_EQ(1, size.width());
+    EXPECT_EQ(1, size.height());
+
+    dragImage = dissolveDragImageToFraction(dragImage, 0.5);
+    ASSERT_TRUE(dragImage);
+
+    deleteDragImage(dragImage);
+}
+
+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()));
+    }
+
+    {
+        // Tests that the drag image is a deep copy.
+        RefPtr<TestImage> testImage(TestImage::create(IntSize(1, 1)));
+        DragImageRef dragImage = createDragImageFromImage(testImage.get());
+        ASSERT_TRUE(dragImage);
+        SkAutoLockPixels lock1(*dragImage), lock2(*(testImage->nativeImageForCurrentFrame()));
+        EXPECT_NE(dragImage->getPixels(), testImage->nativeImageForCurrentFrame()->getPixels());
+    }
+}
+
+} // anonymous namespace