OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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 #ifndef MOJO_COMMON_TEST_MULTIPROCESS_TEST_BASE_H_ | 5 #ifndef MOJO_COMMON_TEST_MULTIPROCESS_TEST_HELPER_H_ |
6 #define MOJO_COMMON_TEST_MULTIPROCESS_TEST_BASE_H_ | 6 #define MOJO_COMMON_TEST_MULTIPROCESS_TEST_HELPER_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
11 #include "base/compiler_specific.h" | 11 #include "base/compiler_specific.h" |
12 #include "base/process/process_handle.h" | 12 #include "base/process/process_handle.h" |
13 #include "base/test/multiprocess_test.h" | 13 #include "base/test/multiprocess_test.h" |
| 14 #include "base/test/test_timeouts.h" |
14 #include "mojo/system/embedder/scoped_platform_handle.h" | 15 #include "mojo/system/embedder/scoped_platform_handle.h" |
15 #include "testing/multiprocess_func_list.h" | 16 #include "testing/multiprocess_func_list.h" |
16 | 17 |
17 namespace mojo { | 18 namespace mojo { |
18 | 19 |
19 namespace embedder { | 20 namespace embedder { |
20 class PlatformChannelPair; | 21 class PlatformChannelPair; |
21 } | 22 } |
22 | 23 |
23 namespace test { | 24 namespace test { |
24 | 25 |
25 class MultiprocessTestBase : public base::MultiProcessTest { | 26 class MultiprocessTestHelper { |
26 public: | 27 public: |
27 MultiprocessTestBase(); | 28 MultiprocessTestHelper(); |
28 virtual ~MultiprocessTestBase(); | 29 ~MultiprocessTestHelper(); |
29 | |
30 virtual void SetUp() OVERRIDE; | |
31 virtual void TearDown() OVERRIDE; | |
32 | 30 |
33 // Start a child process and run the "main" function "named" |test_child_name| | 31 // Start a child process and run the "main" function "named" |test_child_name| |
34 // declared using |MOJO_MULTIPROCESS_TEST_CHILD_MAIN()| (below). | 32 // declared using |MOJO_MULTIPROCESS_TEST_CHILD_MAIN()| (below). |
35 void StartChild(const std::string& test_child_name); | 33 void StartChild(const std::string& test_child_name); |
36 // Wait for the child process to terminate. | 34 // Wait for the child process to terminate. |
37 // Returns the exit code of the child process. Note that, though it's declared | 35 // Returns the exit code of the child process. Note that, though it's declared |
38 // to be an |int|, the exit code is subject to mangling by the OS. E.g., we | 36 // to be an |int|, the exit code is subject to mangling by the OS. E.g., we |
39 // usually return -1 on error in the child (e.g., if |test_child_name| was not | 37 // usually return -1 on error in the child (e.g., if |test_child_name| was not |
40 // found), but this is mangled to 255 on Linux. You should only rely on codes | 38 // found), but this is mangled to 255 on Linux. You should only rely on codes |
41 // 0-127 being preserved, and -1 being outside the range 0-127. | 39 // 0-127 being preserved, and -1 being outside the range 0-127. |
42 int WaitForChildShutdown(); | 40 int WaitForChildShutdown(); |
43 | 41 |
44 // For use by |MOJO_MULTIPROCESS_TEST_CHILD_MAIN()| only: | 42 // For use by |MOJO_MULTIPROCESS_TEST_CHILD_MAIN()| only: |
45 static void ChildSetup(); | 43 static void ChildSetup(); |
46 | 44 |
47 // For use in the main process: | 45 // For use in the main process: |
48 embedder::ScopedPlatformHandle server_platform_handle; | 46 embedder::ScopedPlatformHandle server_platform_handle; |
49 | 47 |
50 // For use (and only valid) in the child process: | 48 // For use (and only valid) in the child process: |
51 static embedder::ScopedPlatformHandle client_platform_handle; | 49 static embedder::ScopedPlatformHandle client_platform_handle; |
52 | 50 |
53 private: | 51 private: |
54 virtual CommandLine MakeCmdLine(const std::string& procname, | |
55 bool debug_on_start) OVERRIDE; | |
56 | |
57 scoped_ptr<embedder::PlatformChannelPair> platform_channel_pair_; | 52 scoped_ptr<embedder::PlatformChannelPair> platform_channel_pair_; |
58 | 53 |
59 // Valid after |StartChild()| and before |WaitForChildShutdown()|. | 54 // Valid after |StartChild()| and before |WaitForChildShutdown()|. |
60 base::ProcessHandle test_child_handle_; | 55 base::ProcessHandle test_child_handle_; |
61 | 56 |
62 DISALLOW_COPY_AND_ASSIGN(MultiprocessTestBase); | 57 DISALLOW_COPY_AND_ASSIGN(MultiprocessTestHelper); |
63 }; | 58 }; |
64 | 59 |
65 // Use this to declare the child process's "main()" function for tests using | 60 // Use this to declare the child process's "main()" function for tests using |
66 // |MultiprocessTestBase|. It returns an |int|, which will be the process's exit | 61 // |MultiprocessTestHelper|. It returns an |int|, which will be the process's |
67 // code (but see the comment about |WaitForChildShutdown()|). | 62 // exit code (but see the comment about |WaitForChildShutdown()|). |
68 #define MOJO_MULTIPROCESS_TEST_CHILD_MAIN(test_child_name) \ | 63 #define MOJO_MULTIPROCESS_TEST_CHILD_MAIN(test_child_name) \ |
69 MULTIPROCESS_TEST_MAIN_WITH_SETUP( \ | 64 MULTIPROCESS_TEST_MAIN_WITH_SETUP( \ |
70 test_child_name ## TestChildMain, \ | 65 test_child_name ## TestChildMain, \ |
71 ::mojo::test::MultiprocessTestBase::ChildSetup) | 66 ::mojo::test::MultiprocessTestHelper::ChildSetup) |
72 | 67 |
73 } // namespace test | 68 } // namespace test |
74 } // namespace mojo | 69 } // namespace mojo |
75 | 70 |
76 #endif // MOJO_COMMON_TEST_MULTIPROCESS_TEST_BASE_H_ | 71 #endif // MOJO_COMMON_TEST_MULTIPROCESS_TEST_HELPER_H_ |
OLD | NEW |