1 files changed, 113 insertions, 0 deletions
diff --git a/src/lib/test-json-tree.c b/src/lib/test-json-tree.c
new file mode 100644
index 0000000..40eff8c
--- /dev/null
+++ b/src/lib/test-json-tree.c
@@ -0,0 +1,113 @@
+/* Copyright (c) 2015-2018 Dovecot authors, see the included COPYING file */
+
+#include "test-lib.h"
+#include "json-tree.h"
+
+struct {
+	enum json_type type;
+	const char *value;
+} test_input[] = {
+	{ JSON_TYPE_OBJECT_KEY, "key-str" },
+	{ JSON_TYPE_STRING, "string" },
+	{ JSON_TYPE_OBJECT_KEY, "key-num" },
+	{ JSON_TYPE_NUMBER, "1234" },
+	{ JSON_TYPE_OBJECT_KEY, "key-true" },
+	{ JSON_TYPE_TRUE, "true" },
+	{ JSON_TYPE_OBJECT_KEY, "key-false" },
+	{ JSON_TYPE_FALSE, "false" },
+	{ JSON_TYPE_OBJECT_KEY, "key-null" },
+	{ JSON_TYPE_NULL, NULL },
+
+	{ JSON_TYPE_OBJECT_KEY, "key-obj-empty" },
+	{ JSON_TYPE_OBJECT, NULL },
+	{ JSON_TYPE_OBJECT_END, NULL },
+
+	{ JSON_TYPE_OBJECT_KEY, "key-obj" },
+	{ JSON_TYPE_OBJECT, NULL },
+	  { JSON_TYPE_OBJECT_KEY, "sub" },
+	  { JSON_TYPE_STRING, "value" },
+	{ JSON_TYPE_OBJECT_END, NULL },
+
+	{ JSON_TYPE_OBJECT_KEY, "key-arr-empty" },
+	{ JSON_TYPE_ARRAY, NULL },
+	{ JSON_TYPE_ARRAY_END, NULL },
+
+	{ JSON_TYPE_OBJECT_KEY, "key-arr" },
+	{ JSON_TYPE_ARRAY, NULL },
+	  { JSON_TYPE_STRING, "foo" },
+	  { JSON_TYPE_ARRAY, NULL },
+	  { JSON_TYPE_TRUE, "true" },
+	  { JSON_TYPE_ARRAY_END, NULL },
+	  { JSON_TYPE_OBJECT, NULL },
+	    { JSON_TYPE_OBJECT_KEY, "aobj" },
+	    { JSON_TYPE_ARRAY, NULL },
+	    { JSON_TYPE_ARRAY_END, NULL },
+	  { JSON_TYPE_OBJECT_END, NULL },
+	  { JSON_TYPE_OBJECT, NULL },
+	    { JSON_TYPE_OBJECT_KEY, "aobj-key" },
+	    { JSON_TYPE_STRING, "value1" },
+	  { JSON_TYPE_OBJECT_END, NULL },
+	  { JSON_TYPE_OBJECT, NULL },
+	    { JSON_TYPE_OBJECT_KEY, "aobj-key" },
+	    { JSON_TYPE_STRING, "value2" },
+	  { JSON_TYPE_OBJECT_END, NULL },
+	{ JSON_TYPE_ARRAY_END, NULL }
+};
+
+void test_json_tree(void)
+{
+	struct json_tree *tree;
+	const struct json_tree_node *root, *node, *node1, *node2;
+	unsigned int i;
+
+	test_begin("json tree");
+	tree = json_tree_init();
+	for (i = 0; i < N_ELEMENTS(test_input); i++) {
+		test_assert_idx(json_tree_append(tree, test_input[i].type,
+						 test_input[i].value) == 0, i);
+	}
+
+	root = json_tree_root(tree);
+	i_assert(root != NULL);
+	test_assert(root->value_type == JSON_TYPE_OBJECT);
+	i_assert(root != NULL);
+
+	for (i = 0; i < 10+2; i += 2) {
+		node = json_tree_find_key(root, test_input[i].value);
+		test_assert(node != NULL &&
+			    node->value_type == test_input[i+1].type &&
+			    null_strcmp(json_tree_get_value_str(node), test_input[i+1].value) == 0);
+	}
+
+	node = json_tree_find_key(root, "key-obj");
+	test_assert(node != NULL);
+
+	node = json_tree_find_key(root, "key-arr-empty");
+	test_assert(node != NULL && node->value_type == JSON_TYPE_ARRAY &&
+		    json_tree_get_child(node) == NULL);
+
+	node = json_tree_find_key(root, "key-arr");
+	test_assert(node != NULL && node->value_type == JSON_TYPE_ARRAY);
+	node = json_tree_get_child(node);
+	test_assert(node != NULL && node->value_type == JSON_TYPE_STRING &&
+		    strcmp(json_tree_get_value_str(node), "foo") == 0);
+	node = node->next;
+	test_assert(node != NULL && node->value_type == JSON_TYPE_ARRAY &&
+		    json_tree_get_child(node) != NULL &&
+		    json_tree_get_child(node)->next == NULL &&
+		    json_tree_get_child(node)->value_type == JSON_TYPE_TRUE);
+	node = node->next;
+	test_assert(node != NULL && node->value_type == JSON_TYPE_OBJECT &&
+		    json_tree_get_child(node) != NULL &&
+		    json_tree_get_child(node)->next == NULL &&
+		    json_tree_get_child(node)->value_type == JSON_TYPE_ARRAY &&
+		    json_tree_get_child(json_tree_get_child(node)) == NULL);
+
+	node1 = json_tree_find_child_with(node->parent, "aobj-key", "value1");
+	node2 = json_tree_find_child_with(node->parent, "aobj-key", "value2");
+	test_assert(node1 != NULL && node2 != NULL && node1 != node2);
+	test_assert(json_tree_find_child_with(node->parent, "aobj-key", "value3") == NULL);
+
+	json_tree_deinit(&tree);
+	test_end();
+}