OLD | NEW |
(Empty) | |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "base/bind.h" |
| 6 #include "base/callback.h" |
| 7 #include "base/location.h" |
| 8 #include "base/memory/ref_counted.h" |
| 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "base/message_loop/message_loop.h" |
| 11 #include "base/run_loop.h" |
| 12 #include "base/sequence_checker_impl.h" |
| 13 #include "base/sequenced_task_runner.h" |
| 14 #include "base/synchronization/waitable_event.h" |
| 15 #include "base/threading/sequenced_task_runner_handle.h" |
| 16 #include "base/threading/sequenced_worker_pool.h" |
| 17 #include "base/threading/simple_thread.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" |
| 19 |
| 20 namespace base { |
| 21 namespace { |
| 22 |
| 23 class SequencedTaskRunnerHandleTest : public ::testing::Test { |
| 24 protected: |
| 25 static void GetTaskRunner(const Closure& callback) { |
| 26 // Use SequenceCheckerImpl to make sure it's not a no-op in Release builds. |
| 27 scoped_ptr<SequenceCheckerImpl> sequence_checker(new SequenceCheckerImpl); |
| 28 ASSERT_TRUE(SequencedTaskRunnerHandle::IsSet()); |
| 29 scoped_refptr<SequencedTaskRunner> task_runner = |
| 30 SequencedTaskRunnerHandle::Get(); |
| 31 ASSERT_TRUE(task_runner); |
| 32 task_runner->PostTask( |
| 33 FROM_HERE, base::Bind(&SequencedTaskRunnerHandleTest::CheckValidThread, |
| 34 base::Passed(&sequence_checker), callback)); |
| 35 } |
| 36 |
| 37 private: |
| 38 static void CheckValidThread(scoped_ptr<SequenceCheckerImpl> sequence_checker, |
| 39 const Closure& callback) { |
| 40 EXPECT_TRUE(sequence_checker->CalledOnValidSequencedThread()); |
| 41 callback.Run(); |
| 42 } |
| 43 |
| 44 MessageLoop message_loop_; |
| 45 }; |
| 46 |
| 47 TEST_F(SequencedTaskRunnerHandleTest, FromMessageLoop) { |
| 48 RunLoop run_loop; |
| 49 GetTaskRunner(run_loop.QuitClosure()); |
| 50 run_loop.Run(); |
| 51 } |
| 52 |
| 53 TEST_F(SequencedTaskRunnerHandleTest, FromSequencedWorkerPool) { |
| 54 scoped_refptr<SequencedWorkerPool> pool(new SequencedWorkerPool(3, "Test")); |
| 55 WaitableEvent event(false, false); |
| 56 pool->PostSequencedWorkerTask( |
| 57 pool->GetSequenceToken(), FROM_HERE, |
| 58 base::Bind(&SequencedTaskRunnerHandleTest::GetTaskRunner, |
| 59 base::Bind(&WaitableEvent::Signal, base::Unretained(&event)))); |
| 60 event.Wait(); |
| 61 } |
| 62 |
| 63 class ThreadRunner : public DelegateSimpleThread::Delegate { |
| 64 public: |
| 65 void Run() override { |
| 66 ASSERT_FALSE(SequencedTaskRunnerHandle::IsSet()); |
| 67 } |
| 68 |
| 69 private: |
| 70 Closure callback_; |
| 71 }; |
| 72 |
| 73 TEST_F(SequencedTaskRunnerHandleTest, FromSimpleThread) { |
| 74 ThreadRunner thread_runner; |
| 75 DelegateSimpleThread thread(&thread_runner, "Background thread"); |
| 76 thread.Start(); |
| 77 thread.Join(); |
| 78 } |
| 79 |
| 80 } // namespace |
| 81 } // namespace base |
OLD | NEW |