src/lib/test-json-tree.c

bcb4e51a409d94ae670de96afb8483a4f7855294Stephan Bosch/* Copyright (c) 2015-2018 Dovecot authors, see the included COPYING file */
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen#include "test-lib.h"
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen#include "json-tree.h"
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainenstruct {
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    enum json_type type;
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    const char *value;
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen} test_input[] = {
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-str" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_STRING, "string" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-num" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_NUMBER, "1234" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-true" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_TRUE, "true" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-false" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_FALSE, "false" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-null" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_NULL, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-obj-empty" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-obj" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_OBJECT_KEY, "sub" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_STRING, "value" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-arr-empty" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_ARRAY, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_ARRAY_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_OBJECT_KEY, "key-arr" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_ARRAY, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_STRING, "foo" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_ARRAY, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_TRUE, "true" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_ARRAY_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_OBJECT, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        { JSON_TYPE_OBJECT_KEY, "aobj" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        { JSON_TYPE_ARRAY, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        { JSON_TYPE_ARRAY_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_OBJECT_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_OBJECT, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        { JSON_TYPE_OBJECT_KEY, "aobj-key" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        { JSON_TYPE_STRING, "value1" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_OBJECT_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_OBJECT, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        { JSON_TYPE_OBJECT_KEY, "aobj-key" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        { JSON_TYPE_STRING, "value2" },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen      { JSON_TYPE_OBJECT_END, NULL },
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    { JSON_TYPE_ARRAY_END, NULL }
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen};
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainenvoid test_json_tree(void)
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen{
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    struct json_tree *tree;
fcd3d6214ce1b8169b6481c78e02d9054901fed3Phil Carmody    const struct json_tree_node *root, *node, *node1, *node2;
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    unsigned int i;
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_begin("json tree");
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    tree = json_tree_init();
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    for (i = 0; i < N_ELEMENTS(test_input); i++) {
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        test_assert_idx(json_tree_append(tree, test_input[i].type,
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen                         test_input[i].value) == 0, i);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    }
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    root = json_tree_root(tree);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    i_assert(root != NULL);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(root->value_type == JSON_TYPE_OBJECT);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    i_assert(root != NULL);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    for (i = 0; i < 10+2; i += 2) {
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        node = json_tree_find_key(root, test_input[i].value);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen        test_assert(node != NULL &&
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen                node->value_type == test_input[i+1].type &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody                null_strcmp(json_tree_get_value_str(node), test_input[i+1].value) == 0);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    }
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    node = json_tree_find_key(root, "key-obj");
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(node != NULL);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    node = json_tree_find_key(root, "key-arr-empty");
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(node != NULL && node->value_type == JSON_TYPE_ARRAY &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(node) == NULL);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    node = json_tree_find_key(root, "key-arr");
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(node != NULL && node->value_type == JSON_TYPE_ARRAY);
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody    node = json_tree_get_child(node);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(node != NULL && node->value_type == JSON_TYPE_STRING &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            strcmp(json_tree_get_value_str(node), "foo") == 0);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    node = node->next;
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(node != NULL && node->value_type == JSON_TYPE_ARRAY &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(node) != NULL &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(node)->next == NULL &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(node)->value_type == JSON_TYPE_TRUE);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    node = node->next;
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(node != NULL && node->value_type == JSON_TYPE_OBJECT &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(node) != NULL &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(node)->next == NULL &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(node)->value_type == JSON_TYPE_ARRAY &&
f946ee9814f653cd8baa8596277861b4582c7887Phil Carmody            json_tree_get_child(json_tree_get_child(node)) == NULL);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    node1 = json_tree_find_child_with(node->parent, "aobj-key", "value1");
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    node2 = json_tree_find_child_with(node->parent, "aobj-key", "value2");
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(node1 != NULL && node2 != NULL && node1 != node2);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_assert(json_tree_find_child_with(node->parent, "aobj-key", "value3") == NULL);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    json_tree_deinit(&tree);
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen    test_end();
31e6dbee5fc7d6c33b44c75c7e18ea576f44184bTimo Sirainen}