Optimize posting of WTF::Closure and improve scheduler test mocks

third_party/WebKit/Source/core/svg/graphics/SVGImageTest.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/svg/graphics/SVGImage.h"
 
 #include "core/svg/graphics/SVGImageChromeClient.h"
 #include "platform/SharedBuffer.h"
 #include "platform/Timer.h"
 #include "platform/geometry/FloatRect.h"
+#include "platform/testing/UnitTestHelpers.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #include "third_party/skia/include/utils/SkNullCanvas.h"
 #include "wtf/PtrUtil.h"
 
 namespace blink {
-namespace {
-
-class MockTaskRunner : public WebTaskRunner {
-public:
-    void setTime(double newTime) { m_time = newTime; }
-
-    MockTaskRunner()
-    : WebTaskRunner(), m_time(0.0), m_currentTask(nullptr)
-    { }
-
-    virtual ~MockTaskRunner()
-    {
-        if (m_currentTask)
-            delete m_currentTask;
-    }
-
-private:
-    void postTask(const WebTraceLocation&, Task*) override { }
-    void postDelayedTask(const WebTraceLocation&, Task* task, double) override
-    {
-        if (m_currentTask)
-            delete m_currentTask;
-        m_currentTask = task;
-
-    }
-    bool runsTasksOnCurrentThread() override { return true; }
-    std::unique_ptr<WebTaskRunner> clone() override { return nullptr; }
-    double virtualTimeSeconds() const override { return 0.0; }
-    double monotonicallyIncreasingVirtualTimeSeconds() const override { return m_time; }
-    SingleThreadTaskRunner* taskRunner() override { return nullptr; }
-
-    double m_time;
-    Task* m_currentTask;
-};
-
-class MockTimer : public TaskRunnerTimer<SVGImageChromeClient> {
-public:
-    using TimerFiredFunction = typename TaskRunnerTimer<SVGImageChromeClient>::TimerFiredFunction;
-
-    MockTimer(SVGImageChromeClient* o, TimerFiredFunction f)
-        : TaskRunnerTimer(&m_taskRunner, o, f)
-    {
-    }
-
-    void fire()
-    {
-        fired();
-        stop();
-    }
-
-    void setTime(double newTime)
-    {
-        m_taskRunner.setTime(newTime);
-    }
-
-private:
-    MockTaskRunner m_taskRunner;
-};
-
-} // namespace
-
-class SVGImageTest : public testing::Test {
+class SVGImageTest : public ::testing::Test {
 public:
     SVGImage& image() { return *m_image; }
 
     void load(const char* data, bool shouldPause)
     {
         m_observer = new PauseControlImageObserver(shouldPause);
         m_image = SVGImage::create(m_observer);
         m_image->setData(SharedBuffer::create(data, strlen(data)), true);
     }
 
@@ skipping 47 matching lines @@
     "}"
     "</style>"
     "<path class='spinner' fill='none' d='M 8,1.125 A 6.875,6.875 0 1 1 1.125,8' stroke-width='2' stroke='blue'/>"
     "</svg>";
 
 TEST_F(SVGImageTest, TimelineSuspendAndResume)
 {
     const bool shouldPause = true;
     load(kAnimatedDocument, shouldPause);
     SVGImageChromeClient& chromeClient = image().chromeClientForTesting();
-    MockTimer* timer = new MockTimer(&chromeClient, &SVGImageChromeClient::animationTimerFired);
+    Timer<SVGImageChromeClient>* timer = new Timer<SVGImageChromeClient>(&chromeClient, &SVGImageChromeClient::animationTimerFired);
     chromeClient.setTimer(wrapUnique(timer));
 
     // Simulate a draw. Cause a frame (timer) to be scheduled.
     pumpFrame();
     EXPECT_TRUE(image().hasAnimations());
     EXPECT_TRUE(timer->isActive());
 
     // Fire the timer/trigger a frame update. Since the observer always returns
     // true for shouldPauseAnimation, this will result in the timeline being
     // suspended.
-    timer->fire();
+    // TODO(alexclarke): Move over to using base::TimeDelta and base::TimeTicks so we can avoid computations like this.
+    testing::runDelayedTasks(1.0 + timer->nextFireInterval() * 1000.0);
     EXPECT_TRUE(chromeClient.isSuspended());
     EXPECT_FALSE(timer->isActive());
 
     // Simulate a draw. This should resume the animation again.
     pumpFrame();
     EXPECT_TRUE(timer->isActive());
     EXPECT_FALSE(chromeClient.isSuspended());
 }
 
 TEST_F(SVGImageTest, ResetAnimation)
 {
     const bool shouldPause = false;
     load(kAnimatedDocument, shouldPause);
     SVGImageChromeClient& chromeClient = image().chromeClientForTesting();
-    MockTimer* timer = new MockTimer(&chromeClient, &SVGImageChromeClient::animationTimerFired);
+    Timer<SVGImageChromeClient>* timer = new Timer<SVGImageChromeClient>(&chromeClient, &SVGImageChromeClient::animationTimerFired);
     chromeClient.setTimer(wrapUnique(timer));
 
     // Simulate a draw. Cause a frame (timer) to be scheduled.
     pumpFrame();
     EXPECT_TRUE(image().hasAnimations());
     EXPECT_TRUE(timer->isActive());
 
     // Reset the animation. This will suspend the timeline but not cancel the
     // timer.
     image().resetAnimation();
     EXPECT_TRUE(chromeClient.isSuspended());
     EXPECT_TRUE(timer->isActive());
 
     // Fire the timer/trigger a frame update. The timeline will remain
     // suspended and no frame will be scheduled.
-    timer->fire();
+    // TODO(alexclarke): Move over to using base::TimeDelta and base::TimeTicks so we can avoid computations like this.
+    testing::runDelayedTasks(1.0 + timer->nextFireInterval() * 1000.0);
     EXPECT_TRUE(chromeClient.isSuspended());
     EXPECT_FALSE(timer->isActive());
 
     // Simulate a draw. This should resume the animation again.
     pumpFrame();
     EXPECT_FALSE(chromeClient.isSuspended());
     EXPECT_TRUE(timer->isActive());
 }
 
 } // namespace blink