OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/test/multiprocess_test_helper.h" | 5 #include "mojo/edk/test/multiprocess_test_helper.h" |
6 | 6 |
7 #include <functional> | 7 #include <functional> |
8 #include <set> | 8 #include <set> |
9 #include <utility> | 9 #include <utility> |
10 | 10 |
11 #include "base/base_switches.h" | 11 #include "base/base_switches.h" |
12 #include "base/bind.h" | 12 #include "base/bind.h" |
13 #include "base/command_line.h" | 13 #include "base/command_line.h" |
14 #include "base/files/file_path.h" | 14 #include "base/files/file_path.h" |
15 #include "base/logging.h" | 15 #include "base/logging.h" |
16 #include "base/memory/ref_counted.h" | 16 #include "base/memory/ref_counted.h" |
17 #include "base/process/kill.h" | 17 #include "base/process/kill.h" |
18 #include "base/process/process_handle.h" | 18 #include "base/process/process_handle.h" |
19 #include "base/run_loop.h" | 19 #include "base/run_loop.h" |
20 #include "base/strings/stringprintf.h" | 20 #include "base/strings/stringprintf.h" |
21 #include "base/task_runner.h" | 21 #include "base/task_runner.h" |
22 #include "base/thread_task_runner_handle.h" | 22 #include "base/thread_task_runner_handle.h" |
23 #include "build/build_config.h" | 23 #include "build/build_config.h" |
24 #include "mojo/edk/embedder/embedder.h" | 24 #include "mojo/edk/embedder/embedder.h" |
25 #include "mojo/edk/embedder/platform_channel_pair.h" | 25 #include "mojo/edk/embedder/platform_channel_pair.h" |
26 #include "testing/gtest/include/gtest/gtest.h" | 26 #include "testing/gtest/include/gtest/gtest.h" |
27 | 27 |
28 #if defined(OS_WIN) | 28 #if defined(OS_WIN) |
29 #include "base/win/windows_version.h" | 29 #include "base/win/windows_version.h" |
| 30 #elif defined(OS_MACOSX) && !defined(OS_IOS) |
| 31 #include "base/mac/mach_port_broker.h" |
30 #endif | 32 #endif |
31 | 33 |
32 namespace mojo { | 34 namespace mojo { |
33 namespace edk { | 35 namespace edk { |
34 namespace test { | 36 namespace test { |
35 | 37 |
36 namespace { | 38 namespace { |
37 | 39 |
38 const char kMojoPrimordialPipeToken[] = "mojo-primordial-pipe-token"; | 40 const char kMojoPrimordialPipeToken[] = "mojo-primordial-pipe-token"; |
39 | 41 |
(...skipping 101 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
141 } | 143 } |
142 | 144 |
143 // static | 145 // static |
144 void MultiprocessTestHelper::ChildSetup() { | 146 void MultiprocessTestHelper::ChildSetup() { |
145 CHECK(base::CommandLine::InitializedForCurrentProcess()); | 147 CHECK(base::CommandLine::InitializedForCurrentProcess()); |
146 | 148 |
147 primordial_pipe_token = base::CommandLine::ForCurrentProcess() | 149 primordial_pipe_token = base::CommandLine::ForCurrentProcess() |
148 ->GetSwitchValueASCII(kMojoPrimordialPipeToken); | 150 ->GetSwitchValueASCII(kMojoPrimordialPipeToken); |
149 CHECK(!primordial_pipe_token.empty()); | 151 CHECK(!primordial_pipe_token.empty()); |
150 | 152 |
| 153 #if defined(OS_MACOSX) && !defined(OS_IOS) |
| 154 CHECK(base::MachPortBroker::ChildSendTaskPortToParent("mojo_test")); |
| 155 #endif |
| 156 |
151 SetParentPipeHandle( | 157 SetParentPipeHandle( |
152 PlatformChannelPair::PassClientHandleFromParentProcess( | 158 PlatformChannelPair::PassClientHandleFromParentProcess( |
153 *base::CommandLine::ForCurrentProcess())); | 159 *base::CommandLine::ForCurrentProcess())); |
154 } | 160 } |
155 | 161 |
156 // static | 162 // static |
157 int MultiprocessTestHelper::RunClientMain( | 163 int MultiprocessTestHelper::RunClientMain( |
158 const base::Callback<int(MojoHandle)>& main) { | 164 const base::Callback<int(MojoHandle)>& main) { |
159 return RunClientFunction([main](MojoHandle handle){ | 165 return RunClientFunction([main](MojoHandle handle){ |
160 return main.Run(handle); | 166 return main.Run(handle); |
161 }); | 167 }); |
162 } | 168 } |
163 | 169 |
164 // static | 170 // static |
165 int MultiprocessTestHelper::RunClientTestMain( | 171 int MultiprocessTestHelper::RunClientTestMain( |
166 const base::Callback<void(MojoHandle)>& main) { | 172 const base::Callback<void(MojoHandle)>& main) { |
167 return RunClientFunction([main](MojoHandle handle) { | 173 return RunClientFunction([main](MojoHandle handle) { |
168 main.Run(handle); | 174 main.Run(handle); |
169 return (::testing::Test::HasFatalFailure() || | 175 return (::testing::Test::HasFatalFailure() || |
170 ::testing::Test::HasNonfatalFailure()) ? 1 : 0; | 176 ::testing::Test::HasNonfatalFailure()) ? 1 : 0; |
171 }); | 177 }); |
172 } | 178 } |
173 | 179 |
174 // static | 180 // static |
175 std::string MultiprocessTestHelper::primordial_pipe_token; | 181 std::string MultiprocessTestHelper::primordial_pipe_token; |
176 | 182 |
177 } // namespace test | 183 } // namespace test |
178 } // namespace edk | 184 } // namespace edk |
179 } // namespace mojo | 185 } // namespace mojo |
OLD | NEW |