Chromium Code Reviews| Index: mojo/edk/test/multiprocess_test_helper.cc |
| diff --git a/mojo/edk/test/multiprocess_test_helper.cc b/mojo/edk/test/multiprocess_test_helper.cc |
| index b07866ef445e7617f22cfa9a3dca02fae03f50bf..af364cf0a9259d28208b2adc386cae1fe52b7333 100644 |
| --- a/mojo/edk/test/multiprocess_test_helper.cc |
| +++ b/mojo/edk/test/multiprocess_test_helper.cc |
| @@ -9,6 +9,7 @@ |
| #include "base/process/kill.h" |
| #include "base/process/process_handle.h" |
| #include "build/build_config.h" |
| +#include "mojo/edk/embedder/embedder.h" |
| #include "mojo/edk/embedder/platform_channel_pair.h" |
| #if defined(OS_WIN) |
| @@ -19,9 +20,12 @@ namespace mojo { |
| namespace edk { |
| namespace test { |
| +const char kBrokerHandleSwitch[] = "broker-switch"; |
|
yzshen1
2015/11/30 16:45:32
It seems "broker-handle" is more clear.
jam
2015/11/30 16:49:20
Done.
|
| + |
| MultiprocessTestHelper::MultiprocessTestHelper() { |
| platform_channel_pair_.reset(new PlatformChannelPair()); |
| server_platform_handle = platform_channel_pair_->PassServerHandle(); |
| + broker_platform_channel_pair_.reset(new PlatformChannelPair()); |
| } |
| MultiprocessTestHelper::~MultiprocessTestHelper() { |
| @@ -50,6 +54,10 @@ void MultiprocessTestHelper::StartChildWithExtraSwitch( |
| platform_channel_pair_->PrepareToPassClientHandleToChildProcess( |
| &command_line, &handle_passing_info); |
| + std::string broker_handle = broker_platform_channel_pair_-> |
| + PrepareToPassClientHandleToChildProcessAsString(&handle_passing_info); |
| + command_line.AppendSwitchASCII(kBrokerHandleSwitch, broker_handle); |
| + |
| if (!switch_string.empty()) { |
| CHECK(!command_line.HasSwitch(switch_string)); |
| if (!switch_value.empty()) |
| @@ -75,6 +83,10 @@ void MultiprocessTestHelper::StartChildWithExtraSwitch( |
| base::SpawnMultiProcessTestChild(test_child_main, command_line, options); |
| platform_channel_pair_->ChildProcessLaunched(); |
| + broker_platform_channel_pair_->ChildProcessLaunched(); |
| + ChildProcessLaunched(test_child_.Handle(), |
| + broker_platform_channel_pair_->PassServerHandle()); |
| + |
| CHECK(test_child_.IsValid()); |
| } |
| @@ -98,6 +110,14 @@ void MultiprocessTestHelper::ChildSetup() { |
| client_platform_handle = |
| PlatformChannelPair::PassClientHandleFromParentProcess( |
| *base::CommandLine::ForCurrentProcess()); |
| + |
| + std::string broker_handle_str = |
| + base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII( |
| + kBrokerHandleSwitch); |
| + ScopedPlatformHandle broker_handle = |
| + PlatformChannelPair::PassClientHandleFromParentProcessFromString( |
| + broker_handle_str); |
| + SetParentPipeHandle(broker_handle.Pass()); |
| } |
| // static |