ceph/src/boost/libs/process/test/env.cpp

   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 }