Introduce WebTaskRunner Patch 3/5

Source/platform/graphics/RecordingImageBufferSurfaceTest.cpp

 // Copyright 2014 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 "config.h"
 #include "platform/graphics/RecordingImageBufferSurface.h"
 
 #include "platform/graphics/GraphicsContext.h"
 #include "platform/graphics/ImageBuffer.h"
 #include "platform/graphics/ImageBufferClient.h"
 #include "platform/graphics/UnacceleratedImageBufferSurface.h"
 #include "public/platform/Platform.h"
+#include "public/platform/WebTaskRunner.h"
 #include "public/platform/WebThread.h"
 #include "public/platform/WebTraceLocation.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #include "third_party/skia/include/core/SkPictureRecorder.h"
 #include "wtf/OwnPtr.h"
 #include "wtf/PassOwnPtr.h"
 #include "wtf/RefPtr.h"
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>
 
@@ skipping 224 matching lines @@
     {
         Platform::initialize(m_oldPlatform);
     }
 
     void enterRunLoop()
     {
         m_mockPlatform.enterRunLoop();
     }
 
 private:
-    class CurrentThreadMock : public WebThread {
+    class MockWebTaskRunner : public WebTaskRunner {
     public:
-        CurrentThreadMock() : m_taskObserver(0), m_task(0) { }
-
-        ~CurrentThreadMock() override
-        {
-            EXPECT_EQ((Task*)0, m_task);
-        }
+        MockWebTaskRunner() : m_task(0) { }
+        ~MockWebTaskRunner() override { }
 
         virtual void postTask(const WebTraceLocation&, Task* task)
         {
             EXPECT_EQ((Task*)0, m_task);
             m_task = task;
         }
 
         void postDelayedTask(const WebTraceLocation&, Task*, long long delayMs) override { ASSERT_NOT_REACHED(); };
 
+        Task* m_task;
+    };
+
+    class CurrentThreadMock : public WebThread {
+    public:
+        CurrentThreadMock() : m_taskObserver(0) { }
+
+        ~CurrentThreadMock() override
+        {
+            EXPECT_EQ((WebTaskRunner::Task*)0, m_taskRunner.m_task);
+        }
+
+        WebTaskRunner* taskRunner() override
+        {
+            return &m_taskRunner;
+        }
+
         bool isCurrentThread() const override { return true; }
+
         PlatformThreadId threadId() const override
         {
             ASSERT_NOT_REACHED();
             return 0;
         }
 
         void addTaskObserver(TaskObserver* taskObserver) override
         {
-            EXPECT_EQ((TaskObserver*)0, m_taskObserver);
+            EXPECT_EQ(nullptr, m_taskObserver);
             m_taskObserver = taskObserver;
         }
 
         void removeTaskObserver(TaskObserver* taskObserver) override
         {
             EXPECT_EQ(m_taskObserver, taskObserver);
             m_taskObserver = 0;
         }
 
         WebScheduler* scheduler() const override
         {
             ASSERT_NOT_REACHED();
             return nullptr;
         }
 
         void enterRunLoop()
         {
             if (m_taskObserver)
                 m_taskObserver->willProcessTask();
-            if (m_task) {
-                m_task->run();
-                delete m_task;
-                m_task = 0;
+            if (m_taskRunner.m_task) {
+                m_taskRunner.m_task->run();
+                delete m_taskRunner.m_task;
+                m_taskRunner.m_task = 0;
             }
             if (m_taskObserver)
                 m_taskObserver->didProcessTask();
         }
 
     private:
+        MockWebTaskRunner m_taskRunner;
         TaskObserver* m_taskObserver;
-        Task* m_task;
     };
 
     class CurrentThreadPlatformMock : public Platform {
     public:
         CurrentThreadPlatformMock() { }
         virtual void cryptographicallyRandomValues(unsigned char* buffer, size_t length) { ASSERT_NOT_REACHED(); }
         WebThread* currentThread() override { return &m_currentThread; }
 
         void enterRunLoop() { m_currentThread.enterRunLoop(); }
     private:
         CurrentThreadMock m_currentThread;
     };
 
     CurrentThreadPlatformMock m_mockPlatform;
     Platform* m_oldPlatform;
 };
 
 } // anonymous namespace
 
 #define DEFINE_TEST_TASK_WRAPPER_CLASS(TEST_METHOD)                                               \
-class TestWrapperTask_ ## TEST_METHOD : public WebThread::Task {                           \
+class TestWrapperTask_ ## TEST_METHOD : public WebTaskRunner::Task {                           \
     public:                                                                                       \
         TestWrapperTask_ ## TEST_METHOD(RecordingImageBufferSurfaceTest* test) : m_test(test) { } \
         void run() override { m_test->TEST_METHOD(); }                                    \
     private:                                                                                      \
         RecordingImageBufferSurfaceTest* m_test;                                                  \
 };
 
 #define CALL_TEST_TASK_WRAPPER(TEST_METHOD)                                                               \
     {                                                                                                     \
         AutoInstallCurrentThreadPlatformMock ctpm;                                                        \
-        Platform::current()->currentThread()->postTask(FROM_HERE, new TestWrapperTask_ ## TEST_METHOD(this)); \
+        Platform::current()->currentThread()->taskRunner()->postTask(FROM_HERE, new TestWrapperTask_ ## TEST_METHOD(this)); \
         ctpm.enterRunLoop();                                      \
     }
 
 TEST_F(RecordingImageBufferSurfaceTest, testEmptyPicture)
 {
     testEmptyPicture();
 }
 
 TEST_F(RecordingImageBufferSurfaceTest, testNoFallbackWithClear)
 {
@@ skipping 32 matching lines @@
 }
 
 DEFINE_TEST_TASK_WRAPPER_CLASS(testClearRect)
 TEST_F(RecordingImageBufferSurfaceTest, testClearRect)
 {
     CALL_TEST_TASK_WRAPPER(testClearRect);
     expectDisplayListEnabled(true);
 }
 
 } // namespace blink