libfdisk: add trivial script test

Signed-off-by: Karel Zak <kzak@redhat.com>

diff --git a/libfdisk/src/script.c b/libfdisk/src/script.c
index 54621ae2895cac3e4b8e4fec54cad76f8fed2b16..e90c18ab071e60951aa666afa2fba7755f992a34 100644 (file)
--- a/libfdisk/src/script.c
+++ b/libfdisk/src/script.c
@@ -1546,6 +1546,24 @@ done:
        return 0;
 }
 
+static int test_tokens(struct fdisk_test *ts, int argc, char *argv[])
+{
+       char *p, *str = argc == 2 ? strdup(argv[1]) : NULL;
+       int i;
+
+       for (i = 1, p = str; p && *p; i++) {
+               char *tk = next_token(&p);
+
+               if (!tk)
+                       break;
+
+               printf("#%d: '%s'\n", i, tk);
+       }
+
+       free(str);
+       return 0;
+}
+
 int main(int argc, char *argv[])
 {
        struct fdisk_test tss[] = {
@@ -1553,6 +1571,7 @@ int main(int argc, char *argv[])
        { "--read",    test_read,    "<file>              read PT script from file" },
        { "--apply",   test_apply,   "<device> <file>     try apply script from file to device" },
        { "--stdin",   test_stdin,   "                    read input like sfdisk" },
+       { "--tokens",  test_tokens,  "<string>            parse string" },
        { NULL }
        };