Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(129)

Side by Side Diff: base/threading/sequenced_task_runner_handle_unittest.cc

Issue 1423773003: Add SequencedTaskRunnerHandle to get a SequencedTaskRunner for the current thread / sequence. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: export SequenceToken Created 5 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698