X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Ftest%2Ftest-unit-name.c;h=fff3e16ed816b82344433112c8dd534139ac8e0c;hb=e5b04c8de83aa17e324bae95f18a822931fb8078;hp=5207781f99128ce07994c152a62e3d7336b73842;hpb=3251c0d2392062b2b8354793a2037d1161f824ee;p=elogind.git diff --git a/src/test/test-unit-name.c b/src/test/test-unit-name.c index 5207781f9..fff3e16ed 100644 --- a/src/test/test-unit-name.c +++ b/src/test/test-unit-name.c @@ -4,6 +4,7 @@ This file is part of systemd. Copyright 2012 Lennart Poettering + Copyright 2013 Zbigniew Jędrzejewski-Szmek 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 @@ -22,15 +23,23 @@ #include #include #include +#include +#include +#include "manager.h" +#include "unit.h" #include "unit-name.h" +#include "unit-printf.h" +#include "install.h" +#include "specifier.h" #include "util.h" +#include "macro.h" +#include "test-helper.h" -int main(int argc, char* argv[]) { - +static void test_replacements(void) { #define expect(pattern, repl, expected) \ { \ - char _cleanup_free_ *t = \ + _cleanup_free_ char *t = \ unit_name_replace_instance(pattern, repl); \ puts(t); \ assert(streq(t, expected)); \ @@ -49,12 +58,12 @@ int main(int argc, char* argv[]) { #undef expect #define expect(path, suffix, expected) \ { \ - char _cleanup_free_ *k, *t = \ + _cleanup_free_ char *k, *t = \ unit_name_from_path(path, suffix); \ puts(t); \ k = unit_name_to_path(t); \ puts(k); \ - assert(streq(k, expected ? expected : path)); \ + assert(streq(k, expected ? expected : path)); \ } expect("/waldo", ".mount", NULL); @@ -65,12 +74,12 @@ int main(int argc, char* argv[]) { puts("-------------------------------------------------"); #undef expect -#define expect(pattern, path, suffix, expected) \ - { \ - char _cleanup_free_ *t = \ +#define expect(pattern, path, suffix, expected) \ + { \ + _cleanup_free_ char *t = \ unit_name_from_path_instance(pattern, path, suffix); \ - puts(t); \ - assert(streq(t, expected)); \ + puts(t); \ + assert(streq(t, expected)); \ } expect("waldo", "/waldo", ".mount", "waldo@waldo.mount"); @@ -80,13 +89,13 @@ int main(int argc, char* argv[]) { puts("-------------------------------------------------"); #undef expect -#define expect(pattern) \ - { \ - char _cleanup_free_ *k, *t; \ - assert_se(t = unit_name_mangle(pattern)); \ - assert_se(k = unit_name_mangle(t)); \ - puts(t); \ - assert_se(streq(t, k)); \ +#define expect(pattern) \ + { \ + _cleanup_free_ char *k, *t; \ + assert_se(t = unit_name_mangle(pattern, MANGLE_NOGLOB)); \ + assert_se(k = unit_name_mangle(t, MANGLE_NOGLOB)); \ + puts(t); \ + assert_se(streq(t, k)); \ } expect("/home"); @@ -97,5 +106,96 @@ int main(int argc, char* argv[]) { expect("_____##@;;;,,,##----.....service"); expect("xxx@@@@/////\\\\\\\\\\yyy.service"); +#undef expect +} + +static int test_unit_printf(void) { + Manager *m = NULL; + Unit *u, *u2; + int r; + + _cleanup_free_ char *mid, *bid, *host, *root_uid; + struct passwd *root; + + assert_se(specifier_machine_id('m', NULL, NULL, &mid) >= 0 && mid); + assert_se(specifier_boot_id('b', NULL, NULL, &bid) >= 0 && bid); + assert_se((host = gethostname_malloc())); + + assert_se((root = getpwnam("root"))); + assert_se(asprintf(&root_uid, "%d", (int) root->pw_uid) > 0); + + r = manager_new(SYSTEMD_USER, &m); + if (r == -EPERM || r == -EACCES || r == -EADDRINUSE) { + puts("manager_new: Permission denied. Skipping test."); + return EXIT_TEST_SKIP; + } + assert(r == 0); + +#define expect(unit, pattern, expected) \ + { \ + char *e; \ + _cleanup_free_ char *t; \ + assert_se(unit_full_printf(unit, pattern, &t) >= 0); \ + printf("result: %s\nexpect: %s\n", t, expected); \ + if ((e = endswith(expected, "*"))) \ + assert(strncmp(t, e, e-expected)); \ + else \ + assert(streq(t, expected)); \ + } + + assert_se(setenv("USER", "root", 1) == 0); + assert_se(setenv("HOME", "/root", 1) == 0); + + assert_se(u = unit_new(m, sizeof(Service))); + assert_se(unit_add_name(u, "blah.service") == 0); + assert_se(unit_add_name(u, "blah.service") == 0); + + /* general tests */ + expect(u, "%%", "%"); + expect(u, "%%s", "%s"); + expect(u, "%", ""); // REALLY? + + /* normal unit */ + expect(u, "%n", "blah.service"); + expect(u, "%N", "blah"); + expect(u, "%p", "blah"); + expect(u, "%P", "blah"); + expect(u, "%i", ""); + expect(u, "%u", root->pw_name); + expect(u, "%U", root_uid); + expect(u, "%h", root->pw_dir); + expect(u, "%m", mid); + expect(u, "%b", bid); + expect(u, "%H", host); + expect(u, "%t", "/run/user/*"); + + /* templated */ + assert_se(u2 = unit_new(m, sizeof(Service))); + assert_se(unit_add_name(u2, "blah@foo-foo.service") == 0); + assert_se(unit_add_name(u2, "blah@foo-foo.service") == 0); + + expect(u2, "%n", "blah@foo-foo.service"); + expect(u2, "%N", "blah@foo-foo"); + expect(u2, "%p", "blah"); + expect(u2, "%P", "blah"); + expect(u2, "%i", "foo-foo"); + expect(u2, "%I", "foo/foo"); + expect(u2, "%u", root->pw_name); + expect(u2, "%U", root_uid); + expect(u2, "%h", root->pw_dir); + expect(u2, "%m", mid); + expect(u2, "%b", bid); + expect(u2, "%H", host); + expect(u2, "%t", "/run/user/*"); + + manager_free(m); + return 0; } + +int main(int argc, char* argv[]) { + int rc = 0; + test_replacements(); + TEST_REQ_RUNNING_SYSTEMD(rc = test_unit_printf()); + return rc; +}