| 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 | #define BOOST_TEST_IGNORE_SIGCHLD |
| 12 | #include <boost/test/included/unit_test.hpp> |
| 13 | |
| 14 | #include <boost/process/args.hpp> |
| 15 | #include <boost/process/child.hpp> |
| 16 | #include <boost/process/env.hpp> |
| 17 | #include <boost/process/environment.hpp> |
| 18 | #include <boost/process/error.hpp> |
| 19 | #include <boost/process/io.hpp> |
| 20 | |
| 21 | #include <boost/algorithm/string/predicate.hpp> |
| 22 | |
| 23 | #include <boost/system/error_code.hpp> |
| 24 | |
| 25 | #include <boost/program_options/environment_iterator.hpp> |
| 26 | #include <string> |
| 27 | #include <stdlib.h> |
| 28 | #include <list> |
| 29 | |
| 30 | namespace bp = boost::process; |
| 31 | |
| 32 | BOOST_AUTO_TEST_CASE(inherit_env, *boost::unit_test::timeout(2)) |
| 33 | { |
| 34 | using boost::unit_test::framework::master_test_suite; |
| 35 | |
| 36 | bp::ipstream st; |
| 37 | |
| 38 | std::error_code ec; |
| 39 | |
| 40 | bp::child c( |
| 41 | master_test_suite().argv[1], |
| 42 | "test" , "--query" , "PATH" , |
| 43 | bp::std_out>st, |
| 44 | ec |
| 45 | ); |
| 46 | BOOST_REQUIRE(!ec); |
| 47 | |
| 48 | std::string s; |
| 49 | |
| 50 | std::getline(is&: st, str&: s); |
| 51 | |
| 52 | auto path = boost::this_process::environment()["PATH" ].to_string(); |
| 53 | |
| 54 | std::cout << "Path : '" << path << "'" << std::endl; |
| 55 | std::cout << "Value: '" << s << "'" << std::endl; |
| 56 | |
| 57 | if(!path.empty()) |
| 58 | { |
| 59 | auto size = (path.size() < s.size()) ? path.size() : s.size(); |
| 60 | |
| 61 | BOOST_CHECK_EQUAL_COLLECTIONS( |
| 62 | s.begin(), s. begin() + size, |
| 63 | path.begin(), path.begin() + size |
| 64 | ); |
| 65 | } |
| 66 | else |
| 67 | BOOST_CHECK(boost::starts_with(s, "************** empty environment **************" )); |
| 68 | c.wait(); |
| 69 | } |
| 70 | |
| 71 | |
| 72 | BOOST_AUTO_TEST_CASE(inherit_mod_env, *boost::unit_test::timeout(2)) |
| 73 | { |
| 74 | using boost::unit_test::framework::master_test_suite; |
| 75 | |
| 76 | auto ie = boost::this_process::environment(); |
| 77 | std::string value = "TestString" ; |
| 78 | ie["BOOST_PROCESS_TEST_1" ] = value; |
| 79 | |
| 80 | { |
| 81 | auto ie2 = boost::this_process::environment(); |
| 82 | auto val = ie2["BOOST_PROCESS_TEST_1" ]; |
| 83 | auto st = val.to_string(); |
| 84 | |
| 85 | BOOST_CHECK_EQUAL_COLLECTIONS( |
| 86 | st.begin(), st.end(), |
| 87 | value.begin(), value.end() |
| 88 | ); |
| 89 | } |
| 90 | bp::ipstream st; |
| 91 | |
| 92 | std::error_code ec; |
| 93 | bp::child c( |
| 94 | master_test_suite().argv[1], |
| 95 | "test" , "--query" , "BOOST_PROCESS_TEST_1" , |
| 96 | bp::std_out>st, |
| 97 | ec |
| 98 | ); |
| 99 | BOOST_REQUIRE(!ec); |
| 100 | |
| 101 | std::string s; |
| 102 | |
| 103 | std::getline(is&: st, str&: s); |
| 104 | |
| 105 | auto size = (value.size() < s.size()) ? value.size() : s.size(); |
| 106 | |
| 107 | BOOST_CHECK_EQUAL_COLLECTIONS( |
| 108 | s.begin(), s. begin() + size, |
| 109 | value.begin(), value.begin() + size |
| 110 | ); |
| 111 | c.wait(); |
| 112 | } |
| 113 | |
| 114 | |
| 115 | BOOST_AUTO_TEST_CASE(modifided_env, *boost::unit_test::timeout(2)) |
| 116 | { |
| 117 | using boost::unit_test::framework::master_test_suite; |
| 118 | |
| 119 | bp::ipstream st; |
| 120 | |
| 121 | boost::process::environment env = boost::this_process::environment(); //empty env, that would fail. |
| 122 | std::string value = "TestString" ; |
| 123 | env["BOOST_PROCESS_TEST_2" ] = value; |
| 124 | |
| 125 | |
| 126 | std::error_code ec; |
| 127 | bp::child c( |
| 128 | master_test_suite().argv[1], |
| 129 | "test" , "--query" , "BOOST_PROCESS_TEST_2" , |
| 130 | bp::std_out>st, |
| 131 | env, |
| 132 | ec |
| 133 | ); |
| 134 | BOOST_REQUIRE(!ec); |
| 135 | BOOST_REQUIRE(boost::this_process::environment().count(value) == 0); |
| 136 | |
| 137 | std::string s; |
| 138 | std::getline(is&: st, str&: s); |
| 139 | |
| 140 | BOOST_CHECK(boost::algorithm::starts_with(s, "TestString" )); |
| 141 | c.wait(); |
| 142 | } |
| 143 | |
| 144 | |
| 145 | BOOST_AUTO_TEST_CASE(append, *boost::unit_test::timeout(5)) |
| 146 | { |
| 147 | using boost::unit_test::framework::master_test_suite; |
| 148 | |
| 149 | bp::ipstream st; |
| 150 | BOOST_TEST_PASSPOINT(); |
| 151 | bp::environment e = boost::this_process::environment(); |
| 152 | |
| 153 | std::error_code ec; |
| 154 | BOOST_REQUIRE_GE(e.size(), 1u); |
| 155 | |
| 156 | std::list<std::string> arg = {"test" , "--query" , "BOOST_PROCESS_TEST_3" }; |
| 157 | bp::child c( |
| 158 | master_test_suite().argv[1], |
| 159 | bp::env["BOOST_PROCESS_TEST_3" ]="some_string" , |
| 160 | bp::env=e, |
| 161 | bp::env["BOOST_PROCESS_TEST_3" ]=boost::none, |
| 162 | bp::env["BOOST_PROCESS_TEST_3" ]+="some_fictional_path_42" , |
| 163 | bp::env["BOOST_PROCESS_TEST_3" ]+={"other" , "next" }, |
| 164 | bp::args=arg, |
| 165 | bp::std_out>st, |
| 166 | ec |
| 167 | ); |
| 168 | |
| 169 | BOOST_REQUIRE(!ec); |
| 170 | BOOST_WARN(c.running()); |
| 171 | std::string s; |
| 172 | |
| 173 | std::getline(is&: st, str&: s); |
| 174 | #if defined(BOOST_WINDOWS_API) |
| 175 | BOOST_CHECK(boost::starts_with(s, "some_fictional_path_42;other;next" )); |
| 176 | #else |
| 177 | BOOST_CHECK(boost::starts_with(s, "some_fictional_path_42:other:next" )); |
| 178 | #endif |
| 179 | c.wait(); |
| 180 | } |
| 181 | |