1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2010 Lennart Poettering
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
29 static void test_strv_env_delete(void) {
30 _cleanup_strv_free_ char **a = NULL, **b = NULL, **c = NULL, **d = NULL;
32 a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
35 b = strv_new("PIEP", "FOO", NULL);
38 c = strv_new("SCHLUMPF", NULL);
41 d = strv_env_delete(a, 2, b, c);
44 assert_se(streq(d[0], "WALDO=WALDO"));
45 assert_se(streq(d[1], "WALDO="));
46 assert_se(strv_length(d) == 2);
49 static void test_strv_env_unset(void) {
50 _cleanup_strv_free_ char **l = NULL;
52 l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
55 assert_se(strv_env_unset(l, "SCHLUMPF") == l);
57 assert_se(streq(l[0], "PIEP"));
58 assert_se(streq(l[1], "NANANANA=YES"));
59 assert_se(strv_length(l) == 2);
62 static void test_strv_env_set(void) {
63 _cleanup_strv_free_ char **l = NULL, **r = NULL;
65 l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
68 r = strv_env_set(l, "WALDO=WALDO");
71 assert_se(streq(r[0], "PIEP"));
72 assert_se(streq(r[1], "SCHLUMPF=SMURFF"));
73 assert_se(streq(r[2], "NANANANA=YES"));
74 assert_se(streq(r[3], "WALDO=WALDO"));
75 assert_se(strv_length(r) == 4);
78 static void test_strv_env_merge(void) {
79 _cleanup_strv_free_ char **a = NULL, **b = NULL, **r = NULL;
81 a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
84 b = strv_new("FOO=KKK", "FOO=", "PIEP=", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
87 r = strv_env_merge(2, a, b);
89 assert_se(streq(r[0], "FOO="));
90 assert_se(streq(r[1], "WALDO="));
91 assert_se(streq(r[2], "PIEP"));
92 assert_se(streq(r[3], "SCHLUMPF=SMURFF"));
93 assert_se(streq(r[4], "PIEP="));
94 assert_se(streq(r[5], "NANANANA=YES"));
95 assert_se(strv_length(r) == 6);
97 assert_se(strv_env_clean(r) == r);
98 assert_se(streq(r[0], "FOO="));
99 assert_se(streq(r[1], "WALDO="));
100 assert_se(streq(r[2], "SCHLUMPF=SMURFF"));
101 assert_se(streq(r[3], "PIEP="));
102 assert_se(streq(r[4], "NANANANA=YES"));
103 assert_se(strv_length(r) == 5);
106 static void test_replace_env_arg(void) {
107 const char *env[] = {
112 const char *line[] = {
124 _cleanup_strv_free_ char **r = NULL;
126 r = replace_env_argv((char**) line, (char**) env);
128 assert_se(streq(r[0], "FOO$FOO"));
129 assert_se(streq(r[1], "FOO$FOOFOO"));
130 assert_se(streq(r[2], "FOOBAR BAR$FOO"));
131 assert_se(streq(r[3], "FOOBAR BAR"));
132 assert_se(streq(r[4], "BAR BAR"));
133 assert_se(streq(r[5], "BAR"));
134 assert_se(streq(r[6], "BAR"));
135 assert_se(streq(r[7], "BAR BARwaldo"));
136 assert_se(streq(r[8], "${FOO"));
137 assert_se(strv_length(r) == 9);
140 static void test_one_normalize(const char *input, const char *output) {
141 _cleanup_free_ char *t;
143 t = normalize_env_assignment(input);
145 assert_se(streq(t, output));
148 static void test_normalize_env_assignment(void) {
149 test_one_normalize("foo=bar", "foo=bar");
150 test_one_normalize("=bar", "=bar");
151 test_one_normalize("foo=", "foo=");
152 test_one_normalize("=", "=");
153 test_one_normalize("", "");
154 test_one_normalize("a=\"waldo\"", "a=waldo");
155 test_one_normalize("a=\"waldo", "a=\"waldo");
156 test_one_normalize("a=waldo\"", "a=waldo\"");
157 test_one_normalize("a=\'", "a='");
158 test_one_normalize("a=\'\'", "a=");
159 test_one_normalize(" xyz ", "xyz");
160 test_one_normalize(" xyz = bar ", "xyz=bar");
161 test_one_normalize(" xyz = 'bar ' ", "xyz=bar ");
162 test_one_normalize(" ' xyz' = 'bar ' ", "' xyz'=bar ");
165 static void test_env_clean(void) {
166 _cleanup_strv_free_ char **e;
168 e = strv_new("FOOBAR=WALDO",
186 assert_se(!strv_env_is_valid(e));
187 assert_se(strv_env_clean(e) == e);
188 assert_se(strv_env_is_valid(e));
190 assert_se(streq(e[0], "FOOBAR=WALDO"));
191 assert_se(streq(e[1], "X="));
192 assert_se(streq(e[2], "F=F"));
193 assert_se(streq(e[3], "abcd=äöüß"));
194 assert_se(streq(e[4], "another=final one"));
195 assert_se(e[5] == NULL);
198 static void test_env_name_is_valid(void) {
199 assert_se(env_name_is_valid("test"));
201 assert_se(!env_name_is_valid(NULL));
202 assert_se(!env_name_is_valid(""));
203 assert_se(!env_name_is_valid("5_starting_with_a_number_is_wrong"));
204 assert_se(!env_name_is_valid("#¤%&?_only_numbers_letters_and_underscore_allowed"));
207 int main(int argc, char *argv[]) {
208 test_strv_env_delete();
209 test_strv_env_unset();
211 test_strv_env_merge();
212 test_replace_env_arg();
213 test_normalize_env_assignment();
215 test_env_name_is_valid();