| 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 "base/task_scheduler/task_tracker.h" | 5 #include "base/task_scheduler/task_tracker.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| (...skipping 18 matching lines...) Expand all Loading... |
| 29 namespace internal { | 29 namespace internal { |
| 30 | 30 |
| 31 namespace { | 31 namespace { |
| 32 | 32 |
| 33 // Calls TaskTracker::Shutdown() asynchronously. | 33 // Calls TaskTracker::Shutdown() asynchronously. |
| 34 class ThreadCallingShutdown : public SimpleThread { | 34 class ThreadCallingShutdown : public SimpleThread { |
| 35 public: | 35 public: |
| 36 explicit ThreadCallingShutdown(TaskTracker* tracker) | 36 explicit ThreadCallingShutdown(TaskTracker* tracker) |
| 37 : SimpleThread("ThreadCallingShutdown"), | 37 : SimpleThread("ThreadCallingShutdown"), |
| 38 tracker_(tracker), | 38 tracker_(tracker), |
| 39 has_returned_(true, false) {} | 39 has_returned_(WaitableEvent::ResetPolicy::MANUAL, |
| 40 WaitableEvent::InitialState::NOT_SIGNALED) {} |
| 40 | 41 |
| 41 // Returns true once the async call to Shutdown() has returned. | 42 // Returns true once the async call to Shutdown() has returned. |
| 42 bool has_returned() { return has_returned_.IsSignaled(); } | 43 bool has_returned() { return has_returned_.IsSignaled(); } |
| 43 | 44 |
| 44 private: | 45 private: |
| 45 void Run() override { | 46 void Run() override { |
| 46 tracker_->Shutdown(); | 47 tracker_->Shutdown(); |
| 47 has_returned_.Signal(); | 48 has_returned_.Signal(); |
| 48 } | 49 } |
| 49 | 50 |
| (...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 156 EXPECT_EQ(0U, num_tasks_executed_); | 157 EXPECT_EQ(0U, num_tasks_executed_); |
| 157 tracker_.RunTask(task.get()); | 158 tracker_.RunTask(task.get()); |
| 158 EXPECT_EQ(1U, num_tasks_executed_); | 159 EXPECT_EQ(1U, num_tasks_executed_); |
| 159 | 160 |
| 160 // Shutdown() shouldn't block. | 161 // Shutdown() shouldn't block. |
| 161 tracker_.Shutdown(); | 162 tracker_.Shutdown(); |
| 162 } | 163 } |
| 163 | 164 |
| 164 TEST_P(TaskSchedulerTaskTrackerTest, WillPostAndRunLongTaskBeforeShutdown) { | 165 TEST_P(TaskSchedulerTaskTrackerTest, WillPostAndRunLongTaskBeforeShutdown) { |
| 165 // Create a task that will block until |event| is signaled. | 166 // Create a task that will block until |event| is signaled. |
| 166 WaitableEvent event(false, false); | 167 WaitableEvent event(WaitableEvent::ResetPolicy::AUTOMATIC, |
| 168 WaitableEvent::InitialState::NOT_SIGNALED); |
| 167 std::unique_ptr<Task> blocked_task( | 169 std::unique_ptr<Task> blocked_task( |
| 168 new Task(FROM_HERE, Bind(&WaitableEvent::Wait, Unretained(&event)), | 170 new Task(FROM_HERE, Bind(&WaitableEvent::Wait, Unretained(&event)), |
| 169 TaskTraits().WithShutdownBehavior(GetParam()), TimeDelta())); | 171 TaskTraits().WithShutdownBehavior(GetParam()), TimeDelta())); |
| 170 | 172 |
| 171 // Inform |task_tracker_| that |blocked_task| will be posted. | 173 // Inform |task_tracker_| that |blocked_task| will be posted. |
| 172 EXPECT_TRUE(tracker_.WillPostTask(blocked_task.get())); | 174 EXPECT_TRUE(tracker_.WillPostTask(blocked_task.get())); |
| 173 | 175 |
| 174 // Run the task asynchronouly. | 176 // Run the task asynchronouly. |
| 175 ThreadRunningTask thread_running_task(&tracker_, blocked_task.get()); | 177 ThreadRunningTask thread_running_task(&tracker_, blocked_task.get()); |
| 176 thread_running_task.Start(); | 178 thread_running_task.Start(); |
| (...skipping 238 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 415 SkipOnShutdown, | 417 SkipOnShutdown, |
| 416 TaskSchedulerTaskTrackerTest, | 418 TaskSchedulerTaskTrackerTest, |
| 417 ::testing::Values(TaskShutdownBehavior::SKIP_ON_SHUTDOWN)); | 419 ::testing::Values(TaskShutdownBehavior::SKIP_ON_SHUTDOWN)); |
| 418 INSTANTIATE_TEST_CASE_P( | 420 INSTANTIATE_TEST_CASE_P( |
| 419 BlockShutdown, | 421 BlockShutdown, |
| 420 TaskSchedulerTaskTrackerTest, | 422 TaskSchedulerTaskTrackerTest, |
| 421 ::testing::Values(TaskShutdownBehavior::BLOCK_SHUTDOWN)); | 423 ::testing::Values(TaskShutdownBehavior::BLOCK_SHUTDOWN)); |
| 422 | 424 |
| 423 } // namespace internal | 425 } // namespace internal |
| 424 } // namespace base | 426 } // namespace base |
| OLD | NEW |