1*6a6c8299Sjmmv // Copyright 2010 Google Inc.
2*6a6c8299Sjmmv // All rights reserved.
3*6a6c8299Sjmmv //
4*6a6c8299Sjmmv // Redistribution and use in source and binary forms, with or without
5*6a6c8299Sjmmv // modification, are permitted provided that the following conditions are
6*6a6c8299Sjmmv // met:
7*6a6c8299Sjmmv //
8*6a6c8299Sjmmv // * Redistributions of source code must retain the above copyright
9*6a6c8299Sjmmv // notice, this list of conditions and the following disclaimer.
10*6a6c8299Sjmmv // * Redistributions in binary form must reproduce the above copyright
11*6a6c8299Sjmmv // notice, this list of conditions and the following disclaimer in the
12*6a6c8299Sjmmv // documentation and/or other materials provided with the distribution.
13*6a6c8299Sjmmv // * Neither the name of Google Inc. nor the names of its contributors
14*6a6c8299Sjmmv // may be used to endorse or promote products derived from this software
15*6a6c8299Sjmmv // without specific prior written permission.
16*6a6c8299Sjmmv //
17*6a6c8299Sjmmv // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18*6a6c8299Sjmmv // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19*6a6c8299Sjmmv // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20*6a6c8299Sjmmv // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21*6a6c8299Sjmmv // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22*6a6c8299Sjmmv // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23*6a6c8299Sjmmv // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24*6a6c8299Sjmmv // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25*6a6c8299Sjmmv // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26*6a6c8299Sjmmv // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27*6a6c8299Sjmmv // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28*6a6c8299Sjmmv
29*6a6c8299Sjmmv #include "engine/config.hpp"
30*6a6c8299Sjmmv
31*6a6c8299Sjmmv #if defined(HAVE_CONFIG_H)
32*6a6c8299Sjmmv # include "config.h"
33*6a6c8299Sjmmv #endif
34*6a6c8299Sjmmv
35*6a6c8299Sjmmv #include <stdexcept>
36*6a6c8299Sjmmv #include <vector>
37*6a6c8299Sjmmv
38*6a6c8299Sjmmv #include <atf-c++.hpp>
39*6a6c8299Sjmmv
40*6a6c8299Sjmmv #include "engine/exceptions.hpp"
41*6a6c8299Sjmmv #include "utils/cmdline/exceptions.hpp"
42*6a6c8299Sjmmv #include "utils/cmdline/parser.hpp"
43*6a6c8299Sjmmv #include "utils/config/tree.ipp"
44*6a6c8299Sjmmv #include "utils/passwd.hpp"
45*6a6c8299Sjmmv
46*6a6c8299Sjmmv namespace config = utils::config;
47*6a6c8299Sjmmv namespace fs = utils::fs;
48*6a6c8299Sjmmv namespace passwd = utils::passwd;
49*6a6c8299Sjmmv
50*6a6c8299Sjmmv using utils::none;
51*6a6c8299Sjmmv using utils::optional;
52*6a6c8299Sjmmv
53*6a6c8299Sjmmv
54*6a6c8299Sjmmv namespace {
55*6a6c8299Sjmmv
56*6a6c8299Sjmmv
57*6a6c8299Sjmmv /// Replaces the system user database with a fake one for testing purposes.
58*6a6c8299Sjmmv static void
set_mock_users(void)59*6a6c8299Sjmmv set_mock_users(void)
60*6a6c8299Sjmmv {
61*6a6c8299Sjmmv std::vector< passwd::user > users;
62*6a6c8299Sjmmv users.push_back(passwd::user("user1", 100, 150));
63*6a6c8299Sjmmv users.push_back(passwd::user("user2", 200, 250));
64*6a6c8299Sjmmv passwd::set_mock_users_for_testing(users);
65*6a6c8299Sjmmv }
66*6a6c8299Sjmmv
67*6a6c8299Sjmmv
68*6a6c8299Sjmmv /// Checks that the default values of a config object match our expectations.
69*6a6c8299Sjmmv ///
70*6a6c8299Sjmmv /// This fails the test case if any field of the input config object is not
71*6a6c8299Sjmmv /// what we expect.
72*6a6c8299Sjmmv ///
73*6a6c8299Sjmmv /// \param config The configuration to validate.
74*6a6c8299Sjmmv static void
validate_defaults(const config::tree & config)75*6a6c8299Sjmmv validate_defaults(const config::tree& config)
76*6a6c8299Sjmmv {
77*6a6c8299Sjmmv ATF_REQUIRE_EQ(
78*6a6c8299Sjmmv KYUA_ARCHITECTURE,
79*6a6c8299Sjmmv config.lookup< config::string_node >("architecture"));
80*6a6c8299Sjmmv
81*6a6c8299Sjmmv ATF_REQUIRE_EQ(
82*6a6c8299Sjmmv KYUA_PLATFORM,
83*6a6c8299Sjmmv config.lookup< config::string_node >("platform"));
84*6a6c8299Sjmmv
85*6a6c8299Sjmmv ATF_REQUIRE(!config.is_set("unprivileged_user"));
86*6a6c8299Sjmmv
87*6a6c8299Sjmmv ATF_REQUIRE(config.all_properties("test_suites").empty());
88*6a6c8299Sjmmv }
89*6a6c8299Sjmmv
90*6a6c8299Sjmmv
91*6a6c8299Sjmmv } // anonymous namespace
92*6a6c8299Sjmmv
93*6a6c8299Sjmmv
94*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(config__defaults);
ATF_TEST_CASE_BODY(config__defaults)95*6a6c8299Sjmmv ATF_TEST_CASE_BODY(config__defaults)
96*6a6c8299Sjmmv {
97*6a6c8299Sjmmv const config::tree user_config = engine::default_config();
98*6a6c8299Sjmmv validate_defaults(user_config);
99*6a6c8299Sjmmv }
100*6a6c8299Sjmmv
101*6a6c8299Sjmmv
102*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(config__load__defaults);
ATF_TEST_CASE_BODY(config__load__defaults)103*6a6c8299Sjmmv ATF_TEST_CASE_BODY(config__load__defaults)
104*6a6c8299Sjmmv {
105*6a6c8299Sjmmv atf::utils::create_file("config", "syntax(2)\n");
106*6a6c8299Sjmmv
107*6a6c8299Sjmmv const config::tree user_config = engine::load_config(fs::path("config"));
108*6a6c8299Sjmmv validate_defaults(user_config);
109*6a6c8299Sjmmv }
110*6a6c8299Sjmmv
111*6a6c8299Sjmmv
112*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(config__load__overrides);
ATF_TEST_CASE_BODY(config__load__overrides)113*6a6c8299Sjmmv ATF_TEST_CASE_BODY(config__load__overrides)
114*6a6c8299Sjmmv {
115*6a6c8299Sjmmv set_mock_users();
116*6a6c8299Sjmmv
117*6a6c8299Sjmmv atf::utils::create_file(
118*6a6c8299Sjmmv "config",
119*6a6c8299Sjmmv "syntax(2)\n"
120*6a6c8299Sjmmv "architecture = 'test-architecture'\n"
121*6a6c8299Sjmmv "platform = 'test-platform'\n"
122*6a6c8299Sjmmv "unprivileged_user = 'user2'\n"
123*6a6c8299Sjmmv "test_suites.mysuite.myvar = 'myvalue'\n");
124*6a6c8299Sjmmv
125*6a6c8299Sjmmv const config::tree user_config = engine::load_config(fs::path("config"));
126*6a6c8299Sjmmv
127*6a6c8299Sjmmv ATF_REQUIRE_EQ("test-architecture",
128*6a6c8299Sjmmv user_config.lookup_string("architecture"));
129*6a6c8299Sjmmv ATF_REQUIRE_EQ("test-platform",
130*6a6c8299Sjmmv user_config.lookup_string("platform"));
131*6a6c8299Sjmmv
132*6a6c8299Sjmmv const passwd::user& user = user_config.lookup< engine::user_node >(
133*6a6c8299Sjmmv "unprivileged_user");
134*6a6c8299Sjmmv ATF_REQUIRE_EQ("user2", user.name);
135*6a6c8299Sjmmv ATF_REQUIRE_EQ(200, user.uid);
136*6a6c8299Sjmmv
137*6a6c8299Sjmmv config::properties_map exp_test_suites;
138*6a6c8299Sjmmv exp_test_suites["test_suites.mysuite.myvar"] = "myvalue";
139*6a6c8299Sjmmv
140*6a6c8299Sjmmv ATF_REQUIRE(exp_test_suites == user_config.all_properties("test_suites"));
141*6a6c8299Sjmmv }
142*6a6c8299Sjmmv
143*6a6c8299Sjmmv
144*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(config__load__lua_error);
ATF_TEST_CASE_BODY(config__load__lua_error)145*6a6c8299Sjmmv ATF_TEST_CASE_BODY(config__load__lua_error)
146*6a6c8299Sjmmv {
147*6a6c8299Sjmmv atf::utils::create_file("config", "this syntax is invalid\n");
148*6a6c8299Sjmmv
149*6a6c8299Sjmmv ATF_REQUIRE_THROW(engine::load_error, engine::load_config(
150*6a6c8299Sjmmv fs::path("config")));
151*6a6c8299Sjmmv }
152*6a6c8299Sjmmv
153*6a6c8299Sjmmv
154*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(config__load__bad_syntax__version);
ATF_TEST_CASE_BODY(config__load__bad_syntax__version)155*6a6c8299Sjmmv ATF_TEST_CASE_BODY(config__load__bad_syntax__version)
156*6a6c8299Sjmmv {
157*6a6c8299Sjmmv atf::utils::create_file("config", "syntax(123)\n");
158*6a6c8299Sjmmv
159*6a6c8299Sjmmv ATF_REQUIRE_THROW_RE(engine::load_error,
160*6a6c8299Sjmmv "Unsupported config version 123",
161*6a6c8299Sjmmv engine::load_config(fs::path("config")));
162*6a6c8299Sjmmv }
163*6a6c8299Sjmmv
164*6a6c8299Sjmmv
165*6a6c8299Sjmmv ATF_TEST_CASE_WITHOUT_HEAD(config__load__missing_file);
ATF_TEST_CASE_BODY(config__load__missing_file)166*6a6c8299Sjmmv ATF_TEST_CASE_BODY(config__load__missing_file)
167*6a6c8299Sjmmv {
168*6a6c8299Sjmmv ATF_REQUIRE_THROW_RE(engine::load_error, "Load of 'missing' failed",
169*6a6c8299Sjmmv engine::load_config(fs::path("missing")));
170*6a6c8299Sjmmv }
171*6a6c8299Sjmmv
172*6a6c8299Sjmmv
ATF_INIT_TEST_CASES(tcs)173*6a6c8299Sjmmv ATF_INIT_TEST_CASES(tcs)
174*6a6c8299Sjmmv {
175*6a6c8299Sjmmv ATF_ADD_TEST_CASE(tcs, config__defaults);
176*6a6c8299Sjmmv ATF_ADD_TEST_CASE(tcs, config__load__defaults);
177*6a6c8299Sjmmv ATF_ADD_TEST_CASE(tcs, config__load__overrides);
178*6a6c8299Sjmmv ATF_ADD_TEST_CASE(tcs, config__load__lua_error);
179*6a6c8299Sjmmv ATF_ADD_TEST_CASE(tcs, config__load__bad_syntax__version);
180*6a6c8299Sjmmv ATF_ADD_TEST_CASE(tcs, config__load__missing_file);
181*6a6c8299Sjmmv }
182