src/mojo/common/test/multiprocess_test_helper.h

   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 #ifndef MOJO_COMMON_TEST_MULTIPROCESS_TEST_HELPER_H_
   6 #define MOJO_COMMON_TEST_MULTIPROCESS_TEST_HELPER_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/process/process_handle.h"
  13 #include "base/test/multiprocess_test.h"
  14 #include "base/test/test_timeouts.h"
  15 #include "mojo/embedder/scoped_platform_handle.h"
  16 #include "testing/multiprocess_func_list.h"
  17
  18 namespace mojo {
  19
  20 namespace embedder {
  21 class PlatformChannelPair;
  22 }
  23
  24 namespace test {
  25
  26 class MultiprocessTestHelper {
  27  public:
  28   MultiprocessTestHelper();
  29   ~MultiprocessTestHelper();
  30
  31   // Start a child process and run the "main" function "named" |test_child_name|
  32   // declared using |MOJO_MULTIPROCESS_TEST_CHILD_MAIN()| or
  33   // |MOJO_MULTIPROCESS_TEST_CHILD_TEST()| (below).
  34   void StartChild(const std::string& test_child_name);
  35   // Wait for the child process to terminate.
  36   // Returns the exit code of the child process. Note that, though it's declared
  37   // to be an |int|, the exit code is subject to mangling by the OS. E.g., we
  38   // usually return -1 on error in the child (e.g., if |test_child_name| was not
  39   // found), but this is mangled to 255 on Linux. You should only rely on codes
  40   // 0-127 being preserved, and -1 being outside the range 0-127.
  41   int WaitForChildShutdown();
  42
  43   // Like |WaitForChildShutdown()|, but returns true on success (exit code of 0)
  44   // and false otherwise. You probably want to do something like
  45   // |EXPECT_TRUE(WaitForChildTestShutdown());|. Mainly for use with
  46   // |MOJO_MULTIPROCESS_TEST_CHILD_TEST()|.
  47   bool WaitForChildTestShutdown();
  48
  49   // For use by |MOJO_MULTIPROCESS_TEST_CHILD_MAIN()| only:
  50   static void ChildSetup();
  51
  52   // For use in the main process:
  53   embedder::ScopedPlatformHandle server_platform_handle;
  54
  55   // For use (and only valid) in the child process:
  56   static embedder::ScopedPlatformHandle client_platform_handle;
  57
  58  private:
  59   scoped_ptr<embedder::PlatformChannelPair> platform_channel_pair_;
  60
  61   // Valid after |StartChild()| and before |WaitForChildShutdown()|.
  62   base::ProcessHandle test_child_handle_;
  63
  64   DISALLOW_COPY_AND_ASSIGN(MultiprocessTestHelper);
  65 };
  66
  67 // Use this to declare the child process's "main()" function for tests using
  68 // |MultiprocessTestHelper|. It returns an |int|, which will be the process's
  69 // exit code (but see the comment about |WaitForChildShutdown()|).
  70 #define MOJO_MULTIPROCESS_TEST_CHILD_MAIN(test_child_name) \
  71     MULTIPROCESS_TEST_MAIN_WITH_SETUP( \
  72         test_child_name ## TestChildMain, \
  73         ::mojo::test::MultiprocessTestHelper::ChildSetup)
  74
  75 // Use this (and |WaitForChildTestShutdown()|) for the child process's "main()",
  76 // if you want to use |EXPECT_...()| or |ASSERT_...()|; it has a |void| return
  77 // type. (Note that while an |ASSERT_...()| failure will abort the test in the
  78 // child, it will not abort the test in the parent.)
  79 #define MOJO_MULTIPROCESS_TEST_CHILD_TEST(test_child_name) \
  80     void test_child_name ## TestChildTest(); \
  81     MOJO_MULTIPROCESS_TEST_CHILD_MAIN(test_child_name) { \
  82       test_child_name ## TestChildTest(); \
  83       return (::testing::Test::HasFatalFailure() || \
  84               ::testing::Test::HasNonfatalFailure()) ? 1 : 0; \
  85     } \
  86     void test_child_name ## TestChildTest()
  87
  88 }  // namespace test
  89 }  // namespace mojo
  90
  91 #endif  // MOJO_COMMON_TEST_MULTIPROCESS_TEST_HELPER_H_