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

Side by Side Diff: mojo/edk/system/message_pipe_test_utils.cc

Issue 814543006: Move //mojo/{public, edk} underneath //third_party (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebase Created 5 years, 11 months 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
« no previous file with comments | « mojo/edk/system/message_pipe_test_utils.h ('k') | mojo/edk/system/message_pipe_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
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
3 // found in the LICENSE file.
4
5 #include "mojo/edk/system/message_pipe_test_utils.h"
6
7 #include "base/bind.h"
8 #include "base/threading/platform_thread.h" // For |Sleep()|.
9 #include "mojo/edk/system/channel.h"
10 #include "mojo/edk/system/channel_endpoint.h"
11 #include "mojo/edk/system/message_pipe.h"
12 #include "mojo/edk/system/waiter.h"
13
14 namespace mojo {
15 namespace system {
16 namespace test {
17
18 MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp,
19 MojoHandleSignals signals,
20 HandleSignalsState* signals_state) {
21 Waiter waiter;
22 waiter.Init();
23
24 MojoResult add_result =
25 mp->AddAwakable(0, &waiter, signals, 0, signals_state);
26 if (add_result != MOJO_RESULT_OK) {
27 return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK
28 : add_result;
29 }
30
31 MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE, nullptr);
32 mp->RemoveAwakable(0, &waiter, signals_state);
33 return wait_result;
34 }
35
36 ChannelThread::ChannelThread(embedder::PlatformSupport* platform_support)
37 : platform_support_(platform_support),
38 test_io_thread_(base::TestIOThread::kManualStart) {
39 }
40
41 ChannelThread::~ChannelThread() {
42 Stop();
43 }
44
45 void ChannelThread::Start(embedder::ScopedPlatformHandle platform_handle,
46 scoped_refptr<ChannelEndpoint> channel_endpoint) {
47 test_io_thread_.Start();
48 test_io_thread_.PostTaskAndWait(
49 FROM_HERE,
50 base::Bind(&ChannelThread::InitChannelOnIOThread, base::Unretained(this),
51 base::Passed(&platform_handle), channel_endpoint));
52 }
53
54 void ChannelThread::Stop() {
55 if (channel_) {
56 // Hack to flush write buffers before quitting.
57 // TODO(vtl): Remove this once |Channel| has a
58 // |FlushWriteBufferAndShutdown()| (or whatever).
59 while (!channel_->IsWriteBufferEmpty())
60 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(20));
61
62 test_io_thread_.PostTaskAndWait(
63 FROM_HERE, base::Bind(&ChannelThread::ShutdownChannelOnIOThread,
64 base::Unretained(this)));
65 }
66 test_io_thread_.Stop();
67 }
68
69 void ChannelThread::InitChannelOnIOThread(
70 embedder::ScopedPlatformHandle platform_handle,
71 scoped_refptr<ChannelEndpoint> channel_endpoint) {
72 CHECK_EQ(base::MessageLoop::current(), test_io_thread_.message_loop());
73 CHECK(platform_handle.is_valid());
74
75 // Create and initialize |Channel|.
76 channel_ = new Channel(platform_support_);
77 channel_->Init(RawChannel::Create(platform_handle.Pass()));
78
79 // Start the bootstrap endpoint.
80 // Note: On the "server" (parent process) side, we need not attach/run the
81 // endpoint immediately. However, on the "client" (child process) side, this
82 // *must* be done here -- otherwise, the |Channel| may receive/process
83 // messages (which it can do as soon as it's hooked up to the IO thread
84 // message loop, and that message loop runs) before the endpoint is attached.
85 channel_->SetBootstrapEndpoint(channel_endpoint);
86 }
87
88 void ChannelThread::ShutdownChannelOnIOThread() {
89 CHECK(channel_);
90 channel_->Shutdown();
91 channel_ = nullptr;
92 }
93
94 #if !defined(OS_IOS)
95 MultiprocessMessagePipeTestBase::MultiprocessMessagePipeTestBase()
96 : channel_thread_(&platform_support_) {
97 }
98
99 MultiprocessMessagePipeTestBase::~MultiprocessMessagePipeTestBase() {
100 }
101
102 void MultiprocessMessagePipeTestBase::Init(scoped_refptr<ChannelEndpoint> ep) {
103 channel_thread_.Start(helper_.server_platform_handle.Pass(), ep);
104 }
105 #endif
106
107 } // namespace test
108 } // namespace system
109 } // namespace mojo
OLDNEW
« no previous file with comments | « mojo/edk/system/message_pipe_test_utils.h ('k') | mojo/edk/system/message_pipe_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698