src/test/test-env-replace.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   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.
  12
  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.
  17
  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/>.
  20 ***/
  21
  22 #include <string.h>
  23
  24 #include "util.h"
  25 #include "strv.h"
  26 #include "env-util.h"
  27
  28 static void test_strv_env_delete(void) {
  29         _cleanup_strv_free_ char **a = NULL, **b = NULL, **c = NULL, **d = NULL;
  30
  31         a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
  32         assert_se(a);
  33
  34         b = strv_new("PIEP", "FOO", NULL);
  35         assert_se(b);
  36
  37         c = strv_new("SCHLUMPF", NULL);
  38         assert_se(c);
  39
  40         d = strv_env_delete(a, 2, b, c);
  41         assert_se(d);
  42
  43         assert_se(streq(d[0], "WALDO=WALDO"));
  44         assert_se(streq(d[1], "WALDO="));
  45         assert_se(strv_length(d) == 2);
  46 }
  47
  48 static void test_strv_env_unset(void) {
  49         _cleanup_strv_free_ char **l = NULL;
  50
  51         l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
  52         assert_se(l);
  53
  54         assert_se(strv_env_unset(l, "SCHLUMPF") == l);
  55
  56         assert_se(streq(l[0], "PIEP"));
  57         assert_se(streq(l[1], "NANANANA=YES"));
  58         assert_se(strv_length(l) == 2);
  59 }
  60
  61 static void test_strv_env_set(void) {
  62         _cleanup_strv_free_ char **l = NULL, **r = NULL;
  63
  64         l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
  65         assert_se(l);
  66
  67         r = strv_env_set(l, "WALDO=WALDO");
  68         assert_se(r);
  69
  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);
  75 }
  76
  77 static void test_strv_env_merge(void) {
  78         _cleanup_strv_free_ char **a = NULL, **b = NULL, **r = NULL;
  79
  80         a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
  81         assert_se(a);
  82
  83         b = strv_new("FOO=KKK", "FOO=", "PIEP=", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
  84         assert_se(b);
  85
  86         r = strv_env_merge(2, a, b);
  87         assert_se(r);
  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);
  95
  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);
 103 }
 104
 105 static void test_replace_env_arg(void) {
 106         const char *env[] = {
 107                 "FOO=BAR BAR",
 108                 "BAR=waldo",
 109                 NULL
 110         };
 111         const char *line[] = {
 112                 "FOO$FOO",
 113                 "FOO$FOOFOO",
 114                 "FOO${FOO}$FOO",
 115                 "FOO${FOO}",
 116                 "${FOO}",
 117                 "$FOO",
 118                 "$FOO$FOO",
 119                 "${FOO}${BAR}",
 120                 "${FOO",
 121                 NULL
 122         };
 123         _cleanup_strv_free_ char **r = NULL;
 124
 125         r = replace_env_argv((char**) line, (char**) env);
 126         assert_se(r);
 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);
 137 }
 138
 139 static void test_env_clean(void) {
 140         _cleanup_strv_free_ char **e;
 141
 142         e = strv_new("FOOBAR=WALDO",
 143                      "FOOBAR=WALDO",
 144                      "FOOBAR",
 145                      "F",
 146                      "X=",
 147                      "F=F",
 148                      "=",
 149                      "=F",
 150                      "",
 151                      "0000=000",
 152                      "äöüß=abcd",
 153                      "abcd=äöüß",
 154                      "xyz\n=xyz",
 155                      "xyz=xyz\n",
 156                      "another=one",
 157                      "another=final one",
 158                      NULL);
 159         assert_se(e);
 160         assert_se(!strv_env_is_valid(e));
 161         assert_se(strv_env_clean(e) == e);
 162         assert_se(strv_env_is_valid(e));
 163
 164         assert_se(streq(e[0], "FOOBAR=WALDO"));
 165         assert_se(streq(e[1], "X="));
 166         assert_se(streq(e[2], "F=F"));
 167         assert_se(streq(e[3], "abcd=äöüß"));
 168         assert_se(streq(e[4], "another=final one"));
 169         assert_se(e[5] == NULL);
 170 }
 171
 172 static void test_env_name_is_valid(void) {
 173         assert_se(env_name_is_valid("test"));
 174
 175         assert_se(!env_name_is_valid(NULL));
 176         assert_se(!env_name_is_valid(""));
 177         assert_se(!env_name_is_valid("5_starting_with_a_number_is_wrong"));
 178         assert_se(!env_name_is_valid("#¤%&?_only_numbers_letters_and_underscore_allowed"));
 179 }
 180
 181 int main(int argc, char *argv[]) {
 182         test_strv_env_delete();
 183         test_strv_env_unset();
 184         test_strv_env_set();
 185         test_strv_env_merge();
 186         test_replace_env_arg();
 187         test_env_clean();
 188         test_env_name_is_valid();
 189
 190         return 0;
 191 }