From 16b5238eb354b2d8faa90cb5c9622963a5d6a1eb Mon Sep 17 00:00:00 2001
From: Karel Zak
Date: Wed, 26 Oct 2016 13:50:10 +0200
Subject: libfdisk: add trivial script test

Signed-off-by: Karel Zak <kzak@redhat.com>
---
 libfdisk/src/script.c | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'libfdisk')

diff --git a/libfdisk/src/script.c b/libfdisk/src/script.c
index 54621ae28..e90c18ab0 100644
--- 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 }
 	};
 
-- 
cgit v1.2.3-55-g7522