| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/mojo_test_base.h" | 5 #include "mojo/edk/test/mojo_test_base.h" |
| 6 | 6 |
| 7 #include "base/memory/ptr_util.h" | 7 #include "base/memory/ptr_util.h" |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
| 10 #include "mojo/edk/embedder/embedder.h" | 10 #include "mojo/edk/embedder/embedder.h" |
| (...skipping 26 matching lines...) Expand all Loading... |
| 37 SetMachPortProvider(g_mach_broker); | 37 SetMachPortProvider(g_mach_broker); |
| 38 } | 38 } |
| 39 #endif | 39 #endif |
| 40 } | 40 } |
| 41 | 41 |
| 42 MojoTestBase::~MojoTestBase() {} | 42 MojoTestBase::~MojoTestBase() {} |
| 43 | 43 |
| 44 MojoTestBase::ClientController& MojoTestBase::StartClient( | 44 MojoTestBase::ClientController& MojoTestBase::StartClient( |
| 45 const std::string& client_name) { | 45 const std::string& client_name) { |
| 46 clients_.push_back(base::MakeUnique<ClientController>( | 46 clients_.push_back(base::MakeUnique<ClientController>( |
| 47 client_name, this, process_error_callback_)); | 47 client_name, this, process_error_callback_, launch_type_)); |
| 48 return *clients_.back(); | 48 return *clients_.back(); |
| 49 } | 49 } |
| 50 | 50 |
| 51 MojoTestBase::ClientController::ClientController( | 51 MojoTestBase::ClientController::ClientController( |
| 52 const std::string& client_name, | 52 const std::string& client_name, |
| 53 MojoTestBase* test, | 53 MojoTestBase* test, |
| 54 const ProcessErrorCallback& process_error_callback) { | 54 const ProcessErrorCallback& process_error_callback, |
| 55 LaunchType launch_type) { |
| 55 #if !defined(OS_IOS) | 56 #if !defined(OS_IOS) |
| 56 #if defined(OS_MACOSX) | 57 #if defined(OS_MACOSX) |
| 57 // This lock needs to be held while launching the child because the Mach port | 58 // This lock needs to be held while launching the child because the Mach port |
| 58 // broker only allows task ports to be received from known child processes. | 59 // broker only allows task ports to be received from known child processes. |
| 59 // However, it can only know the child process's pid after the child has | 60 // However, it can only know the child process's pid after the child has |
| 60 // launched. To prevent a race where the child process sends its task port | 61 // launched. To prevent a race where the child process sends its task port |
| 61 // before the pid has been registered, the lock needs to be held over both | 62 // before the pid has been registered, the lock needs to be held over both |
| 62 // launch and child pid registration. | 63 // launch and child pid registration. |
| 63 base::AutoLock lock(g_mach_broker->GetLock()); | 64 base::AutoLock lock(g_mach_broker->GetLock()); |
| 64 #endif | 65 #endif |
| 65 helper_.set_process_error_callback(process_error_callback); | 66 helper_.set_process_error_callback(process_error_callback); |
| 66 pipe_ = helper_.StartChild(client_name); | 67 pipe_ = helper_.StartChild(client_name, launch_type); |
| 67 #if defined(OS_MACOSX) | 68 #if defined(OS_MACOSX) |
| 68 g_mach_broker->AddPlaceholderForPid(helper_.test_child().Handle()); | 69 g_mach_broker->AddPlaceholderForPid(helper_.test_child().Handle()); |
| 69 #endif | 70 #endif |
| 70 #endif | 71 #endif |
| 71 } | 72 } |
| 72 | 73 |
| 73 MojoTestBase::ClientController::~ClientController() { | 74 MojoTestBase::ClientController::~ClientController() { |
| 74 CHECK(was_shutdown_) | 75 CHECK(was_shutdown_) |
| 75 << "Test clients should be waited on explicitly with WaitForShutdown()."; | 76 << "Test clients should be waited on explicitly with WaitForShutdown()."; |
| 76 } | 77 } |
| (...skipping 225 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 302 MOJO_WRITE_DATA_FLAG_ALL_OR_NONE), | 303 MOJO_WRITE_DATA_FLAG_ALL_OR_NONE), |
| 303 MOJO_RESULT_OK); | 304 MOJO_RESULT_OK); |
| 304 CHECK_EQ(num_bytes, static_cast<uint32_t>(size)); | 305 CHECK_EQ(num_bytes, static_cast<uint32_t>(size)); |
| 305 | 306 |
| 306 return std::string(buffer.data(), buffer.size()); | 307 return std::string(buffer.data(), buffer.size()); |
| 307 } | 308 } |
| 308 | 309 |
| 309 } // namespace test | 310 } // namespace test |
| 310 } // namespace edk | 311 } // namespace edk |
| 311 } // namespace mojo | 312 } // namespace mojo |
| OLD | NEW |