tests: add tests for encode functions from lib/mbsalign.c

Signed-off-by: Vaclav Dolezal <vdolezal@redhat.com>

2 files changed, 55 insertions, 0 deletions

diff --git a/tests/helpers/Makemodule.am b/tests/helpers/Makemodule.am
index 2f2611b67..ab0b3cea9 100644
--- a/tests/helpers/Makemodule.am
+++ b/tests/helpers/Makemodule.am
@@ -1,3 +1,6 @@
+check_PROGRAMS += test_mbsencode
+test_mbsencode_SOURCES = tests/helpers/test_mbsencode.c
+test_mbsencode_LDADD = $(LDADD) libcommon.la
 
 check_PROGRAMS += test_byteswap
 test_byteswap_SOURCES = tests/helpers/test_byteswap.c
diff --git a/tests/helpers/test_mbsencode.c b/tests/helpers/test_mbsencode.c
new file mode 100644
index 000000000..a8f93836c
--- /dev/null
+++ b/tests/helpers/test_mbsencode.c
@@ -0,0 +1,52 @@
+/*
+ * Copyright (C) 2018 Vaclav Dolezal <vdolezal@redhat.com>
+ *
+ * This file is part of util-linux.
+ *
+ * This file is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This file is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <locale.h>
+
+#include "mbsalign.h"
+
+int main(int argc, char **argv)
+{
+	int i = 1;
+	char *(*encode_fn)(const char *, size_t *) = mbs_safe_encode;
+
+	setlocale(LC_ALL, "");
+
+	if (i < argc) {
+		if (!strcmp(argv[i], "--safe")) {
+			i++;
+			encode_fn = mbs_safe_encode;
+		} else if (!strcmp(argv[i], "--invalid")) {
+			i++;
+			encode_fn = mbs_invalid_encode;
+		} else if (!strcmp(argv[i], "--")) {
+			i++;
+		}
+	}
+
+	for (; i < argc; i++) {
+		size_t width;
+		char *res;
+		res = encode_fn(argv[i], &width);
+		printf("%zi %s\n", width, res);
+		free(res);
+	}
+
+	return 0;
+}