OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "mojo/edk/system/message_pipe_test_utils.h" | 5 #include "mojo/edk/system/message_pipe_test_utils.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/threading/platform_thread.h" // For |Sleep()|. | 8 #include "base/threading/platform_thread.h" // For |Sleep()|. |
9 #include "mojo/edk/system/channel.h" | 9 #include "mojo/edk/system/channel.h" |
10 #include "mojo/edk/system/channel_endpoint.h" | 10 #include "mojo/edk/system/channel_endpoint.h" |
(...skipping 28 matching lines...) Expand all Loading... |
39 | 39 |
40 ChannelThread::~ChannelThread() { | 40 ChannelThread::~ChannelThread() { |
41 Stop(); | 41 Stop(); |
42 } | 42 } |
43 | 43 |
44 void ChannelThread::Start(embedder::ScopedPlatformHandle platform_handle, | 44 void ChannelThread::Start(embedder::ScopedPlatformHandle platform_handle, |
45 scoped_refptr<ChannelEndpoint> channel_endpoint) { | 45 scoped_refptr<ChannelEndpoint> channel_endpoint) { |
46 test_io_thread_.Start(); | 46 test_io_thread_.Start(); |
47 test_io_thread_.PostTaskAndWait( | 47 test_io_thread_.PostTaskAndWait( |
48 FROM_HERE, | 48 FROM_HERE, |
49 base::Bind(&ChannelThread::InitChannelOnIOThread, base::Unretained(this), | 49 base::Bind(&ChannelThread::InitChannelOnIOThread, |
50 base::Passed(&platform_handle), channel_endpoint)); | 50 base::Unretained(this), |
| 51 base::Passed(&platform_handle), |
| 52 channel_endpoint)); |
51 } | 53 } |
52 | 54 |
53 void ChannelThread::Stop() { | 55 void ChannelThread::Stop() { |
54 if (channel_.get()) { | 56 if (channel_.get()) { |
55 // Hack to flush write buffers before quitting. | 57 // Hack to flush write buffers before quitting. |
56 // TODO(vtl): Remove this once |Channel| has a | 58 // TODO(vtl): Remove this once |Channel| has a |
57 // |FlushWriteBufferAndShutdown()| (or whatever). | 59 // |FlushWriteBufferAndShutdown()| (or whatever). |
58 while (!channel_->IsWriteBufferEmpty()) | 60 while (!channel_->IsWriteBufferEmpty()) |
59 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(20)); | 61 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(20)); |
60 | 62 |
61 test_io_thread_.PostTaskAndWait( | 63 test_io_thread_.PostTaskAndWait( |
62 FROM_HERE, base::Bind(&ChannelThread::ShutdownChannelOnIOThread, | 64 FROM_HERE, |
63 base::Unretained(this))); | 65 base::Bind(&ChannelThread::ShutdownChannelOnIOThread, |
| 66 base::Unretained(this))); |
64 } | 67 } |
65 test_io_thread_.Stop(); | 68 test_io_thread_.Stop(); |
66 } | 69 } |
67 | 70 |
68 void ChannelThread::InitChannelOnIOThread( | 71 void ChannelThread::InitChannelOnIOThread( |
69 embedder::ScopedPlatformHandle platform_handle, | 72 embedder::ScopedPlatformHandle platform_handle, |
70 scoped_refptr<ChannelEndpoint> channel_endpoint) { | 73 scoped_refptr<ChannelEndpoint> channel_endpoint) { |
71 CHECK_EQ(base::MessageLoop::current(), test_io_thread_.message_loop()); | 74 CHECK_EQ(base::MessageLoop::current(), test_io_thread_.message_loop()); |
72 CHECK(platform_handle.is_valid()); | 75 CHECK(platform_handle.is_valid()); |
73 | 76 |
(...skipping 25 matching lines...) Expand all Loading... |
99 } | 102 } |
100 | 103 |
101 void MultiprocessMessagePipeTestBase::Init(scoped_refptr<ChannelEndpoint> ep) { | 104 void MultiprocessMessagePipeTestBase::Init(scoped_refptr<ChannelEndpoint> ep) { |
102 channel_thread_.Start(helper_.server_platform_handle.Pass(), ep); | 105 channel_thread_.Start(helper_.server_platform_handle.Pass(), ep); |
103 } | 106 } |
104 #endif | 107 #endif |
105 | 108 |
106 } // namespace test | 109 } // namespace test |
107 } // namespace system | 110 } // namespace system |
108 } // namespace mojo | 111 } // namespace mojo |
OLD | NEW |