/* 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();
}