| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "platform/WebTaskRunner.h" | 5 #include "platform/WebTaskRunner.h" |
| 6 | 6 |
| 7 #include "platform/scheduler/test/fake_web_task_runner.h" | 7 #include "platform/scheduler/test/fake_web_task_runner.h" |
| 8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 9 | 9 |
| 10 namespace blink { | 10 namespace blink { |
| (...skipping 91 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 102 #else | 102 #else |
| 103 handle = std::move(handle); | 103 handle = std::move(handle); |
| 104 #endif // COMPILER(CLANG) | 104 #endif // COMPILER(CLANG) |
| 105 EXPECT_EQ(0, count); | 105 EXPECT_EQ(0, count); |
| 106 taskRunner->runUntilIdle(); | 106 taskRunner->runUntilIdle(); |
| 107 EXPECT_EQ(1, count); | 107 EXPECT_EQ(1, count); |
| 108 | 108 |
| 109 // handle->isActive() should switch to false before the task starts running. | 109 // handle->isActive() should switch to false before the task starts running. |
| 110 bool isActive = false; | 110 bool isActive = false; |
| 111 handle = taskRunner->postCancellableTask( | 111 handle = taskRunner->postCancellableTask( |
| 112 BLINK_FROM_HERE, WTF::bind(&getIsActive, WTF::unretained(&isActive), | 112 BLINK_FROM_HERE, |
| 113 WTF::unretained(&handle))); | 113 WTF::bind(&getIsActive, WTF::unretained(&isActive), |
| 114 WTF::unretained(&handle))); |
| 114 EXPECT_TRUE(handle.isActive()); | 115 EXPECT_TRUE(handle.isActive()); |
| 115 taskRunner->runUntilIdle(); | 116 taskRunner->runUntilIdle(); |
| 116 EXPECT_FALSE(isActive); | 117 EXPECT_FALSE(isActive); |
| 117 EXPECT_FALSE(handle.isActive()); | 118 EXPECT_FALSE(handle.isActive()); |
| 118 } | 119 } |
| 119 | 120 |
| 120 TEST(WebTaskRunnerTest, CancellationCheckerTest) { | 121 TEST(WebTaskRunnerTest, CancellationCheckerTest) { |
| 121 RefPtr<scheduler::FakeWebTaskRunner> taskRunner = | 122 RefPtr<scheduler::FakeWebTaskRunner> taskRunner = |
| 122 adoptRef(new scheduler::FakeWebTaskRunner); | 123 adoptRef(new scheduler::FakeWebTaskRunner); |
| 123 | 124 |
| 124 int count = 0; | 125 int count = 0; |
| 125 TaskHandle handle = taskRunner->postCancellableTask( | 126 TaskHandle handle = taskRunner->postCancellableTask( |
| 126 BLINK_FROM_HERE, WTF::bind(&increment, WTF::unretained(&count))); | 127 BLINK_FROM_HERE, WTF::bind(&increment, WTF::unretained(&count))); |
| 127 EXPECT_EQ(0, count); | 128 EXPECT_EQ(0, count); |
| 128 | 129 |
| 129 // TaskHandle::isActive should detect the deletion of posted task. | 130 // TaskHandle::isActive should detect the deletion of posted task. |
| 130 auto queue = taskRunner->takePendingTasksForTesting(); | 131 auto queue = taskRunner->takePendingTasksForTesting(); |
| 131 ASSERT_EQ(1u, queue.size()); | 132 ASSERT_EQ(1u, queue.size()); |
| 132 EXPECT_FALSE(queue[0].IsCancelled()); | 133 EXPECT_FALSE(queue[0].IsCancelled()); |
| 133 EXPECT_TRUE(handle.isActive()); | 134 EXPECT_TRUE(handle.isActive()); |
| 134 queue.clear(); | 135 queue.clear(); |
| 135 EXPECT_FALSE(handle.isActive()); | 136 EXPECT_FALSE(handle.isActive()); |
| 136 EXPECT_EQ(0, count); | 137 EXPECT_EQ(0, count); |
| 137 | 138 |
| 138 count = 0; | 139 count = 0; |
| 139 CancellationTestHelper helper; | 140 CancellationTestHelper helper; |
| 140 handle = taskRunner->postCancellableTask( | 141 handle = taskRunner->postCancellableTask( |
| 141 BLINK_FROM_HERE, WTF::bind(&CancellationTestHelper::incrementCounter, | 142 BLINK_FROM_HERE, |
| 142 helper.createWeakPtr())); | 143 WTF::bind(&CancellationTestHelper::incrementCounter, |
| 144 helper.createWeakPtr())); |
| 143 EXPECT_EQ(0, helper.counter()); | 145 EXPECT_EQ(0, helper.counter()); |
| 144 | 146 |
| 145 // The cancellation of the posted task should be propagated to TaskHandle. | 147 // The cancellation of the posted task should be propagated to TaskHandle. |
| 146 queue = taskRunner->takePendingTasksForTesting(); | 148 queue = taskRunner->takePendingTasksForTesting(); |
| 147 ASSERT_EQ(1u, queue.size()); | 149 ASSERT_EQ(1u, queue.size()); |
| 148 EXPECT_FALSE(queue[0].IsCancelled()); | 150 EXPECT_FALSE(queue[0].IsCancelled()); |
| 149 EXPECT_TRUE(handle.isActive()); | 151 EXPECT_TRUE(handle.isActive()); |
| 150 helper.revokeWeakPtrs(); | 152 helper.revokeWeakPtrs(); |
| 151 EXPECT_TRUE(queue[0].IsCancelled()); | 153 EXPECT_TRUE(queue[0].IsCancelled()); |
| 152 EXPECT_FALSE(handle.isActive()); | 154 EXPECT_FALSE(handle.isActive()); |
| 153 } | 155 } |
| 154 | 156 |
| 155 } // namespace blink | 157 } // namespace blink |
| OLD | NEW |