Mojo: Add NotifyBadMessage API

mojo/edk/test/mojo_test_base.cc

 // Copyright 2016 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/edk/test/mojo_test_base.h"
 
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "mojo/edk/embedder/embedder.h"
@@ skipping 25 matching lines @@
     CHECK(g_mach_broker->Init());
     SetMachPortProvider(g_mach_broker);
   }
 #endif
 }
 
 MojoTestBase::~MojoTestBase() {}
 
 MojoTestBase::ClientController& MojoTestBase::StartClient(
     const std::string& client_name) {
-  clients_.push_back(base::WrapUnique(new ClientController(client_name, this)));
+  clients_.push_back(base::MakeUnique<ClientController>(
+      client_name, this, process_error_callback_));
   return *clients_.back();
 }
 
-MojoTestBase::ClientController::ClientController(const std::string& client_name,
-                                                 MojoTestBase* test) {
+MojoTestBase::ClientController::ClientController(
+    const std::string& client_name,
+    MojoTestBase* test,
+    const ProcessErrorCallback& process_error_callback) {
 #if !defined(OS_IOS)
 #if defined(OS_MACOSX)
   // This lock needs to be held while launching the child because the Mach port
   // broker only allows task ports to be received from known child processes.
   // However, it can only know the child process's pid after the child has
   // launched. To prevent a race where the child process sends its task port
   // before the pid has been registered, the lock needs to be held over both
   // launch and child pid registration.
   base::AutoLock lock(g_mach_broker->GetLock());
 #endif
+  helper_.set_process_error_callback(process_error_callback);
   pipe_ = helper_.StartChild(client_name);
 #if defined(OS_MACOSX)
   g_mach_broker->AddPlaceholderForPid(helper_.test_child().Handle());
 #endif
 #endif
 }
 
 MojoTestBase::ClientController::~ClientController() {
   CHECK(was_shutdown_)
       << "Test clients should be waited on explicitly with WaitForShutdown().";
@@ skipping 226 matching lines @@
                          MOJO_WRITE_DATA_FLAG_ALL_OR_NONE),
            MOJO_RESULT_OK);
   CHECK_EQ(num_bytes, static_cast<uint32_t>(size));
 
   return std::string(buffer.data(), buffer.size());
 }
 
 }  // namespace test
 }  // namespace edk
 }  // namespace mojo