Refactoring Canvas2DLayerBridge to make it easier to write unit tests

Source/core/platform/graphics/chromium/Canvas2DLayerBridgeTest.cpp

 /*
  * Copyright (C) 2011 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:
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  * 2.  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.
  *
  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS 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 APPLE INC. OR ITS 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 "core/platform/graphics/chromium/Canvas2DLayerBridge.h"
-
 #include "SkDeferredCanvas.h"
+#include "SkSurface.h"
 #include "core/platform/graphics/ImageBuffer.h"
-#include "core/tests/FakeWebGraphicsContext3D.h"
+#include "core/platform/graphics/chromium/Canvas2DLayerBridgeTestHelper.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebThread.h"
 #include "third_party/skia/include/core/SkDevice.h"
 #include "wtf/RefPtr.h"
 
-#include <gmock/gmock.h>
 #include <gtest/gtest.h>
 
 using namespace WebCore;
 using namespace WebKit;
 using testing::InSequence;
 using testing::Return;
 using testing::Test;
 
 namespace {
 
-class MockCanvasContext : public FakeWebGraphicsContext3D {
-public:
-    MOCK_METHOD0(flush, void(void));
-    MOCK_METHOD0(createTexture, unsigned(void));
-    MOCK_METHOD1(deleteTexture, void(unsigned));
-
-    virtual GrGLInterface* onCreateGrGLInterface() OVERRIDE { return 0; }
-};
-
 class FakeCanvas2DLayerBridge : public Canvas2DLayerBridge {
 public:
-    static PassOwnPtr<FakeCanvas2DLayerBridge> create(PassRefPtr<GraphicsContext3D> context, SkDeferredCanvas* canvas, OpacityMode opacityMode)
+    static PassOwnPtr<FakeCanvas2DLayerBridge> create(PassOwnPtr<Helper> helper, const IntSize& size, OpacityMode opacityMode, bool* success)
     {
-        return adoptPtr(new FakeCanvas2DLayerBridge(context, canvas, opacityMode));
+        return adoptPtr(new FakeCanvas2DLayerBridge(helper, size, opacityMode, success));
     }
 protected:
-    FakeCanvas2DLayerBridge(PassRefPtr<GraphicsContext3D> context, SkDeferredCanvas* canvas, OpacityMode opacityMode) :
-        Canvas2DLayerBridge(context, canvas, opacityMode)
+    FakeCanvas2DLayerBridge(PassOwnPtr<Helper> helper, const WebCore::IntSize& size, OpacityMode opacityMode, bool* success) :
+        Canvas2DLayerBridge(helper, size, opacityMode, success)
     { }
 };
 
-} // namespace
+}
 
 class Canvas2DLayerBridgeTest : public Test {
 protected:
     void fullLifecycleTest()
     {
-        RefPtr<GraphicsContext3D> mainContext = GraphicsContext3D::createGraphicsContextFromWebContext(adoptPtr(new MockCanvasContext));
+        OwnPtr<Canvas2DLayerBridgeTestHelper> testHelper = adoptPtr(new Canvas2DLayerBridgeTestHelper);
 
-        MockCanvasContext& mainMock = *static_cast<MockCanvasContext*>(mainContext->webContext());
+        MockCanvasContext* mockContext = &testHelper->mock();
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
 
-        SkDevice device(SkBitmap::kARGB_8888_Config, 300, 150);
-        SkDeferredCanvas canvas(&device);
+        IntSize size(300, 150);
+        bool success;
+        OwnPtr<Canvas2DLayerBridge> bridge = FakeCanvas2DLayerBridge::create(testHelper.release(), size, Canvas2DLayerBridge::NonOpaque, &success);
+        EXPECT_TRUE(success);
 
-        ::testing::Mock::VerifyAndClearExpectations(&mainMock);
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
 
-        OwnPtr<Canvas2DLayerBridge> bridge = FakeCanvas2DLayerBridge::create(mainContext.release(), &canvas, Canvas2DLayerBridge::NonOpaque);
-
-        ::testing::Mock::VerifyAndClearExpectations(&mainMock);
-
-        EXPECT_CALL(mainMock, flush());
+        EXPECT_CALL(*mockContext, flush());
         unsigned textureId = bridge->backBufferTexture();
         EXPECT_EQ(textureId, 0u);
 
-        ::testing::Mock::VerifyAndClearExpectations(&mainMock);
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
 
         bridge.clear();
 
-        ::testing::Mock::VerifyAndClearExpectations(&mainMock);
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
+    }
+
+    void loseContextTest()
+    {
+        OwnPtr<Canvas2DLayerBridgeTestHelper> testHelper = adoptPtr(new Canvas2DLayerBridgeTestHelper);
+        MockCanvasContext* mockContext = &testHelper->mock();
+
+        IntSize size(300, 150);
+        bool success;
+        OwnPtr<Canvas2DLayerBridge> bridge = FakeCanvas2DLayerBridge::create(testHelper.release(), size, Canvas2DLayerBridge::NonOpaque, &success);
+        EXPECT_TRUE(success);
+
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
+        EXPECT_TRUE(bridge->isValid());
+
+        mockContext->fakeContextLoss();
+
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
+
+        EXPECT_FALSE(bridge->isValid());
+
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
+
+        static_cast<Canvas2DLayerBridgeTestHelper*>(bridge->helper())->regenerateContext();
+        mockContext = &static_cast<Canvas2DLayerBridgeTestHelper*>(bridge->helper())->mock();
+
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
+
+        EXPECT_TRUE(bridge->isValid());
+
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
+
+        EXPECT_FALSE(bridge->prepareMailbox(0, 0));
+
+        ::testing::Mock::VerifyAndClearExpectations(mockContext);
     }
 };
 
 namespace {
 
-TEST_F(Canvas2DLayerBridgeTest, testFullLifecycleSingleThreaded)
+TEST_F(Canvas2DLayerBridgeTest, testFullLifecycle)
 {
     fullLifecycleTest();
 }
 
+TEST_F(Canvas2DLayerBridgeTest, testLoseContext)
+{
+    loseContextTest();
+}
+
 } // namespace