src/test/test-strip-tab-ansi.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <stdio.h>
   4
   5 #include "alloc-util.h"
   6 #include "pretty-print.h"
   7 #include "string-util.h"
   8 #include "terminal-util.h"
   9 #include "util.h"
  10
  11 int main(int argc, char *argv[]) {
  12         _cleanup_free_ char *urlified = NULL, *q = NULL, *qq = NULL;
  13         char *p, *z;
  14
  15         assert_se(p = strdup("\tFoobar\tbar\twaldo\t"));
  16         assert_se(strip_tab_ansi(&p, NULL, NULL));
  17         fprintf(stdout, "<%s>\n", p);
  18         assert_se(streq(p, "        Foobar        bar        waldo        "));
  19         free(p);
  20
  21         assert_se(p = strdup(ANSI_HIGHLIGHT "Hello" ANSI_NORMAL ANSI_HIGHLIGHT_RED " world!" ANSI_NORMAL));
  22         assert_se(strip_tab_ansi(&p, NULL, NULL));
  23         fprintf(stdout, "<%s>\n", p);
  24         assert_se(streq(p, "Hello world!"));
  25         free(p);
  26
  27         assert_se(p = strdup("\x1B[\x1B[\t\x1B[" ANSI_HIGHLIGHT "\x1B[" "Hello" ANSI_NORMAL ANSI_HIGHLIGHT_RED " world!" ANSI_NORMAL));
  28         assert_se(strip_tab_ansi(&p, NULL, NULL));
  29         assert_se(streq(p, "\x1B[\x1B[        \x1B[\x1B[Hello world!"));
  30         free(p);
  31
  32         assert_se(p = strdup("\x1B[waldo"));
  33         assert_se(strip_tab_ansi(&p, NULL, NULL));
  34         assert_se(streq(p, "\x1B[waldo"));
  35         free(p);
  36
  37         assert_se(p = strdup("\r\rwaldo"));
  38         assert_se(strip_tab_ansi(&p, NULL, NULL));
  39         assert_se(streq(p, "\r\rwaldo"));
  40         free(p);
  41
  42         assert_se(p = strdup("waldo\r\r"));
  43         assert_se(strip_tab_ansi(&p, NULL, NULL));
  44         assert_se(streq(p, "waldo"));
  45         free(p);
  46
  47         assert_se(p = strdup("waldo\r\r\n\r\n"));
  48         assert_se(strip_tab_ansi(&p, NULL, NULL));
  49         assert_se(streq(p, "waldo\n\n"));
  50         free(p);
  51
  52         assert_se(terminal_urlify_path("/etc/fstab", "i am a fabulous link", &urlified) >= 0);
  53         assert_se(p = strjoin("something ", urlified, " something-else"));
  54         assert_se(q = strdup(p));
  55         printf("<%s>\n", p);
  56         assert_se(strip_tab_ansi(&p, NULL, NULL));
  57         printf("<%s>\n", p);
  58         assert_se(streq(p, "something i am a fabulous link something-else"));
  59         p = mfree(p);
  60
  61         /* Truncate the formatted string in the middle of an ANSI sequence (in which case we shouldn't touch the
  62          * incomplete sequence) */
  63         z = strstr(q, "fstab");
  64         if (z) {
  65                 *z = 0;
  66                 assert_se(qq = strdup(q));
  67                 assert_se(strip_tab_ansi(&q, NULL, NULL));
  68                 assert_se(streq(q, qq));
  69         }
  70
  71         return 0;
  72 }