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