| // Copyright (c) 2006, 2007 Julio M. Merino Vidal |
| // Copyright (c) 2008 Ilya Sokolov, Boris Schaeling |
| // Copyright (c) 2009 Boris Schaeling |
| // Copyright (c) 2010 Felipe Tanus, Boris Schaeling |
| // Copyright (c) 2011, 2012 Jeff Flinn, Boris Schaeling |
| // |
| // Distributed under the Boost Software License, Version 1.0. (See accompanying |
| // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) |
| |
| #define BOOST_TEST_MAIN |
| //#define BOOST_TEST_IGNORE_SIGCHLD |
| #include <boost/test/included/unit_test.hpp> |
| |
| #include <boost/process/error.hpp> |
| #include <boost/process/async.hpp> |
| #include <boost/process/io.hpp> |
| #include <boost/process/child.hpp> |
| |
| #include <boost/thread.hpp> |
| #include <future> |
| |
| #include <boost/system/error_code.hpp> |
| #include <boost/algorithm/string/predicate.hpp> |
| #include <boost/asio/deadline_timer.hpp> |
| |
| using namespace std; |
| |
| namespace bp = boost::process; |
| |
| #if __APPLE__ |
| auto abort_sig = signal(SIGALRM, +[](int){std::terminate();}); |
| #endif |
| |
| BOOST_AUTO_TEST_SUITE( async ); |
| |
| |
| BOOST_AUTO_TEST_CASE(async_wait, *boost::unit_test::timeout(5)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| using namespace boost::asio; |
| |
| boost::asio::io_context io_context; |
| |
| std::error_code ec; |
| |
| bool exit_called_for_c1 = false; |
| int exit_code_c1 = 0; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(2)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| bp::child c1(master_test_suite().argv[1], |
| "test", "--exit-code", "123", |
| ec, io_context, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| BOOST_CHECK(!exit_called_for_c1); |
| exit_code_c1 = exit; exit_called_for_c1=true; |
| BOOST_CHECK(!ec_in); |
| timeout.cancel(); |
| })); |
| BOOST_REQUIRE(!ec); |
| |
| bool exit_called_for_c2 = false; |
| int exit_code_c2 = 0; |
| bp::child c2(master_test_suite().argv[1], |
| "test", "--exit-code", "21", |
| ec, io_context, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| BOOST_CHECK(!exit_called_for_c2); |
| exit_code_c2 = exit; exit_called_for_c2=true; |
| BOOST_CHECK(!ec_in); |
| }) |
| ); |
| BOOST_REQUIRE(!ec); |
| |
| io_context.run(); |
| |
| BOOST_CHECK(exit_called_for_c1); |
| BOOST_CHECK_EQUAL(exit_code_c1, 123); |
| BOOST_CHECK_EQUAL(c1.exit_code(), 123); |
| |
| BOOST_CHECK(exit_called_for_c2); |
| BOOST_CHECK_EQUAL(exit_code_c2, 21); |
| BOOST_CHECK_EQUAL(c2.exit_code(), 21); |
| } |
| |
| BOOST_AUTO_TEST_CASE(async_wait_sync_wait, *boost::unit_test::timeout(5)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| using namespace boost::asio; |
| |
| boost::asio::io_context io_context; |
| |
| bool exit_called = false; |
| int exit_code = 0; |
| std::error_code ec; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(3)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| bp::child c1( |
| master_test_suite().argv[1], |
| "test", "--exit-code", "1", |
| ec |
| ); |
| BOOST_REQUIRE(!ec); |
| |
| bp::child c2( |
| master_test_suite().argv[1], |
| "test", "--exit-code", "2", "--wait", "1", |
| ec, |
| io_context, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| exit_code = exit; exit_called=true; |
| BOOST_CHECK(!ec_in); |
| timeout.cancel(); |
| }) |
| ); |
| BOOST_REQUIRE(!ec); |
| |
| io_context.run(); |
| // Regression test for #143: make sure the async SIGCHLD handler on POSIX does not reap the |
| // child c1 is watching (this will error if so) |
| c1.wait(ec); |
| BOOST_REQUIRE(!ec); |
| |
| BOOST_CHECK(exit_called); |
| BOOST_CHECK_EQUAL(exit_code, 2); |
| BOOST_CHECK_EQUAL(c2.exit_code(), 2); |
| } |
| |
| BOOST_AUTO_TEST_CASE(async_wait_different_contexts, *boost::unit_test::timeout(10)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| using namespace boost::asio; |
| |
| boost::asio::io_context io_context1; |
| boost::asio::io_context io_context2; |
| |
| boost::asio::deadline_timer timeout1{io_context1, boost::posix_time::seconds(2)}; |
| timeout1.async_wait([&](boost::system::error_code ec){if (!ec) io_context1.stop();}); |
| |
| boost::asio::deadline_timer timeout2{io_context2, boost::posix_time::seconds(7)}; |
| timeout2.async_wait([&](boost::system::error_code ec){if (!ec) io_context2.stop();}); |
| std::error_code ec; |
| |
| bool exit_called_for_c1 = false; |
| int exit_code_c1 = 0; |
| bp::child c1( |
| master_test_suite().argv[1], |
| "test", "--exit-code", "1", |
| ec, |
| io_context1, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| BOOST_CHECK(!exit_called_for_c1); |
| exit_code_c1 = exit; exit_called_for_c1=true; |
| BOOST_CHECK(!ec_in); |
| timeout1.cancel(); |
| }) |
| ); |
| BOOST_REQUIRE(!ec); |
| |
| bool exit_called_for_c2 = false; |
| int exit_code_c2 = 0; |
| bp::child c2( |
| master_test_suite().argv[1], |
| "test", "--exit-code", "2", "--wait", "4", |
| ec, |
| io_context2, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| BOOST_CHECK(!exit_called_for_c2); |
| exit_code_c2 = exit; exit_called_for_c2=true; |
| BOOST_CHECK(!ec_in); |
| timeout2.cancel(); |
| }) |
| ); |
| BOOST_REQUIRE(!ec); |
| |
| // Regression test for #143: make sure each io_context handles its own children |
| std::thread thr1{[&]{io_context1.run();}}; |
| std::thread thr2{[&]{io_context2.run();}}; |
| |
| thr1.join(); |
| thr2.join(); |
| c1.wait(ec); |
| BOOST_REQUIRE(!ec); |
| |
| BOOST_CHECK(exit_called_for_c1); |
| BOOST_CHECK_EQUAL(exit_code_c1, 1); |
| BOOST_CHECK_EQUAL(c1.exit_code(), 1); |
| BOOST_CHECK(exit_called_for_c2); |
| BOOST_CHECK_EQUAL(exit_code_c2, 2); |
| BOOST_CHECK_EQUAL(c2.exit_code(), 2); |
| } |
| |
| BOOST_AUTO_TEST_CASE(async_wait_abort, *boost::unit_test::timeout(5)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| using namespace boost::asio; |
| |
| boost::asio::io_context io_context; |
| |
| std::error_code ec; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(5)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| bool exit_called = false; |
| int exit_code = 0; |
| bp::child c( |
| master_test_suite().argv[1], |
| "test", "--abort", |
| ec, |
| io_context, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| BOOST_CHECK(!exit_called); |
| exit_code = exit; |
| exit_called=true; |
| BOOST_TEST_MESSAGE(ec_in.message()); |
| BOOST_CHECK(!ec_in); |
| timeout.cancel(); |
| }) |
| ); |
| BOOST_REQUIRE(!ec); |
| |
| io_context.run(); |
| |
| BOOST_CHECK(exit_called); |
| BOOST_CHECK_NE(exit_code, 0); |
| BOOST_CHECK_EQUAL(c.exit_code(), exit_code); |
| } |
| |
| |
| BOOST_AUTO_TEST_CASE(async_future, *boost::unit_test::timeout(3)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| using namespace boost::asio; |
| |
| boost::asio::io_context io_context; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(2)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| std::error_code ec; |
| std::future<int> fut; |
| bp::child c( |
| master_test_suite().argv[1], |
| "test", "--exit-code", "42", |
| ec, |
| io_context, |
| bp::on_exit=fut |
| ); |
| |
| BOOST_REQUIRE(!ec); |
| |
| io_context.run(); |
| |
| BOOST_REQUIRE(fut.valid()); |
| BOOST_CHECK_EQUAL(fut.get(), 42); |
| } |
| |
| |
| BOOST_AUTO_TEST_CASE(async_out_stream, *boost::unit_test::timeout(5)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| |
| boost::asio::io_context io_context; |
| |
| |
| std::error_code ec; |
| |
| boost::asio::streambuf buf; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(2)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| bp::child c(master_test_suite().argv[1], |
| "test", "--echo-stdout", "abc", |
| bp::std_out > buf, |
| io_context, |
| ec); |
| BOOST_REQUIRE(!ec); |
| |
| io_context.run(); |
| |
| std::istream istr(&buf); |
| |
| std::string line; |
| std::getline(istr, line); |
| BOOST_REQUIRE_GE(line.size(), 3u); |
| BOOST_CHECK(boost::algorithm::starts_with(line, "abc")); |
| c.wait(); |
| } |
| |
| |
| |
| BOOST_AUTO_TEST_CASE(async_in_stream, *boost::unit_test::timeout(5)) |
| { |
| |
| using boost::unit_test::framework::master_test_suite; |
| |
| boost::asio::io_context io_context; |
| |
| |
| std::error_code ec; |
| |
| boost::asio::streambuf buf; |
| boost::asio::streambuf in_buf; |
| |
| |
| std::ostream ostr(&in_buf); |
| ostr << "-string" << endl ; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(2)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| bp::child c( |
| master_test_suite().argv[1], |
| "test", "--prefix-once", "test", |
| bp::std_in < in_buf, |
| bp::std_out > buf, |
| io_context, |
| ec |
| ); |
| BOOST_REQUIRE(!ec); |
| |
| io_context.run(); |
| |
| std::istream istr(&buf); |
| |
| std::string line; |
| std::getline(istr, line); |
| |
| std::string val = "test-string"; |
| BOOST_REQUIRE_GE(line.size(), val.size()); |
| if (line >= val) |
| BOOST_CHECK(boost::algorithm::starts_with(line, val)); |
| |
| |
| c.wait(); |
| } |
| |
| |
| BOOST_AUTO_TEST_CASE(async_error, *boost::unit_test::timeout(3)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| using namespace boost::asio; |
| |
| boost::asio::io_context io_context; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(2)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| bool exit_called = false; |
| std::error_code ec; |
| bp::child c( |
| "doesn't exist", |
| ec, |
| io_context, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| exit_called=true; |
| }) |
| ); |
| |
| BOOST_REQUIRE(ec); |
| |
| io_context.run(); |
| |
| BOOST_CHECK(!exit_called); |
| } |
| |
| |
| /* |
| BOOST_AUTO_TEST_CASE(mixed_async, *boost::unit_test::timeout(5)) |
| { |
| using boost::unit_test::framework::master_test_suite; |
| using namespace boost::asio; |
| |
| boost::asio::io_context io_context; |
| |
| boost::asio::deadline_timer timeout{io_context, boost::posix_time::seconds(2)}; |
| timeout.async_wait([&](boost::system::error_code ec){if (!ec) io_context.stop();}); |
| |
| bool exit_called = false; |
| std::error_code ec; |
| |
| bp::child c(master_test_suite().argv[1], |
| "--wait", "1", "--exit-code", "42", |
| ec, |
| io_context, |
| bp::on_exit([&](int exit, const std::error_code& ec_in) |
| { |
| timeout.cancel(); |
| exit_called=true; |
| BOOST_CHECK_EQUAL(exit, 42); |
| }) |
| ); |
| |
| BOOST_REQUIRE(!ec); |
| std::thread thr([&]{c.wait();}); |
| io_context.run(); |
| |
| BOOST_CHECK(exit_called); |
| BOOST_CHECK_EQUAL(c.exit_code(), 42); |
| thr.join(); |
| |
| }*/ |
| |
| BOOST_AUTO_TEST_SUITE_END(); |