1*6b3a42afSjmmv // Copyright 2010 Google Inc.
2*6b3a42afSjmmv // All rights reserved.
3*6b3a42afSjmmv //
4*6b3a42afSjmmv // Redistribution and use in source and binary forms, with or without
5*6b3a42afSjmmv // modification, are permitted provided that the following conditions are
6*6b3a42afSjmmv // met:
7*6b3a42afSjmmv //
8*6b3a42afSjmmv // * Redistributions of source code must retain the above copyright
9*6b3a42afSjmmv // notice, this list of conditions and the following disclaimer.
10*6b3a42afSjmmv // * Redistributions in binary form must reproduce the above copyright
11*6b3a42afSjmmv // notice, this list of conditions and the following disclaimer in the
12*6b3a42afSjmmv // documentation and/or other materials provided with the distribution.
13*6b3a42afSjmmv // * Neither the name of Google Inc. nor the names of its contributors
14*6b3a42afSjmmv // may be used to endorse or promote products derived from this software
15*6b3a42afSjmmv // without specific prior written permission.
16*6b3a42afSjmmv //
17*6b3a42afSjmmv // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18*6b3a42afSjmmv // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19*6b3a42afSjmmv // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20*6b3a42afSjmmv // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21*6b3a42afSjmmv // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22*6b3a42afSjmmv // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23*6b3a42afSjmmv // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24*6b3a42afSjmmv // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25*6b3a42afSjmmv // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26*6b3a42afSjmmv // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27*6b3a42afSjmmv // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28*6b3a42afSjmmv
29*6b3a42afSjmmv #include "utils/env.hpp"
30*6b3a42afSjmmv
31*6b3a42afSjmmv #include <atf-c++.hpp>
32*6b3a42afSjmmv
33*6b3a42afSjmmv #include "utils/optional.ipp"
34*6b3a42afSjmmv
35*6b3a42afSjmmv using utils::optional;
36*6b3a42afSjmmv
37*6b3a42afSjmmv
38*6b3a42afSjmmv ATF_TEST_CASE_WITHOUT_HEAD(getallenv);
ATF_TEST_CASE_BODY(getallenv)39*6b3a42afSjmmv ATF_TEST_CASE_BODY(getallenv)
40*6b3a42afSjmmv {
41*6b3a42afSjmmv utils::unsetenv("test-missing");
42*6b3a42afSjmmv utils::setenv("test-empty", "");
43*6b3a42afSjmmv utils::setenv("test-text", "some-value");
44*6b3a42afSjmmv
45*6b3a42afSjmmv const std::map< std::string, std::string > allenv = utils::getallenv();
46*6b3a42afSjmmv
47*6b3a42afSjmmv {
48*6b3a42afSjmmv const std::map< std::string, std::string >::const_iterator iter =
49*6b3a42afSjmmv allenv.find("test-missing");
50*6b3a42afSjmmv ATF_REQUIRE(iter == allenv.end());
51*6b3a42afSjmmv }
52*6b3a42afSjmmv
53*6b3a42afSjmmv {
54*6b3a42afSjmmv const std::map< std::string, std::string >::const_iterator iter =
55*6b3a42afSjmmv allenv.find("test-empty");
56*6b3a42afSjmmv ATF_REQUIRE(iter != allenv.end());
57*6b3a42afSjmmv ATF_REQUIRE((*iter).second.empty());
58*6b3a42afSjmmv }
59*6b3a42afSjmmv
60*6b3a42afSjmmv {
61*6b3a42afSjmmv const std::map< std::string, std::string >::const_iterator iter =
62*6b3a42afSjmmv allenv.find("test-text");
63*6b3a42afSjmmv ATF_REQUIRE(iter != allenv.end());
64*6b3a42afSjmmv ATF_REQUIRE_EQ("some-value", (*iter).second);
65*6b3a42afSjmmv }
66*6b3a42afSjmmv
67*6b3a42afSjmmv if (utils::getenv("PATH")) {
68*6b3a42afSjmmv const std::map< std::string, std::string >::const_iterator iter =
69*6b3a42afSjmmv allenv.find("PATH");
70*6b3a42afSjmmv ATF_REQUIRE(iter != allenv.end());
71*6b3a42afSjmmv ATF_REQUIRE_EQ(utils::getenv("PATH").get(), (*iter).second);
72*6b3a42afSjmmv }
73*6b3a42afSjmmv }
74*6b3a42afSjmmv
75*6b3a42afSjmmv
76*6b3a42afSjmmv ATF_TEST_CASE_WITHOUT_HEAD(getenv);
ATF_TEST_CASE_BODY(getenv)77*6b3a42afSjmmv ATF_TEST_CASE_BODY(getenv)
78*6b3a42afSjmmv {
79*6b3a42afSjmmv const optional< std::string > path = utils::getenv("PATH");
80*6b3a42afSjmmv ATF_REQUIRE(path);
81*6b3a42afSjmmv ATF_REQUIRE(!path.get().empty());
82*6b3a42afSjmmv
83*6b3a42afSjmmv ATF_REQUIRE(!utils::getenv("__UNDEFINED_VARIABLE__"));
84*6b3a42afSjmmv }
85*6b3a42afSjmmv
86*6b3a42afSjmmv
87*6b3a42afSjmmv ATF_TEST_CASE_WITHOUT_HEAD(getenv_with_default);
ATF_TEST_CASE_BODY(getenv_with_default)88*6b3a42afSjmmv ATF_TEST_CASE_BODY(getenv_with_default)
89*6b3a42afSjmmv {
90*6b3a42afSjmmv ATF_REQUIRE("don't use" !=
91*6b3a42afSjmmv utils::getenv_with_default("PATH", "don't use"));
92*6b3a42afSjmmv
93*6b3a42afSjmmv ATF_REQUIRE_EQ("foo",
94*6b3a42afSjmmv utils::getenv_with_default("__UNDEFINED_VARIABLE__", "foo"));
95*6b3a42afSjmmv }
96*6b3a42afSjmmv
97*6b3a42afSjmmv
98*6b3a42afSjmmv ATF_TEST_CASE_WITHOUT_HEAD(setenv);
ATF_TEST_CASE_BODY(setenv)99*6b3a42afSjmmv ATF_TEST_CASE_BODY(setenv)
100*6b3a42afSjmmv {
101*6b3a42afSjmmv ATF_REQUIRE(utils::getenv("PATH"));
102*6b3a42afSjmmv const std::string oldval = utils::getenv("PATH").get();
103*6b3a42afSjmmv utils::setenv("PATH", "foo-bar");
104*6b3a42afSjmmv ATF_REQUIRE(utils::getenv("PATH").get() != oldval);
105*6b3a42afSjmmv ATF_REQUIRE_EQ("foo-bar", utils::getenv("PATH").get());
106*6b3a42afSjmmv
107*6b3a42afSjmmv ATF_REQUIRE(!utils::getenv("__UNDEFINED_VARIABLE__"));
108*6b3a42afSjmmv utils::setenv("__UNDEFINED_VARIABLE__", "foo2-bar2");
109*6b3a42afSjmmv ATF_REQUIRE_EQ("foo2-bar2", utils::getenv("__UNDEFINED_VARIABLE__").get());
110*6b3a42afSjmmv }
111*6b3a42afSjmmv
112*6b3a42afSjmmv
113*6b3a42afSjmmv ATF_TEST_CASE_WITHOUT_HEAD(unsetenv);
ATF_TEST_CASE_BODY(unsetenv)114*6b3a42afSjmmv ATF_TEST_CASE_BODY(unsetenv)
115*6b3a42afSjmmv {
116*6b3a42afSjmmv ATF_REQUIRE(utils::getenv("PATH"));
117*6b3a42afSjmmv utils::unsetenv("PATH");
118*6b3a42afSjmmv ATF_REQUIRE(!utils::getenv("PATH"));
119*6b3a42afSjmmv }
120*6b3a42afSjmmv
121*6b3a42afSjmmv
ATF_INIT_TEST_CASES(tcs)122*6b3a42afSjmmv ATF_INIT_TEST_CASES(tcs)
123*6b3a42afSjmmv {
124*6b3a42afSjmmv ATF_ADD_TEST_CASE(tcs, getallenv);
125*6b3a42afSjmmv ATF_ADD_TEST_CASE(tcs, getenv);
126*6b3a42afSjmmv ATF_ADD_TEST_CASE(tcs, getenv_with_default);
127*6b3a42afSjmmv ATF_ADD_TEST_CASE(tcs, setenv);
128*6b3a42afSjmmv ATF_ADD_TEST_CASE(tcs, unsetenv);
129*6b3a42afSjmmv }
130