123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416 |
- // 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(), 3);
- 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();
|