Mojo: Add a Mojo-specific multiprocess test base class.

mojo/common/test/multiprocess_test_base.cc

+// Copyright 2013 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/common/test/multiprocess_test_base.h"
+
+#include "base/command_line.h"
+#include "base/logging.h"
+#include "base/process/kill.h"
+#include "base/process/process_handle.h"
+#include "build/build_config.h"
+
+namespace mojo {
+namespace test {
+
+MultiprocessTestBase::MultiprocessTestBase()
+    : test_child_handle_(base::kNullProcessHandle) {
+}
+
+MultiprocessTestBase::~MultiprocessTestBase() {
+  CHECK_EQ(test_child_handle_, base::kNullProcessHandle);
+}
+
+void MultiprocessTestBase::SetUp() {
+  CHECK_EQ(test_child_handle_, base::kNullProcessHandle);
+
+  MultiProcessTest::SetUp();
+
+  platform_server_channel_ =
+      system::PlatformServerChannel::Create("TestChannel");
+}
+
+void MultiprocessTestBase::TearDown() {
+  CHECK_EQ(test_child_handle_, base::kNullProcessHandle);
+
+  platform_server_channel_.reset();
+
+  MultiProcessTest::TearDown();
+}
+
+void MultiprocessTestBase::StartChild(const std::string& test_child_name) {
+  CHECK(platform_server_channel_.get());
+  CHECK(!test_child_name.empty());
+  CHECK_EQ(test_child_handle_, base::kNullProcessHandle);
+
+  std::string test_child_main = test_child_name + "TestChildMain";
+
+#if defined(OS_POSIX)
+  CommandLine unused(CommandLine::NO_PROGRAM);
+  base::FileHandleMappingVector fds_to_map;
+  platform_server_channel_->GetDataNeededToPassClientChannelToChildProcess(
+      &unused, &fds_to_map);
+  test_child_handle_ = SpawnChild(test_child_main, fds_to_map, false);
+#elif defined(OS_WIN)
+  test_child_handle_  = SpawnChild(test_child_main, false);
+#else
+#error "Not supported yet."
+#endif
+  platform_server_channel_->ChildProcessLaunched();
+
+  CHECK_NE(test_child_handle_, base::kNullProcessHandle);
+}
+
+int MultiprocessTestBase::WaitForChildShutdown() {
+  CHECK_NE(test_child_handle_, base::kNullProcessHandle);
+
+  static const int kTimeoutSeconds = 5;
+  int rv = -1;
+  CHECK(base::WaitForExitCodeWithTimeout(
+      test_child_handle_, &rv, base::TimeDelta::FromSeconds(kTimeoutSeconds)));
+  base::CloseProcessHandle(test_child_handle_);
+  test_child_handle_ = base::kNullProcessHandle;
+  return rv;
+}
+
+CommandLine MultiprocessTestBase::MakeCmdLine(const std::string& procname,
+                                              bool debug_on_start) {
+  CHECK(platform_server_channel_.get());
+
+  CommandLine command_line =
+      base::MultiProcessTest::MakeCmdLine(procname, debug_on_start);
+  base::FileHandleMappingVector unused;
+  platform_server_channel_->GetDataNeededToPassClientChannelToChildProcess(
+      &command_line, &unused);
+  return command_line;
+}
+
+// static
+void MultiprocessTestBase::ChildSetup() {
+  // TODO(vtl)
+  NOTIMPLEMENTED();
+}
+
+}  // namespace test
+}  // namespace mojo