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

Unified Diff: mojo/system/message_pipe_test_utils.cc

Issue 501983003: Mojo: add mojo_message_pipe_perftests (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Fixing Windows build error Created 6 years, 4 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « mojo/system/message_pipe_test_utils.h ('k') | mojo/system/multiprocess_message_pipe_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: mojo/system/message_pipe_test_utils.cc
diff --git a/mojo/system/message_pipe_test_utils.cc b/mojo/system/message_pipe_test_utils.cc
new file mode 100644
index 0000000000000000000000000000000000000000..3afc6371518a342da913730c6317cf16cbf056d3
--- /dev/null
+++ b/mojo/system/message_pipe_test_utils.cc
@@ -0,0 +1,110 @@
+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "mojo/system/message_pipe_test_utils.h"
+
+#include "base/bind.h"
+#include "base/threading/platform_thread.h" // For |Sleep()|.
+#include "mojo/system/waiter.h"
+
+namespace mojo {
+namespace system {
+namespace test {
+
+MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp,
+ MojoHandleSignals signals,
+ HandleSignalsState* signals_state) {
+ Waiter waiter;
+ waiter.Init();
+
+ MojoResult add_result = mp->AddWaiter(0, &waiter, signals, 0, signals_state);
+ if (add_result != MOJO_RESULT_OK) {
+ return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK
+ : add_result;
+ }
+
+ MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE, NULL);
+ mp->RemoveWaiter(0, &waiter, signals_state);
+ return wait_result;
+}
+
+ChannelThread::ChannelThread(embedder::PlatformSupport* platform_support)
+ : platform_support_(platform_support),
+ test_io_thread_(test::TestIOThread::kManualStart) {
+}
+
+ChannelThread::~ChannelThread() {
+ Stop();
+}
+
+void ChannelThread::Start(embedder::ScopedPlatformHandle platform_handle,
+ scoped_refptr<MessagePipe> message_pipe) {
+ test_io_thread_.Start();
+ test_io_thread_.PostTaskAndWait(
+ FROM_HERE,
+ base::Bind(&ChannelThread::InitChannelOnIOThread,
+ base::Unretained(this),
+ base::Passed(&platform_handle),
+ message_pipe));
+}
+
+void ChannelThread::Stop() {
+ if (channel_.get()) {
+ // Hack to flush write buffers before quitting.
+ // TODO(vtl): Remove this once |Channel| has a
+ // |FlushWriteBufferAndShutdown()| (or whatever).
+ while (!channel_->IsWriteBufferEmpty())
+ base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(20));
+
+ test_io_thread_.PostTaskAndWait(
+ FROM_HERE,
+ base::Bind(&ChannelThread::ShutdownChannelOnIOThread,
+ base::Unretained(this)));
+ }
+ test_io_thread_.Stop();
+}
+
+void ChannelThread::InitChannelOnIOThread(
+ embedder::ScopedPlatformHandle platform_handle,
+ scoped_refptr<MessagePipe> message_pipe) {
+ CHECK_EQ(base::MessageLoop::current(), test_io_thread_.message_loop());
+ CHECK(platform_handle.is_valid());
+
+ // Create and initialize |Channel|.
+ channel_ = new Channel(platform_support_);
+ CHECK(channel_->Init(RawChannel::Create(platform_handle.Pass())));
+
+ // Attach the message pipe endpoint.
+ // Note: On the "server" (parent process) side, we need not attach the
+ // message pipe endpoint immediately. However, on the "client" (child
+ // process) side, this *must* be done here -- otherwise, the |Channel| may
+ // receive/process messages (which it can do as soon as it's hooked up to
+ // the IO thread message loop, and that message loop runs) before the
+ // message pipe endpoint is attached.
+ CHECK_EQ(channel_->AttachMessagePipeEndpoint(message_pipe, 1),
+ Channel::kBootstrapEndpointId);
+ CHECK(channel_->RunMessagePipeEndpoint(Channel::kBootstrapEndpointId,
+ Channel::kBootstrapEndpointId));
+}
+
+void ChannelThread::ShutdownChannelOnIOThread() {
+ CHECK(channel_.get());
+ channel_->Shutdown();
+ channel_ = NULL;
+}
+
+MultiprocessMessagePipeTestBase::MultiprocessMessagePipeTestBase()
+ : channel_thread_(&platform_support_) {
+}
+
+MultiprocessMessagePipeTestBase::~MultiprocessMessagePipeTestBase() {
+}
+
+void MultiprocessMessagePipeTestBase::Init(scoped_refptr<MessagePipe> mp) {
+ channel_thread_.Start(helper_.server_platform_handle.Pass(), mp);
+}
+
+} // namespace test
+} // namespace system
+} // namespace mojo
« no previous file with comments | « mojo/system/message_pipe_test_utils.h ('k') | mojo/system/multiprocess_message_pipe_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698