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 <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include <memory> | 9 #include <memory> |
10 #include <vector> | 10 #include <vector> |
(...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
104 const Action action_; | 104 const Action action_; |
105 const bool expect_post_succeeds_; | 105 const bool expect_post_succeeds_; |
106 | 106 |
107 DISALLOW_COPY_AND_ASSIGN(ThreadPostingAndRunningTask); | 107 DISALLOW_COPY_AND_ASSIGN(ThreadPostingAndRunningTask); |
108 }; | 108 }; |
109 | 109 |
110 class ScopedSetSingletonAllowed { | 110 class ScopedSetSingletonAllowed { |
111 public: | 111 public: |
112 ScopedSetSingletonAllowed(bool singleton_allowed) | 112 ScopedSetSingletonAllowed(bool singleton_allowed) |
113 : previous_value_( | 113 : previous_value_( |
114 ThreadRestrictions::SetSingletonAllowed(singleton_allowed)) {} | 114 ThreadRestrictions::SetIOAllowed(singleton_allowed)) {} |
115 ~ScopedSetSingletonAllowed() { | 115 ~ScopedSetSingletonAllowed() { |
116 ThreadRestrictions::SetSingletonAllowed(previous_value_); | 116 ThreadRestrictions::SetIOAllowed(previous_value_); |
117 } | 117 } |
118 | 118 |
119 private: | 119 private: |
120 const bool previous_value_; | 120 const bool previous_value_; |
121 }; | 121 }; |
122 | 122 |
123 class TaskSchedulerTaskTrackerTest | 123 class TaskSchedulerTaskTrackerTest |
124 : public testing::TestWithParam<TaskShutdownBehavior> { | 124 : public testing::TestWithParam<TaskShutdownBehavior> { |
125 protected: | 125 protected: |
126 TaskSchedulerTaskTrackerTest() = default; | 126 TaskSchedulerTaskTrackerTest() = default; |
(...skipping 234 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
361 } | 361 } |
362 | 362 |
363 // Verify that BLOCK_SHUTDOWN and SKIP_ON_SHUTDOWN tasks can | 363 // Verify that BLOCK_SHUTDOWN and SKIP_ON_SHUTDOWN tasks can |
364 // AssertSingletonAllowed() but CONTINUE_ON_SHUTDOWN tasks can't. | 364 // AssertSingletonAllowed() but CONTINUE_ON_SHUTDOWN tasks can't. |
365 TEST_P(TaskSchedulerTaskTrackerTest, SingletonAllowed) { | 365 TEST_P(TaskSchedulerTaskTrackerTest, SingletonAllowed) { |
366 const bool can_use_singletons = | 366 const bool can_use_singletons = |
367 (GetParam() != TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN); | 367 (GetParam() != TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN); |
368 | 368 |
369 TaskTracker tracker; | 369 TaskTracker tracker; |
370 std::unique_ptr<Task> task( | 370 std::unique_ptr<Task> task( |
371 new Task(FROM_HERE, Bind(&ThreadRestrictions::AssertSingletonAllowed), | 371 new Task(FROM_HERE, Bind(&ThreadRestrictions::AssertIOAllowed), |
372 TaskTraits().WithShutdownBehavior(GetParam()), TimeDelta())); | 372 TaskTraits().WithShutdownBehavior(GetParam()), TimeDelta())); |
373 EXPECT_TRUE(tracker.WillPostTask(task.get())); | 373 EXPECT_TRUE(tracker.WillPostTask(task.get())); |
374 | 374 |
375 // Set the singleton allowed bit to the opposite of what it is expected to be | 375 // Set the singleton allowed bit to the opposite of what it is expected to be |
376 // when |tracker| runs |task| to verify that |tracker| actually sets the | 376 // when |tracker| runs |task| to verify that |tracker| actually sets the |
377 // correct value. | 377 // correct value. |
378 ScopedSetSingletonAllowed scoped_singleton_allowed(!can_use_singletons); | 378 ScopedSetSingletonAllowed scoped_singleton_allowed(!can_use_singletons); |
379 | 379 |
380 // Running the task should fail iff the task isn't allowed to use singletons. | 380 // Running the task should fail iff the task isn't allowed to use singletons. |
381 if (can_use_singletons) { | 381 if (can_use_singletons) { |
382 tracker.RunNextTaskInSequence( | 382 tracker.RunNextTaskInSequence( |
383 CreateSequenceWithTask(std::move(task)).get()); | 383 CreateSequenceWithTask(std::move(task)).get()); |
384 } else { | 384 } else { |
385 #if !defined(OS_LINUX) // http://crbug.com/634552 | |
386 EXPECT_DCHECK_DEATH( | 385 EXPECT_DCHECK_DEATH( |
387 { | 386 { |
388 tracker.RunNextTaskInSequence( | 387 tracker.RunNextTaskInSequence( |
389 CreateSequenceWithTask(std::move(task)).get()); | 388 CreateSequenceWithTask(std::move(task)).get()); |
390 }); | 389 }); |
391 #endif // !defined(OS_LINUX) | |
392 } | 390 } |
393 } | 391 } |
394 | 392 |
395 static void RunTaskRunnerHandleVerificationTask( | 393 static void RunTaskRunnerHandleVerificationTask( |
396 TaskTracker* tracker, | 394 TaskTracker* tracker, |
397 std::unique_ptr<Task> verify_task) { | 395 std::unique_ptr<Task> verify_task) { |
398 // Pretend |verify_task| is posted to respect TaskTracker's contract. | 396 // Pretend |verify_task| is posted to respect TaskTracker's contract. |
399 EXPECT_TRUE(tracker->WillPostTask(verify_task.get())); | 397 EXPECT_TRUE(tracker->WillPostTask(verify_task.get())); |
400 | 398 |
401 // Confirm that the test conditions are right (no TaskRunnerHandles set | 399 // Confirm that the test conditions are right (no TaskRunnerHandles set |
(...skipping 238 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
640 | 638 |
641 // Unblock shutdown by running |block_shutdown_task|. | 639 // Unblock shutdown by running |block_shutdown_task|. |
642 tracker_.RunNextTaskInSequence( | 640 tracker_.RunNextTaskInSequence( |
643 CreateSequenceWithTask(std::move(block_shutdown_task)).get()); | 641 CreateSequenceWithTask(std::move(block_shutdown_task)).get()); |
644 EXPECT_EQ(kLoadTestNumIterations + 1, NumTasksExecuted()); | 642 EXPECT_EQ(kLoadTestNumIterations + 1, NumTasksExecuted()); |
645 WAIT_FOR_ASYNC_SHUTDOWN_COMPLETED(); | 643 WAIT_FOR_ASYNC_SHUTDOWN_COMPLETED(); |
646 } | 644 } |
647 | 645 |
648 } // namespace internal | 646 } // namespace internal |
649 } // namespace base | 647 } // namespace base |
OLD | NEW |