3e29e810ae
That way, if the test directory does not exist we don't leave behind temporary files (as in that case or on test failure the cleanup actions don't run).
283 lines
8.4 KiB
C
283 lines
8.4 KiB
C
/***
|
|
This file is part of systemd.
|
|
|
|
Copyright 2014 Ronny Chevalier
|
|
|
|
systemd is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation; either version 2.1 of the License, or
|
|
(at your option) any later version.
|
|
|
|
systemd is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with systemd; If not, see <http://www.gnu.org/licenses/>.
|
|
***/
|
|
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
|
|
#include "alloc-util.h"
|
|
#include "fd-util.h"
|
|
#include "fs-util.h"
|
|
#include "macro.h"
|
|
#include "manager.h"
|
|
#include "mkdir.h"
|
|
#include "rm-rf.h"
|
|
#include "string-util.h"
|
|
#include "strv.h"
|
|
#include "test-helper.h"
|
|
#include "tests.h"
|
|
#include "unit.h"
|
|
#include "util.h"
|
|
|
|
typedef void (*test_function_t)(Manager *m);
|
|
|
|
static int setup_test(Manager **m) {
|
|
char **tests_path = STRV_MAKE("exists", "existsglobFOOBAR", "changed", "modified", "unit",
|
|
"directorynotempty", "makedirectory");
|
|
char **test_path;
|
|
Manager *tmp = NULL;
|
|
int r;
|
|
|
|
assert_se(m);
|
|
|
|
r = manager_new(UNIT_FILE_USER, true, &tmp);
|
|
if (MANAGER_SKIP_TEST(r)) {
|
|
log_notice_errno(r, "Skipping test: manager_new: %m");
|
|
return -EXIT_TEST_SKIP;
|
|
}
|
|
assert_se(r >= 0);
|
|
assert_se(manager_startup(tmp, NULL, NULL) >= 0);
|
|
|
|
STRV_FOREACH(test_path, tests_path) {
|
|
_cleanup_free_ char *p = NULL;
|
|
|
|
p = strjoin("/tmp/test-path_", *test_path);
|
|
assert_se(p);
|
|
|
|
(void) rm_rf(p, REMOVE_ROOT|REMOVE_PHYSICAL);
|
|
}
|
|
|
|
*m = tmp;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void shutdown_test(Manager *m) {
|
|
assert_se(m);
|
|
|
|
manager_free(m);
|
|
}
|
|
|
|
static void check_stop_unlink(Manager *m, Unit *unit, const char *test_path, const char *service_name) {
|
|
_cleanup_free_ char *tmp = NULL;
|
|
Unit *service_unit = NULL;
|
|
Service *service = NULL;
|
|
usec_t ts;
|
|
usec_t timeout = 2 * USEC_PER_SEC;
|
|
|
|
assert_se(m);
|
|
assert_se(unit);
|
|
assert_se(test_path);
|
|
|
|
if (!service_name) {
|
|
assert_se(tmp = strreplace(unit->id, ".path", ".service"));
|
|
service_unit = manager_get_unit(m, tmp);
|
|
} else
|
|
service_unit = manager_get_unit(m, service_name);
|
|
assert_se(service_unit);
|
|
service = SERVICE(service_unit);
|
|
|
|
ts = now(CLOCK_MONOTONIC);
|
|
/* We process events until the service related to the path has been successfully started */
|
|
while (service->result != SERVICE_SUCCESS || service->state != SERVICE_START) {
|
|
usec_t n;
|
|
int r;
|
|
|
|
r = sd_event_run(m->event, 100 * USEC_PER_MSEC);
|
|
assert_se(r >= 0);
|
|
|
|
printf("%s: state = %s; result = %s \n",
|
|
service_unit->id,
|
|
service_state_to_string(service->state),
|
|
service_result_to_string(service->result));
|
|
|
|
|
|
/* But we timeout if the service has not been started in the allocated time */
|
|
n = now(CLOCK_MONOTONIC);
|
|
if (ts + timeout < n) {
|
|
log_error("Test timeout when testing %s", unit->id);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
assert_se(UNIT_VTABLE(unit)->stop(unit) >= 0);
|
|
(void) rm_rf(test_path, REMOVE_ROOT|REMOVE_PHYSICAL);
|
|
}
|
|
|
|
static void test_path_exists(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_exists";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(manager_load_unit(m, "path-exists.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_existsglob(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_existsglobFOOBAR";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
assert_se(manager_load_unit(m, "path-existsglob.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_changed(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_changed";
|
|
FILE *f;
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-changed.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
f = fopen(test_path, "w");
|
|
assert_se(f);
|
|
fclose(f);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_modified(Manager *m) {
|
|
_cleanup_fclose_ FILE *f = NULL;
|
|
const char *test_path = "/tmp/test-path_modified";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-modified.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
f = fopen(test_path, "w");
|
|
assert_se(f);
|
|
fputs("test", f);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_unit(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_unit";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(manager_load_unit(m, "path-unit.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, "path-mycustomunit.service");
|
|
}
|
|
|
|
static void test_path_directorynotempty(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_directorynotempty/";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(access(test_path, F_OK) < 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-directorynotempty.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
/* MakeDirectory default to no */
|
|
assert_se(access(test_path, F_OK) < 0);
|
|
|
|
assert_se(mkdir_p(test_path, 0755) >= 0);
|
|
assert_se(touch(strjoina(test_path, "test_file")) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_makedirectory_directorymode(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_makedirectory/";
|
|
Unit *unit = NULL;
|
|
struct stat s;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(access(test_path, F_OK) < 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-makedirectory.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
/* Check if the directory has been created */
|
|
assert_se(access(test_path, F_OK) >= 0);
|
|
|
|
/* Check the mode we specified with DirectoryMode=0744 */
|
|
assert_se(stat(test_path, &s) >= 0);
|
|
assert_se((s.st_mode & S_IRWXU) == 0700);
|
|
assert_se((s.st_mode & S_IRWXG) == 0040);
|
|
assert_se((s.st_mode & S_IRWXO) == 0004);
|
|
|
|
assert_se(UNIT_VTABLE(unit)->stop(unit) >= 0);
|
|
(void) rm_rf(test_path, REMOVE_ROOT|REMOVE_PHYSICAL);
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
static const test_function_t tests[] = {
|
|
test_path_exists,
|
|
test_path_existsglob,
|
|
test_path_changed,
|
|
test_path_modified,
|
|
test_path_unit,
|
|
test_path_directorynotempty,
|
|
test_path_makedirectory_directorymode,
|
|
NULL,
|
|
};
|
|
|
|
_cleanup_(rm_rf_physical_and_freep) char *runtime_dir = NULL;
|
|
const test_function_t *test = NULL;
|
|
Manager *m = NULL;
|
|
|
|
log_parse_environment();
|
|
log_open();
|
|
|
|
assert_se(set_unit_path(TEST_DATA_DIR("/test-path/")) >= 0);
|
|
assert_se(runtime_dir = setup_fake_runtime_dir());
|
|
|
|
for (test = tests; test && *test; test++) {
|
|
int r;
|
|
|
|
/* We create a clean environment for each test */
|
|
r = setup_test(&m);
|
|
if (r < 0)
|
|
return -r;
|
|
|
|
(*test)(m);
|
|
|
|
shutdown_test(m);
|
|
}
|
|
|
|
return 0;
|
|
}
|