[mojo] Ports EDK

mojo/edk/system/message_pipe_perftest.cc

 // Copyright 2014 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 <stddef.h>
 #include <stdint.h>
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/strings/stringprintf.h"
 #include "base/test/perf_time_logger.h"
 #include "mojo/edk/embedder/embedder.h"
 #include "mojo/edk/embedder/scoped_platform_handle.h"
 #include "mojo/edk/system/handle_signals_state.h"
-#include "mojo/edk/system/message_pipe_test_utils.h"
 #include "mojo/edk/system/test_utils.h"
+#include "mojo/edk/test/mojo_test_base.h"
 #include "mojo/edk/test/test_utils.h"
 #include "mojo/public/c/system/functions.h"
 #include "mojo/public/cpp/system/message_pipe.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace mojo {
 namespace edk {
 namespace {
 
-class MultiprocessMessagePipePerfTest
-    : public test::MultiprocessMessagePipeTestBase {
+class MultiprocessMessagePipePerfTest : public test::MojoTestBase {
  public:
   MultiprocessMessagePipePerfTest()
       : message_count_(0),
         message_size_(0) {}
 
   void SetUpMeasurement(int message_count, size_t message_size) {
     message_count_ = message_count;
     message_size_ = message_size;
     payload_ = std::string(message_size, '*');
     read_buffer_.resize(message_size * 2);
@@ skipping 42 matching lines @@
   size_t message_size_;
   std::string payload_;
   std::string read_buffer_;
   scoped_ptr<base::PerfTimeLogger> perf_logger_;
 };
 
 // For each message received, sends a reply message with the same contents
 // repeated twice, until the other end is closed or it receives "quitquitquit"
 // (which it doesn't reply to). It'll return the number of messages received,
 // not including any "quitquitquit" message, modulo 100.
-MOJO_MULTIPROCESS_TEST_CHILD_MAIN(PingPongClient) {
-  ScopedPlatformHandle client_platform_handle =
-      std::move(test::MultiprocessTestHelper::client_platform_handle);
-  CHECK(client_platform_handle.is_valid());
-  ScopedMessagePipeHandle mp =
-      CreateMessagePipe(std::move(client_platform_handle));
-
+DEFINE_TEST_CLIENT_WITH_PIPE(PingPongClient, MultiprocessMessagePipePerfTest,
+                             h) {
   std::string buffer(1000000, '\0');
   int rv = 0;
   while (true) {
     // Wait for our end of the message pipe to be readable.
     HandleSignalsState hss;
     MojoResult result =
-        MojoWait(mp.get().value(), MOJO_HANDLE_SIGNAL_READABLE,
+        MojoWait(h, MOJO_HANDLE_SIGNAL_READABLE,
                  MOJO_DEADLINE_INDEFINITE, &hss);
     if (result != MOJO_RESULT_OK) {
       rv = result;
       break;
     }
 
     uint32_t read_size = static_cast<uint32_t>(buffer.size());
-    CHECK_EQ(MojoReadMessage(mp.get().value(), &buffer[0],
+    CHECK_EQ(MojoReadMessage(h, &buffer[0],
                              &read_size, nullptr,
                              0, MOJO_READ_MESSAGE_FLAG_NONE),
              MOJO_RESULT_OK);
 
     // Empty message indicates quit.
     if (read_size == 0)
       break;
 
-    CHECK_EQ(MojoWriteMessage(mp.get().value(), &buffer[0],
+    CHECK_EQ(MojoWriteMessage(h, &buffer[0],
                               read_size,
                               nullptr, 0, MOJO_WRITE_MESSAGE_FLAG_NONE),
              MOJO_RESULT_OK);
   }
 
   return rv;
 }
 
 // Repeatedly sends messages as previous one got replied by the child.
 // Waits for the child to close its end before quitting once specified
 // number of messages has been sent.
 #if defined(OS_ANDROID)
 // Android multi-process tests are not executing the new process. This is flaky.
 #define MAYBE_PingPong DISABLED_PingPong
 #else
 #define MAYBE_PingPong PingPong
 #endif  // defined(OS_ANDROID)
 TEST_F(MultiprocessMessagePipePerfTest, MAYBE_PingPong) {
-  helper()->StartChild("PingPongClient");
+  RUN_CHILD_ON_PIPE(PingPongClient, h)
+    // This values are set to align with one at ipc_pertests.cc for comparison.
+    const size_t kMsgSize[5] = {12, 144, 1728, 20736, 248832};
+    const int kMessageCount[5] = {50000, 50000, 50000, 12000, 1000};
 
-  ScopedMessagePipeHandle mp =
-      CreateMessagePipe(std::move(helper()->server_platform_handle));
+    for (size_t i = 0; i < 5; i++) {
+      SetUpMeasurement(kMessageCount[i], kMsgSize[i]);
+      Measure(h);
+    }
 
-  // This values are set to align with one at ipc_pertests.cc for comparison.
-  const size_t kMsgSize[5] = {12, 144, 1728, 20736, 248832};
-  const int kMessageCount[5] = {50000, 50000, 50000, 12000, 1000};
-
-  for (size_t i = 0; i < 5; i++) {
-    SetUpMeasurement(kMessageCount[i], kMsgSize[i]);
-    Measure(mp.get().value());
-  }
-
-  SendQuitMessage(mp.get().value());
-  EXPECT_EQ(0, helper()->WaitForChildShutdown());
+    SendQuitMessage(h);
+  END_CHILD()
 }
 
 }  // namespace
 }  // namespace edk
 }  // namespace mojo