]>
Commit | Line | Data |
---|---|---|
db9ecf05 | 1 | /* SPDX-License-Identifier: LGPL-2.1-or-later */ |
7d1e91d1 ZJS |
2 | |
3 | #include "path-lookup.h" | |
80276543 | 4 | #include "set.h" |
da33cba0 | 5 | #include "special.h" |
7d1e91d1 ZJS |
6 | #include "strv.h" |
7 | #include "tests.h" | |
8 | #include "unit-file.h" | |
9 | ||
10 | static void test_unit_validate_alias_symlink_and_warn(void) { | |
11 | log_info("/* %s */", __func__); | |
12 | ||
13 | assert_se(unit_validate_alias_symlink_and_warn("/path/a.service", "/other/b.service") == 0); | |
14 | assert_se(unit_validate_alias_symlink_and_warn("/path/a.service", "/other/b.socket") == -EXDEV); | |
15 | assert_se(unit_validate_alias_symlink_and_warn("/path/a.service", "/other/b.foobar") == -EXDEV); | |
16 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@.service", "/other/b@.service") == 0); | |
17 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@.service", "/other/b@.socket") == -EXDEV); | |
18 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@XXX.service", "/other/b@YYY.service") == -EXDEV); | |
19 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@XXX.service", "/other/b@YYY.socket") == -EXDEV); | |
20 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@.service", "/other/b@YYY.service") == -EXDEV); | |
21 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@XXX.service", "/other/b@XXX.service") == 0); | |
22 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@XXX.service", "/other/b@.service") == 0); | |
23 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@.service", "/other/b.service") == -EXDEV); | |
24 | assert_se(unit_validate_alias_symlink_and_warn("/path/a.service", "/other/b@.service") == -EXDEV); | |
25 | assert_se(unit_validate_alias_symlink_and_warn("/path/a@.slice", "/other/b.slice") == -EINVAL); | |
26 | assert_se(unit_validate_alias_symlink_and_warn("/path/a.slice", "/other/b.slice") == -EINVAL); | |
27 | } | |
28 | ||
80276543 | 29 | static void test_unit_file_build_name_map(char **ids) { |
e8630e69 ZJS |
30 | _cleanup_(lookup_paths_free) LookupPaths lp = {}; |
31 | _cleanup_hashmap_free_ Hashmap *unit_ids = NULL; | |
32 | _cleanup_hashmap_free_ Hashmap *unit_names = NULL; | |
e8630e69 ZJS |
33 | const char *k, *dst; |
34 | char **v; | |
91e0ee5f ZJS |
35 | usec_t mtime = 0; |
36 | int r; | |
e8630e69 ZJS |
37 | |
38 | assert_se(lookup_paths_init(&lp, UNIT_FILE_SYSTEM, 0, NULL) >= 0); | |
39 | ||
91e0ee5f | 40 | assert_se(unit_file_build_name_map(&lp, &mtime, &unit_ids, &unit_names, NULL) == 1); |
e8630e69 | 41 | |
90e74a66 | 42 | HASHMAP_FOREACH_KEY(dst, k, unit_ids) |
e8630e69 ZJS |
43 | log_info("ids: %s → %s", k, dst); |
44 | ||
90e74a66 | 45 | HASHMAP_FOREACH_KEY(v, k, unit_names) { |
e8630e69 ZJS |
46 | _cleanup_free_ char *j = strv_join(v, ", "); |
47 | log_info("aliases: %s ← %s", k, j); | |
48 | } | |
91e0ee5f ZJS |
49 | |
50 | char buf[FORMAT_TIMESTAMP_MAX]; | |
51 | log_debug("Last modification time: %s", format_timestamp(buf, sizeof buf, mtime)); | |
52 | ||
53 | r = unit_file_build_name_map(&lp, &mtime, &unit_ids, &unit_names, NULL); | |
54 | assert_se(IN_SET(r, 0, 1)); | |
55 | if (r == 0) | |
56 | log_debug("Cache rebuild skipped based on mtime."); | |
80276543 | 57 | |
80276543 ZJS |
58 | char **id; |
59 | STRV_FOREACH(id, ids) { | |
60 | const char *fragment, *name; | |
80276543 ZJS |
61 | _cleanup_set_free_free_ Set *names = NULL; |
62 | log_info("*** %s ***", *id); | |
63 | r = unit_file_find_fragment(unit_ids, | |
64 | unit_names, | |
65 | *id, | |
66 | &fragment, | |
67 | &names); | |
f21b863e | 68 | assert_se(r == 0); |
80276543 ZJS |
69 | log_info("fragment: %s", fragment); |
70 | log_info("names:"); | |
90e74a66 | 71 | SET_FOREACH(name, names) |
80276543 ZJS |
72 | log_info(" %s", name); |
73 | } | |
11e9347b DDM |
74 | |
75 | /* Make sure everything still works if we don't collect names. */ | |
76 | STRV_FOREACH(id, ids) { | |
77 | const char *fragment; | |
78 | log_info("*** %s ***", *id); | |
79 | r = unit_file_find_fragment(unit_ids, | |
80 | unit_names, | |
81 | *id, | |
82 | &fragment, | |
83 | NULL); | |
84 | assert_se(r == 0); | |
85 | log_info("fragment: %s", fragment); | |
86 | } | |
e8630e69 ZJS |
87 | } |
88 | ||
da33cba0 ZJS |
89 | static void test_runlevel_to_target(void) { |
90 | log_info("/* %s */", __func__); | |
91 | ||
92 | in_initrd_force(false); | |
93 | assert_se(streq_ptr(runlevel_to_target(NULL), NULL)); | |
94 | assert_se(streq_ptr(runlevel_to_target("unknown-runlevel"), NULL)); | |
95 | assert_se(streq_ptr(runlevel_to_target("rd.unknown-runlevel"), NULL)); | |
96 | assert_se(streq_ptr(runlevel_to_target("3"), SPECIAL_MULTI_USER_TARGET)); | |
97 | assert_se(streq_ptr(runlevel_to_target("rd.rescue"), NULL)); | |
98 | ||
99 | in_initrd_force(true); | |
100 | assert_se(streq_ptr(runlevel_to_target(NULL), NULL)); | |
101 | assert_se(streq_ptr(runlevel_to_target("unknown-runlevel"), NULL)); | |
102 | assert_se(streq_ptr(runlevel_to_target("rd.unknown-runlevel"), NULL)); | |
103 | assert_se(streq_ptr(runlevel_to_target("3"), NULL)); | |
104 | assert_se(streq_ptr(runlevel_to_target("rd.rescue"), SPECIAL_RESCUE_TARGET)); | |
105 | } | |
106 | ||
7d1e91d1 | 107 | int main(int argc, char **argv) { |
32464e18 | 108 | log_show_color(true); |
7d1e91d1 ZJS |
109 | test_setup_logging(LOG_DEBUG); |
110 | ||
111 | test_unit_validate_alias_symlink_and_warn(); | |
80276543 | 112 | test_unit_file_build_name_map(strv_skip(argv, 1)); |
da33cba0 | 113 | test_runlevel_to_target(); |
7d1e91d1 ZJS |
114 | |
115 | return 0; | |
116 | } |