src/mojo/edk/test/multiprocess_test_helper_unittest.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "mojo/edk/test/multiprocess_test_helper.h"
   6
   7 #include "base/logging.h"
   8 #include "build/build_config.h"
   9 #include "mojo/edk/embedder/scoped_platform_handle.h"
  10 #include "mojo/edk/test/test_utils.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 #if defined(OS_POSIX)
  14 #include <fcntl.h>
  15 #endif
  16
  17 namespace mojo {
  18 namespace test {
  19 namespace {
  20
  21 bool IsNonBlocking(const embedder::PlatformHandle& handle) {
  22 #if defined(OS_WIN)
  23   // Haven't figured out a way to query whether a HANDLE was created with
  24   // FILE_FLAG_OVERLAPPED.
  25   return true;
  26 #else
  27   return fcntl(handle.fd, F_GETFL) & O_NONBLOCK;
  28 #endif
  29 }
  30
  31 bool WriteByte(const embedder::PlatformHandle& handle, char c) {
  32   size_t bytes_written = 0;
  33   BlockingWrite(handle, &c, 1, &bytes_written);
  34   return bytes_written == 1;
  35 }
  36
  37 bool ReadByte(const embedder::PlatformHandle& handle, char* c) {
  38   size_t bytes_read = 0;
  39   BlockingRead(handle, c, 1, &bytes_read);
  40   return bytes_read == 1;
  41 }
  42
  43 typedef testing::Test MultiprocessTestHelperTest;
  44
  45 TEST_F(MultiprocessTestHelperTest, RunChild) {
  46   MultiprocessTestHelper helper;
  47   EXPECT_TRUE(helper.server_platform_handle.is_valid());
  48
  49   helper.StartChild("RunChild");
  50   EXPECT_EQ(123, helper.WaitForChildShutdown());
  51 }
  52
  53 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(RunChild) {
  54   CHECK(MultiprocessTestHelper::client_platform_handle.is_valid());
  55   return 123;
  56 }
  57
  58 TEST_F(MultiprocessTestHelperTest, TestChildMainNotFound) {
  59   MultiprocessTestHelper helper;
  60   helper.StartChild("NoSuchTestChildMain");
  61   int result = helper.WaitForChildShutdown();
  62   EXPECT_FALSE(result >= 0 && result <= 127);
  63 }
  64
  65 TEST_F(MultiprocessTestHelperTest, PassedChannel) {
  66   MultiprocessTestHelper helper;
  67   EXPECT_TRUE(helper.server_platform_handle.is_valid());
  68   helper.StartChild("PassedChannel");
  69
  70   // Take ownership of the handle.
  71   embedder::ScopedPlatformHandle handle = helper.server_platform_handle.Pass();
  72
  73   // The handle should be non-blocking.
  74   EXPECT_TRUE(IsNonBlocking(handle.get()));
  75
  76   // Write a byte.
  77   const char c = 'X';
  78   EXPECT_TRUE(WriteByte(handle.get(), c));
  79
  80   // It'll echo it back to us, incremented.
  81   char d = 0;
  82   EXPECT_TRUE(ReadByte(handle.get(), &d));
  83   EXPECT_EQ(c + 1, d);
  84
  85   // And return it, incremented again.
  86   EXPECT_EQ(c + 2, helper.WaitForChildShutdown());
  87 }
  88
  89 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(PassedChannel) {
  90   CHECK(MultiprocessTestHelper::client_platform_handle.is_valid());
  91
  92   // Take ownership of the handle.
  93   embedder::ScopedPlatformHandle handle =
  94       MultiprocessTestHelper::client_platform_handle.Pass();
  95
  96   // The handle should be non-blocking.
  97   EXPECT_TRUE(IsNonBlocking(handle.get()));
  98
  99   // Read a byte.
 100   char c = 0;
 101   EXPECT_TRUE(ReadByte(handle.get(), &c));
 102
 103   // Write it back, incremented.
 104   c++;
 105   EXPECT_TRUE(WriteByte(handle.get(), c));
 106
 107   // And return it, incremented again.
 108   c++;
 109   return static_cast<int>(c);
 110 }
 111
 112 TEST_F(MultiprocessTestHelperTest, ChildTestPasses) {
 113   MultiprocessTestHelper helper;
 114   EXPECT_TRUE(helper.server_platform_handle.is_valid());
 115   helper.StartChild("ChildTestPasses");
 116   EXPECT_TRUE(helper.WaitForChildTestShutdown());
 117 }
 118
 119 MOJO_MULTIPROCESS_TEST_CHILD_TEST(ChildTestPasses) {
 120   ASSERT_TRUE(MultiprocessTestHelper::client_platform_handle.is_valid());
 121   EXPECT_TRUE(
 122       IsNonBlocking(MultiprocessTestHelper::client_platform_handle.get()));
 123 }
 124
 125 TEST_F(MultiprocessTestHelperTest, ChildTestFailsAssert) {
 126   MultiprocessTestHelper helper;
 127   EXPECT_TRUE(helper.server_platform_handle.is_valid());
 128   helper.StartChild("ChildTestFailsAssert");
 129   EXPECT_FALSE(helper.WaitForChildTestShutdown());
 130 }
 131
 132 MOJO_MULTIPROCESS_TEST_CHILD_TEST(ChildTestFailsAssert) {
 133   ASSERT_FALSE(MultiprocessTestHelper::client_platform_handle.is_valid())
 134       << "DISREGARD: Expected failure in child process";
 135   ASSERT_FALSE(
 136       IsNonBlocking(MultiprocessTestHelper::client_platform_handle.get()))
 137       << "Not reached";
 138   CHECK(false) << "Not reached";
 139 }
 140
 141 TEST_F(MultiprocessTestHelperTest, ChildTestFailsExpect) {
 142   MultiprocessTestHelper helper;
 143   EXPECT_TRUE(helper.server_platform_handle.is_valid());
 144   helper.StartChild("ChildTestFailsExpect");
 145   EXPECT_FALSE(helper.WaitForChildTestShutdown());
 146 }
 147
 148 MOJO_MULTIPROCESS_TEST_CHILD_TEST(ChildTestFailsExpect) {
 149   EXPECT_FALSE(MultiprocessTestHelper::client_platform_handle.is_valid())
 150       << "DISREGARD: Expected failure #1 in child process";
 151   EXPECT_FALSE(
 152       IsNonBlocking(MultiprocessTestHelper::client_platform_handle.get()))
 153       << "DISREGARD: Expected failure #2 in child process";
 154 }
 155
 156 }  // namespace
 157 }  // namespace test
 158 }  // namespace mojo