src/test/test-specifier.c

   1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
   2
   3 #include "alloc-util.h"
   4 #include "log.h"
   5 #include "specifier.h"
   6 #include "stdio-util.h"
   7 #include "string-util.h"
   8 #include "strv.h"
   9 #include "tests.h"
  10
  11 static void test_specifier_escape_one(const char *a, const char *b) {
  12         _cleanup_free_ char *x = NULL;
  13
  14         x = specifier_escape(a);
  15         assert_se(streq_ptr(x, b));
  16 }
  17
  18 static void test_specifier_escape(void) {
  19         log_info("/* %s */", __func__);
  20
  21         test_specifier_escape_one(NULL, NULL);
  22         test_specifier_escape_one("", "");
  23         test_specifier_escape_one("%", "%%");
  24         test_specifier_escape_one("foo bar", "foo bar");
  25         test_specifier_escape_one("foo%bar", "foo%%bar");
  26         test_specifier_escape_one("%%%%%", "%%%%%%%%%%");
  27 }
  28
  29 static void test_specifier_escape_strv_one(char **a, char **b) {
  30         _cleanup_strv_free_ char **x = NULL;
  31
  32         assert_se(specifier_escape_strv(a, &x) >= 0);
  33         assert_se(strv_equal(x, b));
  34 }
  35
  36 static void test_specifier_escape_strv(void) {
  37         log_info("/* %s */", __func__);
  38
  39         test_specifier_escape_strv_one(NULL, NULL);
  40         test_specifier_escape_strv_one(STRV_MAKE(NULL), STRV_MAKE(NULL));
  41         test_specifier_escape_strv_one(STRV_MAKE(""), STRV_MAKE(""));
  42         test_specifier_escape_strv_one(STRV_MAKE("foo"), STRV_MAKE("foo"));
  43         test_specifier_escape_strv_one(STRV_MAKE("%"), STRV_MAKE("%%"));
  44         test_specifier_escape_strv_one(STRV_MAKE("foo", "%", "foo%", "%foo", "foo%foo", "quux", "%%%"),
  45                                        STRV_MAKE("foo", "%%", "foo%%", "%%foo", "foo%%foo", "quux", "%%%%%%"));
  46 }
  47
  48 /* Any specifier functions which don't need an argument. */
  49 static const Specifier specifier_table[] = {
  50         COMMON_SYSTEM_SPECIFIERS,
  51
  52         COMMON_CREDS_SPECIFIERS,
  53         { 'h', specifier_user_home,       NULL },
  54
  55         COMMON_TMP_SPECIFIERS,
  56         {}
  57 };
  58
  59 static void test_specifier_printf(void) {
  60         static const Specifier table[] = {
  61                 { 'X', specifier_string,         (char*) "AAAA" },
  62                 { 'Y', specifier_string,         (char*) "BBBB" },
  63                 COMMON_SYSTEM_SPECIFIERS,
  64                 {}
  65         };
  66
  67         _cleanup_free_ char *w = NULL;
  68         int r;
  69
  70         log_info("/* %s */", __func__);
  71
  72         r = specifier_printf("xxx a=%X b=%Y yyy", table, NULL, &w);
  73         assert_se(r >= 0);
  74         assert_se(w);
  75
  76         puts(w);
  77         assert_se(streq(w, "xxx a=AAAA b=BBBB yyy"));
  78
  79         free(w);
  80         r = specifier_printf("machine=%m, boot=%b, host=%H, version=%v, arch=%a", table, NULL, &w);
  81         assert_se(r >= 0);
  82         assert_se(w);
  83         puts(w);
  84
  85         w = mfree(w);
  86         specifier_printf("os=%o, os-version=%w, build=%B, variant=%W", table, NULL, &w);
  87         if (w)
  88                 puts(w);
  89 }
  90
  91 static void test_specifiers(void) {
  92         log_info("/* %s */", __func__);
  93
  94         for (const Specifier *s = specifier_table; s->specifier; s++) {
  95                 char spec[3];
  96                 _cleanup_free_ char *resolved = NULL;
  97
  98                 xsprintf(spec, "%%%c", s->specifier);
  99
 100                 assert_se(specifier_printf(spec, specifier_table, NULL, &resolved) >= 0);
 101
 102                 log_info("%%%c → %s", s->specifier, resolved);
 103         }
 104 }
 105
 106 int main(int argc, char *argv[]) {
 107         test_setup_logging(LOG_DEBUG);
 108
 109         test_specifier_escape();
 110         test_specifier_escape_strv();
 111         test_specifier_printf();
 112         test_specifiers();
 113
 114         return 0;
 115 }