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/>.
28 static void test_strv_env_delete(void) {
29 _cleanup_strv_free_ char **a = NULL, **b = NULL, **c = NULL, **d = NULL;
31 a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
34 b = strv_new("PIEP", "FOO", NULL);
37 c = strv_new("SCHLUMPF", NULL);
40 d = strv_env_delete(a, 2, b, c);
43 assert_se(streq(d[0], "WALDO=WALDO"));
44 assert_se(streq(d[1], "WALDO="));
45 assert_se(strv_length(d) == 2);
48 static void test_strv_env_unset(void) {
49 _cleanup_strv_free_ char **l = NULL;
51 l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
54 assert_se(strv_env_unset(l, "SCHLUMPF") == l);
56 assert_se(streq(l[0], "PIEP"));
57 assert_se(streq(l[1], "NANANANA=YES"));
58 assert_se(strv_length(l) == 2);
61 static void test_strv_env_set(void) {
62 _cleanup_strv_free_ char **l = NULL, **r = NULL;
64 l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
67 r = strv_env_set(l, "WALDO=WALDO");
70 assert_se(streq(r[0], "PIEP"));
71 assert_se(streq(r[1], "SCHLUMPF=SMURFF"));
72 assert_se(streq(r[2], "NANANANA=YES"));
73 assert_se(streq(r[3], "WALDO=WALDO"));
74 assert_se(strv_length(r) == 4);
77 static void test_strv_env_merge(void) {
78 _cleanup_strv_free_ char **a = NULL, **b = NULL, **r = NULL;
80 a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
83 b = strv_new("FOO=KKK", "FOO=", "PIEP=", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
86 r = strv_env_merge(2, a, b);
88 assert_se(streq(r[0], "FOO="));
89 assert_se(streq(r[1], "WALDO="));
90 assert_se(streq(r[2], "PIEP"));
91 assert_se(streq(r[3], "SCHLUMPF=SMURFF"));
92 assert_se(streq(r[4], "PIEP="));
93 assert_se(streq(r[5], "NANANANA=YES"));
94 assert_se(strv_length(r) == 6);
96 assert_se(strv_env_clean(r) == r);
97 assert_se(streq(r[0], "FOO="));
98 assert_se(streq(r[1], "WALDO="));
99 assert_se(streq(r[2], "SCHLUMPF=SMURFF"));
100 assert_se(streq(r[3], "PIEP="));
101 assert_se(streq(r[4], "NANANANA=YES"));
102 assert_se(strv_length(r) == 5);
105 static void test_replace_env_arg(void) {
106 const char *env[] = {
111 const char *line[] = {
123 _cleanup_strv_free_ char **r = NULL;
125 r = replace_env_argv((char**) line, (char**) env);
127 assert_se(streq(r[0], "FOO$FOO"));
128 assert_se(streq(r[1], "FOO$FOOFOO"));
129 assert_se(streq(r[2], "FOOBAR BAR$FOO"));
130 assert_se(streq(r[3], "FOOBAR BAR"));
131 assert_se(streq(r[4], "BAR BAR"));
132 assert_se(streq(r[5], "BAR"));
133 assert_se(streq(r[6], "BAR"));
134 assert_se(streq(r[7], "BAR BARwaldo"));
135 assert_se(streq(r[8], "${FOO"));
136 assert_se(strv_length(r) == 9);
139 static void test_one_normalize(const char *input, const char *output) {
140 _cleanup_free_ char *t;
142 t = normalize_env_assignment(input);
144 assert_se(streq(t, output));
147 static void test_normalize_env_assignment(void) {
148 test_one_normalize("foo=bar", "foo=bar");
149 test_one_normalize("=bar", "=bar");
150 test_one_normalize("foo=", "foo=");
151 test_one_normalize("=", "=");
152 test_one_normalize("", "");
153 test_one_normalize("a=\"waldo\"", "a=waldo");
154 test_one_normalize("a=\"waldo", "a=\"waldo");
155 test_one_normalize("a=waldo\"", "a=waldo\"");
156 test_one_normalize("a=\'", "a='");
157 test_one_normalize("a=\'\'", "a=");
158 test_one_normalize(" xyz ", "xyz");
159 test_one_normalize(" xyz = bar ", "xyz=bar");
160 test_one_normalize(" xyz = 'bar ' ", "xyz=bar ");
161 test_one_normalize(" ' xyz' = 'bar ' ", "' xyz'=bar ");
164 static void test_env_clean(void) {
165 _cleanup_strv_free_ char **e;
167 e = strv_new("FOOBAR=WALDO",
185 assert_se(!strv_env_is_valid(e));
186 assert_se(strv_env_clean(e) == e);
187 assert_se(strv_env_is_valid(e));
189 assert_se(streq(e[0], "FOOBAR=WALDO"));
190 assert_se(streq(e[1], "X="));
191 assert_se(streq(e[2], "F=F"));
192 assert_se(streq(e[3], "abcd=äöüß"));
193 assert_se(streq(e[4], "another=final one"));
194 assert_se(e[5] == NULL);
197 static void test_env_name_is_valid(void) {
198 assert_se(env_name_is_valid("test"));
200 assert_se(!env_name_is_valid(NULL));
201 assert_se(!env_name_is_valid(""));
202 assert_se(!env_name_is_valid("5_starting_with_a_number_is_wrong"));
203 assert_se(!env_name_is_valid("#¤%&?_only_numbers_letters_and_underscore_allowed"));
206 int main(int argc, char *argv[]) {
207 test_strv_env_delete();
208 test_strv_env_unset();
210 test_strv_env_merge();
211 test_replace_env_arg();
212 test_normalize_env_assignment();
214 test_env_name_is_valid();