| 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.h" | 5 #include "base/task_scheduler/task.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/location.h" | 8 #include "base/location.h" |
| 9 #include "base/task_scheduler/task_traits.h" | 9 #include "base/task_scheduler/task_traits.h" |
| 10 #include "base/time/time.h" | 10 #include "base/time/time.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 | 12 |
| 13 namespace base { | 13 namespace base { |
| 14 namespace internal { | 14 namespace internal { |
| 15 | 15 |
| 16 // Verify that the shutdown behavior of a BLOCK_SHUTDOWN delayed task is | 16 // Verify that the shutdown behavior of a BLOCK_SHUTDOWN delayed task is |
| 17 // adjusted to SKIP_ON_SHUTDOWN. The shutown behavior of other delayed tasks | 17 // adjusted to SKIP_ON_SHUTDOWN. The shutown behavior of other delayed tasks |
| 18 // should not change. | 18 // should not change. |
| 19 TEST(TaskSchedulerTaskTest, ShutdownBehaviorChangeWithDelay) { | 19 TEST(TaskSchedulerTaskTest, ShutdownBehaviorChangeWithDelay) { |
| 20 Task continue_on_shutdown(FROM_HERE, Bind(&DoNothing), | 20 Task continue_on_shutdown(FROM_HERE, BindOnce(&DoNothing), |
| 21 TaskTraits().WithShutdownBehavior( | 21 TaskTraits().WithShutdownBehavior( |
| 22 TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN), | 22 TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN), |
| 23 TimeDelta::FromSeconds(1)); | 23 TimeDelta::FromSeconds(1)); |
| 24 EXPECT_EQ(TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN, | 24 EXPECT_EQ(TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN, |
| 25 continue_on_shutdown.traits.shutdown_behavior()); | 25 continue_on_shutdown.traits.shutdown_behavior()); |
| 26 | 26 |
| 27 Task skip_on_shutdown( | 27 Task skip_on_shutdown( |
| 28 FROM_HERE, Bind(&DoNothing), | 28 FROM_HERE, BindOnce(&DoNothing), |
| 29 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::SKIP_ON_SHUTDOWN), | 29 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::SKIP_ON_SHUTDOWN), |
| 30 TimeDelta::FromSeconds(1)); | 30 TimeDelta::FromSeconds(1)); |
| 31 EXPECT_EQ(TaskShutdownBehavior::SKIP_ON_SHUTDOWN, | 31 EXPECT_EQ(TaskShutdownBehavior::SKIP_ON_SHUTDOWN, |
| 32 skip_on_shutdown.traits.shutdown_behavior()); | 32 skip_on_shutdown.traits.shutdown_behavior()); |
| 33 | 33 |
| 34 Task block_shutdown( | 34 Task block_shutdown( |
| 35 FROM_HERE, Bind(&DoNothing), | 35 FROM_HERE, BindOnce(&DoNothing), |
| 36 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::BLOCK_SHUTDOWN), | 36 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::BLOCK_SHUTDOWN), |
| 37 TimeDelta::FromSeconds(1)); | 37 TimeDelta::FromSeconds(1)); |
| 38 EXPECT_EQ(TaskShutdownBehavior::SKIP_ON_SHUTDOWN, | 38 EXPECT_EQ(TaskShutdownBehavior::SKIP_ON_SHUTDOWN, |
| 39 block_shutdown.traits.shutdown_behavior()); | 39 block_shutdown.traits.shutdown_behavior()); |
| 40 } | 40 } |
| 41 | 41 |
| 42 // Verify that the shutdown behavior of undelayed tasks is not adjusted. | 42 // Verify that the shutdown behavior of undelayed tasks is not adjusted. |
| 43 TEST(TaskSchedulerTaskTest, NoShutdownBehaviorChangeNoDelay) { | 43 TEST(TaskSchedulerTaskTest, NoShutdownBehaviorChangeNoDelay) { |
| 44 Task continue_on_shutdown(FROM_HERE, Bind(&DoNothing), | 44 Task continue_on_shutdown(FROM_HERE, BindOnce(&DoNothing), |
| 45 TaskTraits().WithShutdownBehavior( | 45 TaskTraits().WithShutdownBehavior( |
| 46 TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN), | 46 TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN), |
| 47 TimeDelta()); | 47 TimeDelta()); |
| 48 EXPECT_EQ(TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN, | 48 EXPECT_EQ(TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN, |
| 49 continue_on_shutdown.traits.shutdown_behavior()); | 49 continue_on_shutdown.traits.shutdown_behavior()); |
| 50 | 50 |
| 51 Task skip_on_shutdown( | 51 Task skip_on_shutdown( |
| 52 FROM_HERE, Bind(&DoNothing), | 52 FROM_HERE, BindOnce(&DoNothing), |
| 53 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::SKIP_ON_SHUTDOWN), | 53 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::SKIP_ON_SHUTDOWN), |
| 54 TimeDelta()); | 54 TimeDelta()); |
| 55 EXPECT_EQ(TaskShutdownBehavior::SKIP_ON_SHUTDOWN, | 55 EXPECT_EQ(TaskShutdownBehavior::SKIP_ON_SHUTDOWN, |
| 56 skip_on_shutdown.traits.shutdown_behavior()); | 56 skip_on_shutdown.traits.shutdown_behavior()); |
| 57 | 57 |
| 58 Task block_shutdown( | 58 Task block_shutdown( |
| 59 FROM_HERE, Bind(&DoNothing), | 59 FROM_HERE, BindOnce(&DoNothing), |
| 60 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::BLOCK_SHUTDOWN), | 60 TaskTraits().WithShutdownBehavior(TaskShutdownBehavior::BLOCK_SHUTDOWN), |
| 61 TimeDelta()); | 61 TimeDelta()); |
| 62 EXPECT_EQ(TaskShutdownBehavior::BLOCK_SHUTDOWN, | 62 EXPECT_EQ(TaskShutdownBehavior::BLOCK_SHUTDOWN, |
| 63 block_shutdown.traits.shutdown_behavior()); | 63 block_shutdown.traits.shutdown_behavior()); |
| 64 } | 64 } |
| 65 | 65 |
| 66 } // namespace internal | 66 } // namespace internal |
| 67 } // namespace base | 67 } // namespace base |
| OLD | NEW |