1 files changed, 84 insertions, 0 deletions
diff --git a/tests/libutil/str_table.c b/tests/libutil/str_table.c
new file mode 100644
index 0000000..83526af
--- /dev/null
+++ b/tests/libutil/str_table.c
@@ -0,0 +1,84 @@
+/* SPDX-License-Identifier: GPL-3.0-or-later */
+/*
+ * str_table.c
+ *
+ * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
+ */
+#include "config.h"
+
+#include "str_table.h"
+#include "fstream.h"
+#include "compat.h"
+#include "../test.h"
+
+static char *strings[1000];
+
+static int read_strings(void)
+{
+	istream_t *fp;
+	ssize_t ret;
+	char *line;
+	int i;
+
+	fp = istream_open_file("words.txt");
+	TEST_NOT_NULL(fp);
+
+	for (i = 0; i < 1000; ++i) {
+		ret = istream_get_line(fp, &line, NULL, 0);
+		TEST_EQUAL_I(ret, 0);
+
+		strings[i] = line;
+	}
+
+	sqfs_destroy(fp);
+	return 0;
+}
+
+int main(void)
+{
+	str_table_t table;
+	size_t i, j, idx;
+	const char *str;
+
+	TEST_ASSERT(chdir(TEST_PATH) == 0);
+
+	if (read_strings())
+		return EXIT_FAILURE;
+
+	TEST_ASSERT(str_table_init(&table, 64) == 0);
+
+	for (i = 0; i < 1000; ++i) {
+		TEST_ASSERT(str_table_get_index(&table, strings[i], &idx) == 0);
+
+		TEST_EQUAL_UI(idx, i);
+
+		for (j = 0; j <= i; ++j) {
+			str = str_table_get_string(&table, j);
+
+			TEST_NOT_NULL(str);
+			TEST_ASSERT(str != strings[i]);
+			TEST_STR_EQUAL(str, strings[j]);
+		}
+
+		for (; j < 1000; ++j)
+			TEST_NULL(str_table_get_string(&table, j));
+	}
+
+	for (i = 0; i < 1000; ++i) {
+		TEST_ASSERT(str_table_get_index(&table, strings[i], &idx) == 0);
+		TEST_EQUAL_UI(idx, i);
+
+		str = str_table_get_string(&table, i);
+
+		TEST_NOT_NULL(str);
+		TEST_ASSERT(str != strings[i]);
+		TEST_STR_EQUAL(str, strings[i]);
+	}
+
+	str_table_cleanup(&table);
+
+	for (i = 0; i < 1000; ++i)
+		free(strings[i]);
+
+	return EXIT_SUCCESS;
+}