Limit stack-allocated uses of CancellableTaskFactory(PassOwnPtr<Closure>).

Source/platform/scheduler/CancellableTaskFactoryTest.cpp

 // Copyright 2015 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/scheduler/CancellableTaskFactory.h"
 
 #include <gtest/gtest.h>
 
 namespace blink {
 
+namespace {
+
+class TestCancellableTaskFactory : public CancellableTaskFactory {
+public:
+    explicit TestCancellableTaskFactory(PassOwnPtr<Closure> closure)
+        : CancellableTaskFactory(closure)
+    {
+    }
+};
+
+} // namespace
+
 using CancellableTaskFactoryTest = testing::Test;
 
 TEST_F(CancellableTaskFactoryTest, IsPending_TaskNotCreated)
 {
-    CancellableTaskFactory factory(nullptr);
+    TestCancellableTaskFactory factory(nullptr);
 
     EXPECT_FALSE(factory.isPending());
 }
 
 TEST_F(CancellableTaskFactoryTest, IsPending_TaskCreated)
 {
-    CancellableTaskFactory factory(nullptr);
+    TestCancellableTaskFactory factory(nullptr);
     OwnPtr<WebTaskRunner::Task> task = adoptPtr(factory.cancelAndCreate());
 
     EXPECT_TRUE(factory.isPending());
 }
 
 void EmptyFn()
 {
 }
 
 TEST_F(CancellableTaskFactoryTest, IsPending_TaskCreatedAndRun)
 {
-    CancellableTaskFactory factory(WTF::bind(&EmptyFn));
+    TestCancellableTaskFactory factory(WTF::bind(&EmptyFn));
     {
         OwnPtr<WebTaskRunner::Task> task = adoptPtr(factory.cancelAndCreate());
         task->run();
     }
 
     EXPECT_FALSE(factory.isPending());
 }
 
 TEST_F(CancellableTaskFactoryTest, IsPending_TaskCreatedAndDestroyed)
 {
-    CancellableTaskFactory factory(nullptr);
+    TestCancellableTaskFactory factory(nullptr);
     delete factory.cancelAndCreate();
 
     EXPECT_FALSE(factory.isPending());
 }
 
 TEST_F(CancellableTaskFactoryTest, IsPending_TaskCreatedAndCancelled)
 {
-    CancellableTaskFactory factory(nullptr);
+    TestCancellableTaskFactory factory(nullptr);
     OwnPtr<WebTaskRunner::Task> task = adoptPtr(factory.cancelAndCreate());
     factory.cancel();
 
     EXPECT_FALSE(factory.isPending());
 }
 
 class TestClass {
 public:
-    CancellableTaskFactory m_factory;
+    OwnPtr<CancellableTaskFactory> m_factory;
 
     TestClass()
-        : m_factory(WTF::bind(&TestClass::TestFn, this))
+        : m_factory(CancellableTaskFactory::create(this, &TestClass::TestFn))
     {
     }
 
     void TestFn()
     {
-        EXPECT_FALSE(m_factory.isPending());
+        EXPECT_FALSE(m_factory->isPending());
     }
 };
 
 TEST_F(CancellableTaskFactoryTest, IsPending_InCallback)
 {
     TestClass testClass;
-    OwnPtr<WebTaskRunner::Task> task = adoptPtr(testClass.m_factory.cancelAndCreate());
+    OwnPtr<WebTaskRunner::Task> task = adoptPtr(testClass.m_factory->cancelAndCreate());
     task->run();
 }
 
 void AddOne(int* ptr)
 {
     *ptr += 1;
 }
 
 TEST_F(CancellableTaskFactoryTest, Run_ClosureIsExecuted)
 {
     int executionCount = 0;
-    CancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
+    TestCancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
     OwnPtr<WebTaskRunner::Task> task = adoptPtr(factory.cancelAndCreate());
     task->run();
 
     EXPECT_EQ(1, executionCount);
 }
 
 TEST_F(CancellableTaskFactoryTest, Run_ClosureIsExecutedOnlyOnce)
 {
     int executionCount = 0;
-    CancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
+    TestCancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
     OwnPtr<WebTaskRunner::Task> task = adoptPtr(factory.cancelAndCreate());
     task->run();
     task->run();
     task->run();
     task->run();
 
     EXPECT_EQ(1, executionCount);
 }
 
 TEST_F(CancellableTaskFactoryTest, Run_FactoryDestructionPreventsExecution)
 {
     int executionCount = 0;
     OwnPtr<WebTaskRunner::Task> task;
     {
-        CancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
+        TestCancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
         task = adoptPtr(factory.cancelAndCreate());
     }
     task->run();
 
     EXPECT_EQ(0, executionCount);
 }
 
 TEST_F(CancellableTaskFactoryTest, Run_TasksInSequence)
 {
     int executionCount = 0;
-    CancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
+    TestCancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
 
     OwnPtr<WebTaskRunner::Task> taskA = adoptPtr(factory.cancelAndCreate());
     taskA->run();
     EXPECT_EQ(1, executionCount);
 
     OwnPtr<WebTaskRunner::Task> taskB = adoptPtr(factory.cancelAndCreate());
     taskB->run();
     EXPECT_EQ(2, executionCount);
 
     OwnPtr<WebTaskRunner::Task> taskC = adoptPtr(factory.cancelAndCreate());
     taskC->run();
     EXPECT_EQ(3, executionCount);
 }
 
 TEST_F(CancellableTaskFactoryTest, Cancel)
 {
     int executionCount = 0;
-    CancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
+    TestCancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
     OwnPtr<WebTaskRunner::Task> task = adoptPtr(factory.cancelAndCreate());
     factory.cancel();
     task->run();
 
     EXPECT_EQ(0, executionCount);
 }
 
 TEST_F(CancellableTaskFactoryTest, CreatingANewTaskCancelsPreviousOnes)
 {
     int executionCount = 0;
-    CancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
+    TestCancellableTaskFactory factory(WTF::bind(&AddOne, &executionCount));
 
     OwnPtr<WebTaskRunner::Task> taskA = adoptPtr(factory.cancelAndCreate());
     OwnPtr<WebTaskRunner::Task> taskB = adoptPtr(factory.cancelAndCreate());
 
     taskA->run();
     EXPECT_EQ(0, executionCount);
 
     taskB->run();
     EXPECT_EQ(1, executionCount);
 }
 
 } // namespace blink