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(st, 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(st, 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(st, 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(st, 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