1 // Copyright (c) 2006, 2007 Julio M. Merino Vidal
2 // Copyright (c) 2008 Ilya Sokolov, Boris Schaeling
3 // Copyright (c) 2009 Boris Schaeling
4 // Copyright (c) 2010 Felipe Tanus, Boris Schaeling
5 // Copyright (c) 2011, 2012 Jeff Flinn, Boris Schaeling
6 //
7 // Distributed under the Boost Software License, Version 1.0. (See accompanying
8 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
9
10 #define BOOST_TEST_MAIN
11
12 #include <boost/test/included/unit_test.hpp>
13
14 #include <boost/process/error.hpp>
15 #include <boost/process/async.hpp>
16 #include <boost/process/child.hpp>
17
18 #include <boost/system/error_code.hpp>
19 #include <boost/asio.hpp>
20 #if defined(BOOST_WINDOWS_API)
21 # include <windows.h>
22 typedef boost::asio::windows::stream_handle pipe_end;
23 #elif defined(BOOST_POSIX_API)
24 # include <sys/wait.h>
25 # include <signal.h>
26 typedef boost::asio::posix::stream_descriptor pipe_end;
27 #endif
28
29 namespace bp = boost::process;
30
BOOST_AUTO_TEST_CASE(sync_wait)31 BOOST_AUTO_TEST_CASE(sync_wait)
32 {
33 using boost::unit_test::framework::master_test_suite;
34
35 std::error_code ec;
36 bp::child c(
37 master_test_suite().argv[1],
38 "test", "--exit-code", "123",
39 ec
40 );
41 BOOST_REQUIRE(!ec);
42 c.wait();
43 int exit_code = c.exit_code();
44
45
46 BOOST_CHECK_EQUAL(123, exit_code);
47
48 c.wait();
49 }
50
BOOST_AUTO_TEST_CASE(sync_wait_abort)51 BOOST_AUTO_TEST_CASE(sync_wait_abort)
52 {
53 using boost::unit_test::framework::master_test_suite;
54
55 std::error_code ec;
56 bp::child c(
57 master_test_suite().argv[1],
58 "test", "--abort",
59 ec
60 );
61 BOOST_REQUIRE(!ec);
62 c.wait();
63 int exit_code = c.exit_code();
64
65
66 BOOST_CHECK(exit_code != 0);
67
68 c.wait();
69 }
70
71 #if defined(BOOST_WINDOWS_API)
72 struct wait_handler
73 {
74 HANDLE handle_;
75 bool &called_;
wait_handlerwait_handler76 wait_handler(HANDLE handle, bool &called) : handle_(handle), called_(called) {}
77
operator ()wait_handler78 void operator()(const boost::system::error_code &ec)
79 {
80 called_ = true;
81 BOOST_REQUIRE(!ec);
82 DWORD exit_code;
83 BOOST_REQUIRE(GetExitCodeProcess(handle_, &exit_code));
84 BOOST_CHECK_EQUAL(123, exit_code);
85 }
86 };
87 #elif defined(BOOST_POSIX_API)
88 struct wait_handler
89 {
90 bool &called_;
91
wait_handlerwait_handler92 wait_handler (bool & called) : called_(called) {}
93
operator ()wait_handler94 void operator()(const boost::system::error_code &ec, int signal)
95 {
96 called_ = true;
97 BOOST_REQUIRE(!ec);
98 BOOST_REQUIRE_EQUAL(SIGCHLD, signal);
99 int status;
100 wait(&status);
101 BOOST_CHECK_EQUAL(123, WEXITSTATUS(status));
102 }
103 };
104 #endif
105
BOOST_AUTO_TEST_CASE(async_wait)106 BOOST_AUTO_TEST_CASE(async_wait)
107 {
108 using boost::unit_test::framework::master_test_suite;
109 using namespace boost::asio;
110
111 boost::asio::io_context io_context;
112
113 boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(3)};
114 timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();});
115
116
117 bool wh_called = false;
118 #if defined(BOOST_POSIX_API)
119 signal_set set(io_context, SIGCHLD);
120 set.async_wait(wait_handler(wh_called));
121 #endif
122
123 std::error_code ec;
124 bp::child c(
125 master_test_suite().argv[1],
126 "test", "--exit-code", "123",
127 ec
128 );
129 BOOST_REQUIRE(!ec);
130
131 #if defined(BOOST_WINDOWS_API)
132 windows::object_handle handle(io_context.get_executor(), c.native_handle());
133 handle.async_wait(wait_handler(handle.native_handle(), wh_called));
134 #endif
135 io_context.run();
136 BOOST_CHECK_MESSAGE(wh_called, "Wait handler not called");
137 }
138
139
140
BOOST_AUTO_TEST_CASE(async_nowait)141 BOOST_AUTO_TEST_CASE(async_nowait)
142 {
143 // No need to call wait when passing an io_context
144 using boost::unit_test::framework::master_test_suite;
145 using namespace boost::asio;
146
147 std::error_code ec;
148 boost::asio::io_context io_context;
149 bp::child c(
150 master_test_suite().argv[1],
151 "test", "--exit-code", "221",
152 ec,
153 bp::on_exit=[](int exit_code, std::error_code) mutable {},
154 io_context
155 );
156 BOOST_REQUIRE(!ec);
157 io_context.run();
158 BOOST_CHECK_EQUAL(221, c.exit_code());
159 }