1*b0d29bc4SBrooks Davis // Copyright 2011 The Kyua Authors.
2*b0d29bc4SBrooks Davis // All rights reserved.
3*b0d29bc4SBrooks Davis //
4*b0d29bc4SBrooks Davis // Redistribution and use in source and binary forms, with or without
5*b0d29bc4SBrooks Davis // modification, are permitted provided that the following conditions are
6*b0d29bc4SBrooks Davis // met:
7*b0d29bc4SBrooks Davis //
8*b0d29bc4SBrooks Davis // * Redistributions of source code must retain the above copyright
9*b0d29bc4SBrooks Davis //   notice, this list of conditions and the following disclaimer.
10*b0d29bc4SBrooks Davis // * Redistributions in binary form must reproduce the above copyright
11*b0d29bc4SBrooks Davis //   notice, this list of conditions and the following disclaimer in the
12*b0d29bc4SBrooks Davis //   documentation and/or other materials provided with the distribution.
13*b0d29bc4SBrooks Davis // * Neither the name of Google Inc. nor the names of its contributors
14*b0d29bc4SBrooks Davis //   may be used to endorse or promote products derived from this software
15*b0d29bc4SBrooks Davis //   without specific prior written permission.
16*b0d29bc4SBrooks Davis //
17*b0d29bc4SBrooks Davis // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18*b0d29bc4SBrooks Davis // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19*b0d29bc4SBrooks Davis // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20*b0d29bc4SBrooks Davis // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21*b0d29bc4SBrooks Davis // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22*b0d29bc4SBrooks Davis // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23*b0d29bc4SBrooks Davis // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24*b0d29bc4SBrooks Davis // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25*b0d29bc4SBrooks Davis // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26*b0d29bc4SBrooks Davis // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27*b0d29bc4SBrooks Davis // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28*b0d29bc4SBrooks Davis 
29*b0d29bc4SBrooks Davis extern "C" {
30*b0d29bc4SBrooks Davis #include <unistd.h>
31*b0d29bc4SBrooks Davis }
32*b0d29bc4SBrooks Davis 
33*b0d29bc4SBrooks Davis #include <atf-c++.hpp>
34*b0d29bc4SBrooks Davis 
35*b0d29bc4SBrooks Davis #include "engine/config.hpp"
36*b0d29bc4SBrooks Davis #include "engine/kyuafile.hpp"
37*b0d29bc4SBrooks Davis #include "engine/plain.hpp"
38*b0d29bc4SBrooks Davis #include "engine/scheduler.hpp"
39*b0d29bc4SBrooks Davis #include "model/metadata.hpp"
40*b0d29bc4SBrooks Davis #include "model/test_program.hpp"
41*b0d29bc4SBrooks Davis #include "utils/config/tree.ipp"
42*b0d29bc4SBrooks Davis #include "utils/env.hpp"
43*b0d29bc4SBrooks Davis #include "utils/fs/operations.hpp"
44*b0d29bc4SBrooks Davis #include "utils/fs/path.hpp"
45*b0d29bc4SBrooks Davis #include "utils/logging/operations.hpp"
46*b0d29bc4SBrooks Davis #include "utils/optional.ipp"
47*b0d29bc4SBrooks Davis #include "utils/passwd.hpp"
48*b0d29bc4SBrooks Davis 
49*b0d29bc4SBrooks Davis namespace config = utils::config;
50*b0d29bc4SBrooks Davis namespace fs = utils::fs;
51*b0d29bc4SBrooks Davis namespace passwd = utils::passwd;
52*b0d29bc4SBrooks Davis namespace scheduler = engine::scheduler;
53*b0d29bc4SBrooks Davis 
54*b0d29bc4SBrooks Davis using utils::none;
55*b0d29bc4SBrooks Davis 
56*b0d29bc4SBrooks Davis 
57*b0d29bc4SBrooks Davis namespace {
58*b0d29bc4SBrooks Davis 
59*b0d29bc4SBrooks Davis 
60*b0d29bc4SBrooks Davis /// Gets the path to an example file.
61*b0d29bc4SBrooks Davis ///
62*b0d29bc4SBrooks Davis /// \param name The name of the example file.
63*b0d29bc4SBrooks Davis ///
64*b0d29bc4SBrooks Davis /// \return A path to the desired example file.  This can either be inside the
65*b0d29bc4SBrooks Davis /// source tree before installing Kyua or in the target installation directory
66*b0d29bc4SBrooks Davis /// after installation.
67*b0d29bc4SBrooks Davis static fs::path
example_file(const char * name)68*b0d29bc4SBrooks Davis example_file(const char* name)
69*b0d29bc4SBrooks Davis {
70*b0d29bc4SBrooks Davis     const fs::path examplesdir(utils::getenv_with_default(
71*b0d29bc4SBrooks Davis         "KYUA_EXAMPLESDIR", KYUA_EXAMPLESDIR));
72*b0d29bc4SBrooks Davis     return examplesdir / name;
73*b0d29bc4SBrooks Davis }
74*b0d29bc4SBrooks Davis 
75*b0d29bc4SBrooks Davis 
76*b0d29bc4SBrooks Davis }  // anonymous namespace
77*b0d29bc4SBrooks Davis 
78*b0d29bc4SBrooks Davis 
79*b0d29bc4SBrooks Davis ATF_TEST_CASE(kyua_conf);
ATF_TEST_CASE_HEAD(kyua_conf)80*b0d29bc4SBrooks Davis ATF_TEST_CASE_HEAD(kyua_conf)
81*b0d29bc4SBrooks Davis {
82*b0d29bc4SBrooks Davis     utils::logging::set_inmemory();
83*b0d29bc4SBrooks Davis     set_md_var("require.files", example_file("kyua.conf").str());
84*b0d29bc4SBrooks Davis }
ATF_TEST_CASE_BODY(kyua_conf)85*b0d29bc4SBrooks Davis ATF_TEST_CASE_BODY(kyua_conf)
86*b0d29bc4SBrooks Davis {
87*b0d29bc4SBrooks Davis     std::vector< passwd::user > users;
88*b0d29bc4SBrooks Davis     users.push_back(passwd::user("nobody", 1, 2));
89*b0d29bc4SBrooks Davis     passwd::set_mock_users_for_testing(users);
90*b0d29bc4SBrooks Davis 
91*b0d29bc4SBrooks Davis     const config::tree user_config = engine::load_config(
92*b0d29bc4SBrooks Davis         example_file("kyua.conf"));
93*b0d29bc4SBrooks Davis 
94*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(
95*b0d29bc4SBrooks Davis         "x86_64",
96*b0d29bc4SBrooks Davis         user_config.lookup< config::string_node >("architecture"));
97*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(
98*b0d29bc4SBrooks Davis         16,
99*b0d29bc4SBrooks Davis         user_config.lookup< config::positive_int_node >("parallelism"));
100*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(
101*b0d29bc4SBrooks Davis         "amd64",
102*b0d29bc4SBrooks Davis         user_config.lookup< config::string_node >("platform"));
103*b0d29bc4SBrooks Davis 
104*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(
105*b0d29bc4SBrooks Davis         "nobody",
106*b0d29bc4SBrooks Davis         user_config.lookup< engine::user_node >("unprivileged_user").name);
107*b0d29bc4SBrooks Davis 
108*b0d29bc4SBrooks Davis     config::properties_map exp_test_suites;
109*b0d29bc4SBrooks Davis     exp_test_suites["test_suites.kyua.run_coredump_tests"] = "false";
110*b0d29bc4SBrooks Davis     exp_test_suites["test_suites.FreeBSD.iterations"] = "1000";
111*b0d29bc4SBrooks Davis     exp_test_suites["test_suites.FreeBSD.run_old_tests"] = "false";
112*b0d29bc4SBrooks Davis     exp_test_suites["test_suites.NetBSD.file_systems"] = "ffs lfs ext2fs";
113*b0d29bc4SBrooks Davis     exp_test_suites["test_suites.NetBSD.iterations"] = "100";
114*b0d29bc4SBrooks Davis     exp_test_suites["test_suites.NetBSD.run_broken_tests"] = "true";
115*b0d29bc4SBrooks Davis     ATF_REQUIRE(exp_test_suites == user_config.all_properties("test_suites"));
116*b0d29bc4SBrooks Davis }
117*b0d29bc4SBrooks Davis 
118*b0d29bc4SBrooks Davis 
119*b0d29bc4SBrooks Davis ATF_TEST_CASE(kyuafile_top__no_matches);
ATF_TEST_CASE_HEAD(kyuafile_top__no_matches)120*b0d29bc4SBrooks Davis ATF_TEST_CASE_HEAD(kyuafile_top__no_matches)
121*b0d29bc4SBrooks Davis {
122*b0d29bc4SBrooks Davis     utils::logging::set_inmemory();
123*b0d29bc4SBrooks Davis     set_md_var("require.files", example_file("Kyuafile.top").str());
124*b0d29bc4SBrooks Davis }
ATF_TEST_CASE_BODY(kyuafile_top__no_matches)125*b0d29bc4SBrooks Davis ATF_TEST_CASE_BODY(kyuafile_top__no_matches)
126*b0d29bc4SBrooks Davis {
127*b0d29bc4SBrooks Davis     scheduler::scheduler_handle handle = scheduler::setup();
128*b0d29bc4SBrooks Davis 
129*b0d29bc4SBrooks Davis     fs::mkdir(fs::path("root"), 0755);
130*b0d29bc4SBrooks Davis     const fs::path source_path = example_file("Kyuafile.top");
131*b0d29bc4SBrooks Davis     ATF_REQUIRE(::symlink(source_path.c_str(), "root/Kyuafile") != -1);
132*b0d29bc4SBrooks Davis 
133*b0d29bc4SBrooks Davis     atf::utils::create_file("root/file", "");
134*b0d29bc4SBrooks Davis     fs::mkdir(fs::path("root/subdir"), 0755);
135*b0d29bc4SBrooks Davis 
136*b0d29bc4SBrooks Davis     const engine::kyuafile kyuafile = engine::kyuafile::load(
137*b0d29bc4SBrooks Davis         fs::path("root/Kyuafile"), none, engine::default_config(), handle);
138*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(fs::path("root"), kyuafile.source_root());
139*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(fs::path("root"), kyuafile.build_root());
140*b0d29bc4SBrooks Davis     ATF_REQUIRE(kyuafile.test_programs().empty());
141*b0d29bc4SBrooks Davis 
142*b0d29bc4SBrooks Davis     handle.cleanup();
143*b0d29bc4SBrooks Davis }
144*b0d29bc4SBrooks Davis 
145*b0d29bc4SBrooks Davis 
146*b0d29bc4SBrooks Davis ATF_TEST_CASE(kyuafile_top__some_matches);
ATF_TEST_CASE_HEAD(kyuafile_top__some_matches)147*b0d29bc4SBrooks Davis ATF_TEST_CASE_HEAD(kyuafile_top__some_matches)
148*b0d29bc4SBrooks Davis {
149*b0d29bc4SBrooks Davis     utils::logging::set_inmemory();
150*b0d29bc4SBrooks Davis     set_md_var("require.files", example_file("Kyuafile.top").str());
151*b0d29bc4SBrooks Davis }
ATF_TEST_CASE_BODY(kyuafile_top__some_matches)152*b0d29bc4SBrooks Davis ATF_TEST_CASE_BODY(kyuafile_top__some_matches)
153*b0d29bc4SBrooks Davis {
154*b0d29bc4SBrooks Davis     scheduler::scheduler_handle handle = scheduler::setup();
155*b0d29bc4SBrooks Davis 
156*b0d29bc4SBrooks Davis     fs::mkdir(fs::path("root"), 0755);
157*b0d29bc4SBrooks Davis     const fs::path source_path = example_file("Kyuafile.top");
158*b0d29bc4SBrooks Davis     ATF_REQUIRE(::symlink(source_path.c_str(), "root/Kyuafile") != -1);
159*b0d29bc4SBrooks Davis 
160*b0d29bc4SBrooks Davis     atf::utils::create_file("root/file", "");
161*b0d29bc4SBrooks Davis 
162*b0d29bc4SBrooks Davis     fs::mkdir(fs::path("root/subdir1"), 0755);
163*b0d29bc4SBrooks Davis     atf::utils::create_file("root/subdir1/Kyuafile",
164*b0d29bc4SBrooks Davis                             "syntax(2)\n"
165*b0d29bc4SBrooks Davis                             "plain_test_program{name='a', test_suite='b'}\n");
166*b0d29bc4SBrooks Davis     atf::utils::create_file("root/subdir1/a", "");
167*b0d29bc4SBrooks Davis 
168*b0d29bc4SBrooks Davis     fs::mkdir(fs::path("root/subdir2"), 0755);
169*b0d29bc4SBrooks Davis     atf::utils::create_file("root/subdir2/Kyuafile",
170*b0d29bc4SBrooks Davis                             "syntax(2)\n"
171*b0d29bc4SBrooks Davis                             "plain_test_program{name='c', test_suite='d'}\n");
172*b0d29bc4SBrooks Davis     atf::utils::create_file("root/subdir2/c", "");
173*b0d29bc4SBrooks Davis     atf::utils::create_file("root/subdir2/Kyuafile.etc", "invalid");
174*b0d29bc4SBrooks Davis 
175*b0d29bc4SBrooks Davis     const engine::kyuafile kyuafile = engine::kyuafile::load(
176*b0d29bc4SBrooks Davis         fs::path("root/Kyuafile"), none, engine::default_config(), handle);
177*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(fs::path("root"), kyuafile.source_root());
178*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(fs::path("root"), kyuafile.build_root());
179*b0d29bc4SBrooks Davis 
180*b0d29bc4SBrooks Davis     const model::test_program exp_test_program_a = model::test_program_builder(
181*b0d29bc4SBrooks Davis         "plain", fs::path("subdir1/a"), fs::path("root").to_absolute(), "b")
182*b0d29bc4SBrooks Davis         .add_test_case("main")
183*b0d29bc4SBrooks Davis         .build();
184*b0d29bc4SBrooks Davis     const model::test_program exp_test_program_c = model::test_program_builder(
185*b0d29bc4SBrooks Davis         "plain", fs::path("subdir2/c"), fs::path("root").to_absolute(), "d")
186*b0d29bc4SBrooks Davis         .add_test_case("main")
187*b0d29bc4SBrooks Davis         .build();
188*b0d29bc4SBrooks Davis 
189*b0d29bc4SBrooks Davis     ATF_REQUIRE_EQ(2, kyuafile.test_programs().size());
190*b0d29bc4SBrooks Davis     ATF_REQUIRE((exp_test_program_a == *kyuafile.test_programs()[0] &&
191*b0d29bc4SBrooks Davis                  exp_test_program_c == *kyuafile.test_programs()[1])
192*b0d29bc4SBrooks Davis                 ||
193*b0d29bc4SBrooks Davis                 (exp_test_program_a == *kyuafile.test_programs()[1] &&
194*b0d29bc4SBrooks Davis                  exp_test_program_c == *kyuafile.test_programs()[0]));
195*b0d29bc4SBrooks Davis 
196*b0d29bc4SBrooks Davis     handle.cleanup();
197*b0d29bc4SBrooks Davis }
198*b0d29bc4SBrooks Davis 
199*b0d29bc4SBrooks Davis 
ATF_INIT_TEST_CASES(tcs)200*b0d29bc4SBrooks Davis ATF_INIT_TEST_CASES(tcs)
201*b0d29bc4SBrooks Davis {
202*b0d29bc4SBrooks Davis     scheduler::register_interface(
203*b0d29bc4SBrooks Davis         "plain", std::shared_ptr< scheduler::interface >(
204*b0d29bc4SBrooks Davis             new engine::plain_interface()));
205*b0d29bc4SBrooks Davis 
206*b0d29bc4SBrooks Davis     ATF_ADD_TEST_CASE(tcs, kyua_conf);
207*b0d29bc4SBrooks Davis 
208*b0d29bc4SBrooks Davis     ATF_ADD_TEST_CASE(tcs, kyuafile_top__no_matches);
209*b0d29bc4SBrooks Davis     ATF_ADD_TEST_CASE(tcs, kyuafile_top__some_matches);
210*b0d29bc4SBrooks Davis }
211