X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Ftest%2Ftest-unit-name.c;h=256e820c224ee85c5c5f688a5b3142515800885f;hp=392e358ac2f4efc7cccfda0aee17a90ea6ff55bf;hb=ea7061e4d431c3e03212243f65148f2e985d80b7;hpb=35eb6b124ebdf82bd77aad6e44962a9a039c4d33 diff --git a/src/test/test-unit-name.c b/src/test/test-unit-name.c index 392e358ac..256e820c2 100644 --- a/src/test/test-unit-name.c +++ b/src/test/test-unit-name.c @@ -4,6 +4,8 @@ This file is part of systemd. Copyright 2012 Lennart Poettering + Copyright 2013 Zbigniew Jędrzejewski-Szmek + 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 @@ -22,112 +24,306 @@ #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[]) { - char *t, *k; - - t = unit_name_from_path("/waldo", ".mount"); - puts(t); - k = unit_name_to_path(t); - puts(k); - free(k); - free(t); - - t = unit_name_from_path("/waldo/quuix", ".mount"); - puts(t); - k = unit_name_to_path(t); - puts(k); - free(k); - free(t); - - t = unit_name_from_path("/waldo/quuix/", ".mount"); - puts(t); - k = unit_name_to_path(t); - puts(k); - free(k); - free(t); - - t = unit_name_from_path("/", ".mount"); - puts(t); - k = unit_name_to_path(t); - puts(k); - free(k); - free(t); - - t = unit_name_from_path("///", ".mount"); - puts(t); - k = unit_name_to_path(t); - puts(k); - free(k); - free(t); - - t = unit_name_from_path_instance("waldo", "/waldo", ".mount"); - puts(t); - free(t); - - t = unit_name_from_path_instance("waldo", "/waldo////quuix////", ".mount"); - puts(t); - free(t); - - t = unit_name_from_path_instance("waldo", "/", ".mount"); - puts(t); - free(t); - - t = unit_name_from_path_instance("wa--ldo", "/--", ".mount"); - puts(t); - free(t); - - assert_se(t = unit_name_mangle("/home")); - assert_se(k = unit_name_mangle(t)); - puts(t); - assert_se(streq(t, k)); - free(t); - free(k); - - assert_se(t = unit_name_mangle("/dev/sda")); - assert_se(k = unit_name_mangle(t)); - puts(t); - assert_se(streq(t, k)); - free(t); - free(k); - - assert_se(t = unit_name_mangle("üxknürz.service")); - assert_se(k = unit_name_mangle(t)); - puts(t); - assert_se(streq(t, k)); - free(t); - free(k); - - assert_se(t = unit_name_mangle("foobar-meh...waldi.service")); - assert_se(k = unit_name_mangle(t)); - puts(t); - assert_se(streq(t, k)); - free(t); - free(k); - - assert_se(t = unit_name_mangle("_____####----.....service")); - assert_se(k = unit_name_mangle(t)); - puts(t); - assert_se(streq(t, k)); - free(t); - free(k); - - assert_se(t = unit_name_mangle("_____##@;;;,,,##----.....service")); - assert_se(k = unit_name_mangle(t)); - puts(t); - assert_se(streq(t, k)); - free(t); - free(k); - - assert_se(t = unit_name_mangle("xxx@@@@/////\\\\\\\\\\yyy.service")); - assert_se(k = unit_name_mangle(t)); - puts(t); - assert_se(streq(t, k)); - free(t); - free(k); +static void test_replacements(void) { +#define expect(pattern, repl, expected) \ + { \ + _cleanup_free_ char *t = \ + unit_name_replace_instance(pattern, repl); \ + puts(t); \ + assert_se(streq(t, expected)); \ + } + + expect("foo@.service", "waldo", "foo@waldo.service"); + expect("foo@xyz.service", "waldo", "foo@waldo.service"); + expect("xyz", "waldo", "xyz"); + expect("", "waldo", ""); + expect("foo.service", "waldo", "foo.service"); + expect(".service", "waldo", ".service"); + expect("foo@", "waldo", "foo@waldo"); + expect("@bar", "waldo", "@waldo"); + + puts("-------------------------------------------------"); +#undef expect +#define expect(path, suffix, expected) \ + { \ + _cleanup_free_ char *k, *t = \ + unit_name_from_path(path, suffix); \ + puts(t); \ + k = unit_name_to_path(t); \ + puts(k); \ + assert_se(streq(k, expected ? expected : path)); \ + } + + expect("/waldo", ".mount", NULL); + expect("/waldo/quuix", ".mount", NULL); + expect("/waldo/quuix/", ".mount", "/waldo/quuix"); + expect("/", ".mount", NULL); + expect("///", ".mount", "/"); + + puts("-------------------------------------------------"); +#undef expect +#define expect(pattern, path, suffix, expected) \ + { \ + _cleanup_free_ char *t = \ + unit_name_from_path_instance(pattern, path, suffix); \ + puts(t); \ + assert_se(streq(t, expected)); \ + } + + expect("waldo", "/waldo", ".mount", "waldo@waldo.mount"); + expect("waldo", "/waldo////quuix////", ".mount", "waldo@waldo-quuix.mount"); + expect("waldo", "/", ".mount", "waldo@-.mount"); + expect("wa--ldo", "/--", ".mount", "wa--ldo@\\x2d\\x2d.mount"); + + puts("-------------------------------------------------"); +#undef expect +#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"); + expect("/dev/sda"); + expect("üxknürz.service"); + expect("foobar-meh...waldi.service"); + expect("_____####----.....service"); + 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, true, &m); + if (r == -EPERM || r == -EACCES || r == -EADDRINUSE) { + puts("manager_new: Permission denied. Skipping test."); + return EXIT_TEST_SKIP; + } + assert_se(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_se(strncmp(t, e, e-expected)); \ + else \ + assert_se(streq(t, expected)); \ + } + + assert_se(setenv("USER", "root", 1) == 0); + assert_se(setenv("HOME", "/root", 1) == 0); + assert_se(setenv("XDG_RUNTIME_DIR", "/run/user/1/", 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; } + +static void test_unit_instance_is_valid(void) { + assert_se(unit_instance_is_valid("fooBar")); + assert_se(unit_instance_is_valid("foo-bar")); + assert_se(unit_instance_is_valid("foo.stUff")); + assert_se(unit_instance_is_valid("fOo123.stuff")); + assert_se(unit_instance_is_valid("@f_oo123.Stuff")); + + assert_se(!unit_instance_is_valid("$¢£")); + assert_se(!unit_instance_is_valid("")); + assert_se(!unit_instance_is_valid("foo bar")); + assert_se(!unit_instance_is_valid("foo/bar")); +} + +static void test_unit_prefix_is_valid(void) { + assert_se(unit_prefix_is_valid("fooBar")); + assert_se(unit_prefix_is_valid("foo-bar")); + assert_se(unit_prefix_is_valid("foo.stUff")); + assert_se(unit_prefix_is_valid("fOo123.stuff")); + assert_se(unit_prefix_is_valid("foo123.Stuff")); + + assert_se(!unit_prefix_is_valid("$¢£")); + assert_se(!unit_prefix_is_valid("")); + assert_se(!unit_prefix_is_valid("foo bar")); + assert_se(!unit_prefix_is_valid("foo/bar")); + assert_se(!unit_prefix_is_valid("@foo-bar")); +} + +static void test_unit_name_change_suffix(void) { + char *r; + + r = unit_name_change_suffix("foo.bar", ".service"); + assert_se(r); + assert_se(streq(r, "foo.service")); + free(r); + + r = unit_name_change_suffix("foo@stuff.bar", ".boo"); + assert_se(r); + assert_se(streq(r, "foo@stuff.boo")); + free(r); +} + +static void test_unit_name_build(void) { + char *r; + + r = unit_name_build("foo", "bar", ".service"); + assert_se(r); + assert_se(streq(r, "foo@bar.service")); + free(r); + + r = unit_name_build("fo0-stUff_b", "bar", ".mount"); + assert_se(r); + assert_se(streq(r, "fo0-stUff_b@bar.mount")); + free(r); + + r = unit_name_build("foo", NULL, ".service"); + assert_se(r); + assert_se(streq(r, "foo.service")); + free(r); +} + +static void test_unit_name_is_instance(void) { + assert_se(unit_name_is_instance("a@b.service")); + assert_se(unit_name_is_instance("a-c_c01Aj@b05Dii_-oioi.service")); + + assert_se(!unit_name_is_instance("a.service")); + assert_se(!unit_name_is_instance("junk")); + assert_se(!unit_name_is_instance("")); +} + +static void test_build_subslice(void) { + char *a; + char *b; + + assert_se(build_subslice("-.slice", "foo", &a) >= 0); + assert_se(build_subslice(a, "bar", &b) >= 0); + free(a); + assert_se(build_subslice(b, "barfoo", &a) >= 0); + free(b); + assert_se(build_subslice(a, "foobar", &b) >= 0); + free(a); + assert_se(streq(b, "foo-bar-barfoo-foobar.slice")); + free(b); + + assert_se(build_subslice("foo.service", "bar", &a) < 0); + assert_se(build_subslice("foo", "bar", &a) < 0); +} + +static void test_unit_name_to_instance(void) { + char *instance; + int r; + + r = unit_name_to_instance("foo@bar.service", &instance); + assert_se(r >= 0); + assert_se(streq(instance, "bar")); + free(instance); + + r = unit_name_to_instance("fo0-stUff_b@b.e", &instance); + assert_se(r >= 0); + assert_se(streq(instance, "b")); + free(instance); + + r = unit_name_to_instance("foo.bar", &instance); + assert_se(r >= 0); + assert_se(!instance); + + r = unit_name_to_instance("fooj@unk", &instance); + assert_se(r < 0); +} + +static void test_unit_name_escape(void) { + _cleanup_free_ char *r; + + r = unit_name_escape("ab+-c.a/bc@foo.service"); + assert_se(r); + assert_se(streq(r, "ab\\x2b\\x2dc.a-bc\\x40foo.service")); +} + +int main(int argc, char* argv[]) { + int rc = 0; + test_replacements(); + TEST_REQ_RUNNING_SYSTEMD(rc = test_unit_printf()); + test_unit_instance_is_valid(); + test_unit_prefix_is_valid(); + test_unit_name_change_suffix(); + test_unit_name_build(); + test_unit_name_is_instance(); + test_build_subslice(); + test_unit_name_to_instance(); + test_unit_name_escape(); + + return rc; +}