[mirror_qemu.git] / tests / check-qdict.c

/*
 * QDict unit-tests.
 *
 * Copyright (C) 2009 Red Hat Inc.
 *
 * Authors:
 *  Luiz Capitulino <lcapitulino@redhat.com>
 *
 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
 * See the COPYING.LIB file in the top-level directory.
 */
#include "qemu/osdep.h"
#include <glib.h>

#include "qapi/qmp/qint.h"
#include "qapi/qmp/qdict.h"
#include "qapi/qmp/qstring.h"
#include "qemu-common.h"

/*
 * Public Interface test-cases
 *
 * (with some violations to access 'private' data)
 */

static void qdict_new_test(void)
{
    QDict *qdict;

    qdict = qdict_new();
    g_assert(qdict != NULL);
    g_assert(qdict_size(qdict) == 0);
    g_assert(qdict->base.refcnt == 1);
    g_assert(qobject_type(QOBJECT(qdict)) == QTYPE_QDICT);

    // destroy doesn't exit yet
    g_free(qdict);
}

static void qdict_put_obj_test(void)
{
    QInt *qi;
    QDict *qdict;
    QDictEntry *ent;
    const int num = 42;

    qdict = qdict_new();

    // key "" will have tdb hash 12345
    qdict_put_obj(qdict, "", QOBJECT(qint_from_int(num)));

    g_assert(qdict_size(qdict) == 1);
    ent = QLIST_FIRST(&qdict->table[12345 % QDICT_BUCKET_MAX]);
    qi = qobject_to_qint(ent->value);
    g_assert(qint_get_int(qi) == num);

    // destroy doesn't exit yet
    QDECREF(qi);
    g_free(ent->key);
    g_free(ent);
    g_free(qdict);
}

static void qdict_destroy_simple_test(void)
{
    QDict *qdict;

    qdict = qdict_new();
    qdict_put_obj(qdict, "num", QOBJECT(qint_from_int(0)));
    qdict_put_obj(qdict, "str", QOBJECT(qstring_from_str("foo")));

    QDECREF(qdict);
}

static void qdict_get_test(void)
{
    QInt *qi;
    QObject *obj;
    const int value = -42;
    const char *key = "test";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qint_from_int(value));

    obj = qdict_get(tests_dict, key);
    g_assert(obj != NULL);

    qi = qobject_to_qint(obj);
    g_assert(qint_get_int(qi) == value);

    QDECREF(tests_dict);
}

static void qdict_get_int_test(void)
{
    int ret;
    const int value = 100;
    const char *key = "int";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qint_from_int(value));

    ret = qdict_get_int(tests_dict, key);
    g_assert(ret == value);

    QDECREF(tests_dict);
}

static void qdict_get_try_int_test(void)
{
    int ret;
    const int value = 100;
    const char *key = "int";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qint_from_int(value));

    ret = qdict_get_try_int(tests_dict, key, 0);
    g_assert(ret == value);

    QDECREF(tests_dict);
}

static void qdict_get_str_test(void)
{
    const char *p;
    const char *key = "key";
    const char *str = "string";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qstring_from_str(str));

    p = qdict_get_str(tests_dict, key);
    g_assert(p != NULL);
    g_assert(strcmp(p, str) == 0);

    QDECREF(tests_dict);
}

static void qdict_get_try_str_test(void)
{
    const char *p;
    const char *key = "key";
    const char *str = "string";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qstring_from_str(str));

    p = qdict_get_try_str(tests_dict, key);
    g_assert(p != NULL);
    g_assert(strcmp(p, str) == 0);

    QDECREF(tests_dict);
}

static void qdict_defaults_test(void)
{
    QDict *dict, *copy;

    dict = qdict_new();
    copy = qdict_new();

    qdict_set_default_str(dict, "foo", "abc");
    qdict_set_default_str(dict, "foo", "def");
    g_assert_cmpstr(qdict_get_str(dict, "foo"), ==, "abc");
    qdict_set_default_str(dict, "bar", "ghi");

    qdict_copy_default(copy, dict, "foo");
    g_assert_cmpstr(qdict_get_str(copy, "foo"), ==, "abc");
    qdict_set_default_str(copy, "bar", "xyz");
    qdict_copy_default(copy, dict, "bar");
    g_assert_cmpstr(qdict_get_str(copy, "bar"), ==, "xyz");

    QDECREF(copy);
    QDECREF(dict);
}

static void qdict_haskey_not_test(void)
{
    QDict *tests_dict = qdict_new();
    g_assert(qdict_haskey(tests_dict, "test") == 0);

    QDECREF(tests_dict);
}

static void qdict_haskey_test(void)
{
    const char *key = "test";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qint_from_int(0));
    g_assert(qdict_haskey(tests_dict, key) == 1);

    QDECREF(tests_dict);
}

static void qdict_del_test(void)
{
    const char *key = "key test";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qstring_from_str("foo"));
    g_assert(qdict_size(tests_dict) == 1);

    qdict_del(tests_dict, key);

    g_assert(qdict_size(tests_dict) == 0);
    g_assert(qdict_haskey(tests_dict, key) == 0);

    QDECREF(tests_dict);
}

static void qobject_to_qdict_test(void)
{
    QDict *tests_dict = qdict_new();
    g_assert(qobject_to_qdict(QOBJECT(tests_dict)) == tests_dict);

    QDECREF(tests_dict);
}

static void qdict_iterapi_test(void)
{
    int count;
    const QDictEntry *ent;
    QDict *tests_dict = qdict_new();

    g_assert(qdict_first(tests_dict) == NULL);

    qdict_put(tests_dict, "key1", qint_from_int(1));
    qdict_put(tests_dict, "key2", qint_from_int(2));
    qdict_put(tests_dict, "key3", qint_from_int(3));

    count = 0;
    for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
        g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
        count++;
    }

    g_assert(count == qdict_size(tests_dict));

    /* Do it again to test restarting */
    count = 0;
    for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
        g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
        count++;
    }

    g_assert(count == qdict_size(tests_dict));

    QDECREF(tests_dict);
}

static void qdict_flatten_test(void)
{
    QList *list1 = qlist_new();
    QList *list2 = qlist_new();
    QDict *dict1 = qdict_new();
    QDict *dict2 = qdict_new();
    QDict *dict3 = qdict_new();

    /*
     * Test the flattening of
     *
     * {
     *     "e": [
     *         42,
     *         [
     *             23,
     *             66,
     *             {
     *                 "a": 0,
     *                 "b": 1
     *             }
     *         ]
     *     ],
     *     "f": {
     *         "c": 2,
     *         "d": 3,
     *     },
     *     "g": 4
     * }
     *
     * to
     *
     * {
     *     "e.0": 42,
     *     "e.1.0": 23,
     *     "e.1.1": 66,
     *     "e.1.2.a": 0,
     *     "e.1.2.b": 1,
     *     "f.c": 2,
     *     "f.d": 3,
     *     "g": 4
     * }
     */

    qdict_put(dict1, "a", qint_from_int(0));
    qdict_put(dict1, "b", qint_from_int(1));

    qlist_append_obj(list1, QOBJECT(qint_from_int(23)));
    qlist_append_obj(list1, QOBJECT(qint_from_int(66)));
    qlist_append_obj(list1, QOBJECT(dict1));
    qlist_append_obj(list2, QOBJECT(qint_from_int(42)));
    qlist_append_obj(list2, QOBJECT(list1));

    qdict_put(dict2, "c", qint_from_int(2));
    qdict_put(dict2, "d", qint_from_int(3));
    qdict_put_obj(dict3, "e", QOBJECT(list2));
    qdict_put_obj(dict3, "f", QOBJECT(dict2));
    qdict_put(dict3, "g", qint_from_int(4));

    qdict_flatten(dict3);

    g_assert(qdict_get_int(dict3, "e.0") == 42);
    g_assert(qdict_get_int(dict3, "e.1.0") == 23);
    g_assert(qdict_get_int(dict3, "e.1.1") == 66);
    g_assert(qdict_get_int(dict3, "e.1.2.a") == 0);
    g_assert(qdict_get_int(dict3, "e.1.2.b") == 1);
    g_assert(qdict_get_int(dict3, "f.c") == 2);
    g_assert(qdict_get_int(dict3, "f.d") == 3);
    g_assert(qdict_get_int(dict3, "g") == 4);

    g_assert(qdict_size(dict3) == 8);

    QDECREF(dict3);
}

static void qdict_array_split_test(void)
{
    QDict *test_dict = qdict_new();
    QDict *dict1, *dict2;
    QInt *int1;
    QList *test_list;

    /*
     * Test the split of
     *
     * {
     *     "1.x": 0,
     *     "4.y": 1,
     *     "0.a": 42,
     *     "o.o": 7,
     *     "0.b": 23,
     *     "2": 66
     * }
     *
     * to
     *
     * [
     *     {
     *         "a": 42,
     *         "b": 23
     *     },
     *     {
     *         "x": 0
     *     },
     *     66
     * ]
     *
     * and
     *
     * {
     *     "4.y": 1,
     *     "o.o": 7
     * }
     *
     * (remaining in the old QDict)
     *
     * This example is given in the comment of qdict_array_split().
     */

    qdict_put(test_dict, "1.x", qint_from_int(0));
    qdict_put(test_dict, "4.y", qint_from_int(1));
    qdict_put(test_dict, "0.a", qint_from_int(42));
    qdict_put(test_dict, "o.o", qint_from_int(7));
    qdict_put(test_dict, "0.b", qint_from_int(23));
    qdict_put(test_dict, "2", qint_from_int(66));

    qdict_array_split(test_dict, &test_list);

    dict1 = qobject_to_qdict(qlist_pop(test_list));
    dict2 = qobject_to_qdict(qlist_pop(test_list));
    int1 = qobject_to_qint(qlist_pop(test_list));

    g_assert(dict1);
    g_assert(dict2);
    g_assert(int1);
    g_assert(qlist_empty(test_list));

    QDECREF(test_list);

    g_assert(qdict_get_int(dict1, "a") == 42);
    g_assert(qdict_get_int(dict1, "b") == 23);

    g_assert(qdict_size(dict1) == 2);

    QDECREF(dict1);

    g_assert(qdict_get_int(dict2, "x") == 0);

    g_assert(qdict_size(dict2) == 1);

    QDECREF(dict2);

    g_assert(qint_get_int(int1) == 66);

    QDECREF(int1);

    g_assert(qdict_get_int(test_dict, "4.y") == 1);
    g_assert(qdict_get_int(test_dict, "o.o") == 7);

    g_assert(qdict_size(test_dict) == 2);

    QDECREF(test_dict);


    /*
     * Test the split of
     *
     * {
     *     "0": 42,
     *     "1": 23,
     *     "1.x": 84
     * }
     *
     * to
     *
     * [
     *     42
     * ]
     *
     * and
     *
     * {
     *     "1": 23,
     *     "1.x": 84
     * }
     *
     * That is, test whether splitting stops if there is both an entry with key
     * of "%u" and other entries with keys prefixed "%u." for the same index.
     */

    test_dict = qdict_new();

    qdict_put(test_dict, "0", qint_from_int(42));
    qdict_put(test_dict, "1", qint_from_int(23));
    qdict_put(test_dict, "1.x", qint_from_int(84));

    qdict_array_split(test_dict, &test_list);

    int1 = qobject_to_qint(qlist_pop(test_list));

    g_assert(int1);
    g_assert(qlist_empty(test_list));

    QDECREF(test_list);

    g_assert(qint_get_int(int1) == 42);

    QDECREF(int1);

    g_assert(qdict_get_int(test_dict, "1") == 23);
    g_assert(qdict_get_int(test_dict, "1.x") == 84);

    g_assert(qdict_size(test_dict) == 2);

    QDECREF(test_dict);
}

static void qdict_array_entries_test(void)
{
    QDict *dict = qdict_new();

    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);

    qdict_put(dict, "bar", qint_from_int(0));
    qdict_put(dict, "baz.0", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);

    qdict_put(dict, "foo.1", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
    qdict_put(dict, "foo.0", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 2);
    qdict_put(dict, "foo.bar", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
    qdict_del(dict, "foo.bar");

    qdict_put(dict, "foo.2.a", qint_from_int(0));
    qdict_put(dict, "foo.2.b", qint_from_int(0));
    qdict_put(dict, "foo.2.c", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 3);
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);

    QDECREF(dict);

    dict = qdict_new();
    qdict_put(dict, "1", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
    qdict_put(dict, "0", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, 2);
    qdict_put(dict, "bar", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
    qdict_del(dict, "bar");

    qdict_put(dict, "2.a", qint_from_int(0));
    qdict_put(dict, "2.b", qint_from_int(0));
    qdict_put(dict, "2.c", qint_from_int(0));
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, 3);

    QDECREF(dict);
}

static void qdict_join_test(void)
{
    QDict *dict1, *dict2;
    bool overwrite = false;
    int i;

    dict1 = qdict_new();
    dict2 = qdict_new();


    /* Test everything once without overwrite and once with */
    do
    {
        /* Test empty dicts */
        qdict_join(dict1, dict2, overwrite);

        g_assert(qdict_size(dict1) == 0);
        g_assert(qdict_size(dict2) == 0);


        /* First iteration: Test movement */
        /* Second iteration: Test empty source and non-empty destination */
        qdict_put(dict2, "foo", qint_from_int(42));

        for (i = 0; i < 2; i++) {
            qdict_join(dict1, dict2, overwrite);

            g_assert(qdict_size(dict1) == 1);
            g_assert(qdict_size(dict2) == 0);

            g_assert(qdict_get_int(dict1, "foo") == 42);
        }


        /* Test non-empty source and destination without conflict */
        qdict_put(dict2, "bar", qint_from_int(23));

        qdict_join(dict1, dict2, overwrite);

        g_assert(qdict_size(dict1) == 2);
        g_assert(qdict_size(dict2) == 0);

        g_assert(qdict_get_int(dict1, "foo") == 42);
        g_assert(qdict_get_int(dict1, "bar") == 23);


        /* Test conflict */
        qdict_put(dict2, "foo", qint_from_int(84));

        qdict_join(dict1, dict2, overwrite);

        g_assert(qdict_size(dict1) == 2);
        g_assert(qdict_size(dict2) == !overwrite);

        g_assert(qdict_get_int(dict1, "foo") == overwrite ? 84 : 42);
        g_assert(qdict_get_int(dict1, "bar") == 23);

        if (!overwrite) {
            g_assert(qdict_get_int(dict2, "foo") == 84);
        }


        /* Check the references */
        g_assert(qdict_get(dict1, "foo")->refcnt == 1);
        g_assert(qdict_get(dict1, "bar")->refcnt == 1);

        if (!overwrite) {
            g_assert(qdict_get(dict2, "foo")->refcnt == 1);
        }


        /* Clean up */
        qdict_del(dict1, "foo");
        qdict_del(dict1, "bar");

        if (!overwrite) {
            qdict_del(dict2, "foo");
        }
    }
    while (overwrite ^= true);


    QDECREF(dict1);
    QDECREF(dict2);
}

/*
 * Errors test-cases
 */

static void qdict_put_exists_test(void)
{
    int value;
    const char *key = "exists";
    QDict *tests_dict = qdict_new();

    qdict_put(tests_dict, key, qint_from_int(1));
    qdict_put(tests_dict, key, qint_from_int(2));

    value = qdict_get_int(tests_dict, key);
    g_assert(value == 2);

    g_assert(qdict_size(tests_dict) == 1);

    QDECREF(tests_dict);
}

static void qdict_get_not_exists_test(void)
{
    QDict *tests_dict = qdict_new();
    g_assert(qdict_get(tests_dict, "foo") == NULL);

    QDECREF(tests_dict);
}

/*
 * Stress test-case
 *
 * This is a lot big for a unit-test, but there is no other place
 * to have it.
 */

static void remove_dots(char *string)
{
    char *p = strchr(string, ':');
    if (p)
        *p = '\0';
}

static QString *read_line(FILE *file, char *key)
{
    char value[128];

    if (fscanf(file, "%127s%127s", key, value) == EOF) {
        return NULL;
    }
    remove_dots(key);
    return qstring_from_str(value);
}

#define reset_file(file)    fseek(file, 0L, SEEK_SET)

static void qdict_stress_test(void)
{
    size_t lines;
    char key[128];
    FILE *test_file;
    QDict *qdict;
    QString *value;
    const char *test_file_path = "qdict-test-data.txt";

    test_file = fopen(test_file_path, "r");
    g_assert(test_file != NULL);

    // Create the dict
    qdict = qdict_new();
    g_assert(qdict != NULL);

    // Add everything from the test file
    for (lines = 0;; lines++) {
        value = read_line(test_file, key);
        if (!value)
            break;

        qdict_put(qdict, key, value);
    }
    g_assert(qdict_size(qdict) == lines);

    // Check if everything is really in there
    reset_file(test_file);
    for (;;) {
        const char *str1, *str2;

        value = read_line(test_file, key);
        if (!value)
            break;

        str1 = qstring_get_str(value);

        str2 = qdict_get_str(qdict, key);
        g_assert(str2 != NULL);

        g_assert(strcmp(str1, str2) == 0);

        QDECREF(value);
    }

    // Delete everything
    reset_file(test_file);
    for (;;) {
        value = read_line(test_file, key);
        if (!value)
            break;

        qdict_del(qdict, key);
        QDECREF(value);

        g_assert(qdict_haskey(qdict, key) == 0);
    }
    fclose(test_file);

    g_assert(qdict_size(qdict) == 0);
    QDECREF(qdict);
}

int main(int argc, char **argv)
{
    g_test_init(&argc, &argv, NULL);

    g_test_add_func("/public/new", qdict_new_test);
    g_test_add_func("/public/put_obj", qdict_put_obj_test);
    g_test_add_func("/public/destroy_simple", qdict_destroy_simple_test);

    /* Continue, but now with fixtures */
    g_test_add_func("/public/get", qdict_get_test);
    g_test_add_func("/public/get_int", qdict_get_int_test);
    g_test_add_func("/public/get_try_int", qdict_get_try_int_test);
    g_test_add_func("/public/get_str", qdict_get_str_test);
    g_test_add_func("/public/get_try_str", qdict_get_try_str_test);
    g_test_add_func("/public/defaults", qdict_defaults_test);
    g_test_add_func("/public/haskey_not", qdict_haskey_not_test);
    g_test_add_func("/public/haskey", qdict_haskey_test);
    g_test_add_func("/public/del", qdict_del_test);
    g_test_add_func("/public/to_qdict", qobject_to_qdict_test);
    g_test_add_func("/public/iterapi", qdict_iterapi_test);
    g_test_add_func("/public/flatten", qdict_flatten_test);
    g_test_add_func("/public/array_split", qdict_array_split_test);
    g_test_add_func("/public/array_entries", qdict_array_entries_test);
    g_test_add_func("/public/join", qdict_join_test);

    g_test_add_func("/errors/put_exists", qdict_put_exists_test);
    g_test_add_func("/errors/get_not_exists", qdict_get_not_exists_test);

    /* The Big one */
    if (g_test_slow()) {
        g_test_add_func("/stress/test", qdict_stress_test);
    }

    return g_test_run();
}
Commit	Line	Data
7b8c51ad LC	1	/*
	2	* QDict unit-tests.
	3	*
	4	* Copyright (C) 2009 Red Hat Inc.
	5	*
	6	* Authors:
	7	* Luiz Capitulino <lcapitulino@redhat.com>
41836a9f LC	8	*
	9	* This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
	10	* See the COPYING.LIB file in the top-level directory.
7b8c51ad	11	*/
681c28a3	12	#include "qemu/osdep.h"
ac531cb6	13	#include <glib.h>
7b8c51ad	14
7b1b5d19 PB	15	#include "qapi/qmp/qint.h"
	16	#include "qapi/qmp/qdict.h"
	17	#include "qapi/qmp/qstring.h"
7b8c51ad LC	18	#include "qemu-common.h"
	19
	20	/*
	21	* Public Interface test-cases
	22	*
	23	* (with some violations to access 'private' data)
	24	*/
	25
ac531cb6	26	static void qdict_new_test(void)
7b8c51ad LC	27	{
	28	QDict *qdict;
	29
	30	qdict = qdict_new();
ac531cb6 AL	31	g_assert(qdict != NULL);
	32	g_assert(qdict_size(qdict) == 0);
	33	g_assert(qdict->base.refcnt == 1);
	34	g_assert(qobject_type(QOBJECT(qdict)) == QTYPE_QDICT);
7b8c51ad LC	35
7b8c51ad LC	36	// destroy doesn't exit yet
ac531cb6	37	g_free(qdict);
7b8c51ad	38	}
7b8c51ad	39
ac531cb6	40	static void qdict_put_obj_test(void)
7b8c51ad LC	41	{
	42	QInt *qi;
	43	QDict *qdict;
	44	QDictEntry *ent;
	45	const int num = 42;
	46
	47	qdict = qdict_new();
	48
	49	// key "" will have tdb hash 12345
	50	qdict_put_obj(qdict, "", QOBJECT(qint_from_int(num)));
	51
ac531cb6	52	g_assert(qdict_size(qdict) == 1);
c8bc3cd7	53	ent = QLIST_FIRST(&qdict->table[12345 % QDICT_BUCKET_MAX]);
7b8c51ad	54	qi = qobject_to_qint(ent->value);
ac531cb6	55	g_assert(qint_get_int(qi) == num);
7b8c51ad LC	56
	57	// destroy doesn't exit yet
	58	QDECREF(qi);
7267c094 AL	59	g_free(ent->key);
	60	g_free(ent);
	61	g_free(qdict);
7b8c51ad	62	}
7b8c51ad	63
ac531cb6	64	static void qdict_destroy_simple_test(void)
7b8c51ad LC	65	{
	66	QDict *qdict;
	67
	68	qdict = qdict_new();
	69	qdict_put_obj(qdict, "num", QOBJECT(qint_from_int(0)));
	70	qdict_put_obj(qdict, "str", QOBJECT(qstring_from_str("foo")));
	71
	72	QDECREF(qdict);
	73	}
7b8c51ad	74
ac531cb6	75	static void qdict_get_test(void)
7b8c51ad LC	76	{
	77	QInt *qi;
	78	QObject *obj;
	79	const int value = -42;
	80	const char *key = "test";
ac531cb6	81	QDict *tests_dict = qdict_new();
7b8c51ad LC	82
	83	qdict_put(tests_dict, key, qint_from_int(value));
	84
	85	obj = qdict_get(tests_dict, key);
ac531cb6	86	g_assert(obj != NULL);
7b8c51ad LC	87
7b8c51ad LC	88	qi = qobject_to_qint(obj);
ac531cb6 AL	89	g_assert(qint_get_int(qi) == value);
	90
	91	QDECREF(tests_dict);
7b8c51ad	92	}
7b8c51ad	93
ac531cb6	94	static void qdict_get_int_test(void)
7b8c51ad LC	95	{
	96	int ret;
	97	const int value = 100;
	98	const char *key = "int";
ac531cb6	99	QDict *tests_dict = qdict_new();
7b8c51ad LC	100
	101	qdict_put(tests_dict, key, qint_from_int(value));
	102
	103	ret = qdict_get_int(tests_dict, key);
ac531cb6 AL	104	g_assert(ret == value);
	105
	106	QDECREF(tests_dict);
7b8c51ad	107	}
7b8c51ad	108
ac531cb6	109	static void qdict_get_try_int_test(void)
7b8c51ad LC	110	{
	111	int ret;
	112	const int value = 100;
	113	const char *key = "int";
ac531cb6	114	QDict *tests_dict = qdict_new();
7b8c51ad LC	115
	116	qdict_put(tests_dict, key, qint_from_int(value));
	117
	118	ret = qdict_get_try_int(tests_dict, key, 0);
ac531cb6 AL	119	g_assert(ret == value);
	120
	121	QDECREF(tests_dict);
7b8c51ad	122	}
7b8c51ad	123
ac531cb6	124	static void qdict_get_str_test(void)
7b8c51ad LC	125	{
	126	const char *p;
	127	const char *key = "key";
	128	const char *str = "string";
ac531cb6	129	QDict *tests_dict = qdict_new();
7b8c51ad LC	130
	131	qdict_put(tests_dict, key, qstring_from_str(str));
	132
	133	p = qdict_get_str(tests_dict, key);
ac531cb6 AL	134	g_assert(p != NULL);
	135	g_assert(strcmp(p, str) == 0);
	136
	137	QDECREF(tests_dict);
7b8c51ad	138	}
7b8c51ad	139
ac531cb6	140	static void qdict_get_try_str_test(void)
7b8c51ad LC	141	{
	142	const char *p;
	143	const char *key = "key";
	144	const char *str = "string";
ac531cb6	145	QDict *tests_dict = qdict_new();
7b8c51ad LC	146
	147	qdict_put(tests_dict, key, qstring_from_str(str));
	148
	149	p = qdict_get_try_str(tests_dict, key);
ac531cb6 AL	150	g_assert(p != NULL);
	151	g_assert(strcmp(p, str) == 0);
	152
	153	QDECREF(tests_dict);
7b8c51ad	154	}
7b8c51ad	155
ef1919df KW	156	static void qdict_defaults_test(void)
	157	{
	158	QDict dict, copy;
	159
	160	dict = qdict_new();
	161	copy = qdict_new();
	162
	163	qdict_set_default_str(dict, "foo", "abc");
	164	qdict_set_default_str(dict, "foo", "def");
	165	g_assert_cmpstr(qdict_get_str(dict, "foo"), ==, "abc");
	166	qdict_set_default_str(dict, "bar", "ghi");
	167
	168	qdict_copy_default(copy, dict, "foo");
	169	g_assert_cmpstr(qdict_get_str(copy, "foo"), ==, "abc");
	170	qdict_set_default_str(copy, "bar", "xyz");
	171	qdict_copy_default(copy, dict, "bar");
	172	g_assert_cmpstr(qdict_get_str(copy, "bar"), ==, "xyz");
	173
	174	QDECREF(copy);
	175	QDECREF(dict);
	176	}
	177
ac531cb6	178	static void qdict_haskey_not_test(void)
7b8c51ad	179	{
ac531cb6 AL	180	QDict *tests_dict = qdict_new();
	181	g_assert(qdict_haskey(tests_dict, "test") == 0);
	182
	183	QDECREF(tests_dict);
7b8c51ad	184	}
7b8c51ad	185
ac531cb6	186	static void qdict_haskey_test(void)
7b8c51ad LC	187	{
7b8c51ad LC	188	const char *key = "test";
ac531cb6	189	QDict *tests_dict = qdict_new();
7b8c51ad LC	190
7b8c51ad LC	191	qdict_put(tests_dict, key, qint_from_int(0));
ac531cb6 AL	192	g_assert(qdict_haskey(tests_dict, key) == 1);
	193
	194	QDECREF(tests_dict);
7b8c51ad	195	}
7b8c51ad	196
ac531cb6	197	static void qdict_del_test(void)
7b8c51ad LC	198	{
7b8c51ad LC	199	const char *key = "key test";
ac531cb6	200	QDict *tests_dict = qdict_new();
7b8c51ad LC	201
7b8c51ad LC	202	qdict_put(tests_dict, key, qstring_from_str("foo"));
ac531cb6	203	g_assert(qdict_size(tests_dict) == 1);
7b8c51ad LC	204
	205	qdict_del(tests_dict, key);
	206
ac531cb6 AL	207	g_assert(qdict_size(tests_dict) == 0);
	208	g_assert(qdict_haskey(tests_dict, key) == 0);
	209
	210	QDECREF(tests_dict);
7b8c51ad	211	}
7b8c51ad	212
ac531cb6	213	static void qobject_to_qdict_test(void)
7b8c51ad	214	{
ac531cb6 AL	215	QDict *tests_dict = qdict_new();
	216	g_assert(qobject_to_qdict(QOBJECT(tests_dict)) == tests_dict);
	217
	218	QDECREF(tests_dict);
7b8c51ad	219	}
7b8c51ad	220
ac531cb6	221	static void qdict_iterapi_test(void)
d02c6bd4 LC	222	{
	223	int count;
	224	const QDictEntry *ent;
ac531cb6	225	QDict *tests_dict = qdict_new();
d02c6bd4	226
ac531cb6	227	g_assert(qdict_first(tests_dict) == NULL);
d02c6bd4 LC	228
	229	qdict_put(tests_dict, "key1", qint_from_int(1));
	230	qdict_put(tests_dict, "key2", qint_from_int(2));
	231	qdict_put(tests_dict, "key3", qint_from_int(3));
	232
	233	count = 0;
	234	for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
ac531cb6	235	g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
d02c6bd4 LC	236	count++;
	237	}
	238
ac531cb6	239	g_assert(count == qdict_size(tests_dict));
d02c6bd4 LC	240
	241	/* Do it again to test restarting */
	242	count = 0;
	243	for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
ac531cb6	244	g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
d02c6bd4 LC	245	count++;
	246	}
	247
ac531cb6 AL	248	g_assert(count == qdict_size(tests_dict));
	249
	250	QDECREF(tests_dict);
d02c6bd4	251	}
d02c6bd4	252
3fb11779 HR	253	static void qdict_flatten_test(void)
	254	{
	255	QList *list1 = qlist_new();
	256	QList *list2 = qlist_new();
	257	QDict *dict1 = qdict_new();
	258	QDict *dict2 = qdict_new();
	259	QDict *dict3 = qdict_new();
	260
	261	/*
	262	* Test the flattening of
	263	*
	264	* {
	265	* "e": [
	266	* 42,
	267	* [
	268	* 23,
	269	* 66,
	270	* {
	271	* "a": 0,
	272	* "b": 1
	273	* }
	274	* ]
	275	* ],
	276	* "f": {
	277	* "c": 2,
	278	* "d": 3,
	279	* },
	280	* "g": 4
	281	* }
	282	*
	283	* to
	284	*
	285	* {
	286	* "e.0": 42,
	287	* "e.1.0": 23,
	288	* "e.1.1": 66,
	289	* "e.1.2.a": 0,
	290	* "e.1.2.b": 1,
	291	* "f.c": 2,
	292	* "f.d": 3,
	293	* "g": 4
	294	* }
	295	*/
	296
	297	qdict_put(dict1, "a", qint_from_int(0));
	298	qdict_put(dict1, "b", qint_from_int(1));
	299
	300	qlist_append_obj(list1, QOBJECT(qint_from_int(23)));
	301	qlist_append_obj(list1, QOBJECT(qint_from_int(66)));
	302	qlist_append_obj(list1, QOBJECT(dict1));
	303	qlist_append_obj(list2, QOBJECT(qint_from_int(42)));
	304	qlist_append_obj(list2, QOBJECT(list1));
	305
	306	qdict_put(dict2, "c", qint_from_int(2));
	307	qdict_put(dict2, "d", qint_from_int(3));
	308	qdict_put_obj(dict3, "e", QOBJECT(list2));
	309	qdict_put_obj(dict3, "f", QOBJECT(dict2));
	310	qdict_put(dict3, "g", qint_from_int(4));
	311
	312	qdict_flatten(dict3);
	313
	314	g_assert(qdict_get_int(dict3, "e.0") == 42);
	315	g_assert(qdict_get_int(dict3, "e.1.0") == 23);
	316	g_assert(qdict_get_int(dict3, "e.1.1") == 66);
317	g_assert(qdict_get_int(dict3, "e.1.2.a") == 0);
318	g_assert(qdict_get_int(dict3, "e.1.2.b") == 1);
319	g_assert(qdict_get_int(dict3, "f.c") == 2);
320	g_assert(qdict_get_int(dict3, "f.d") == 3);
321	g_assert(qdict_get_int(dict3, "g") == 4);
322
323	g_assert(qdict_size(dict3) == 8);
324
325	QDECREF(dict3);
326	}
327
be331341 HR	328	static void qdict_array_split_test(void)
	329	{
	330	QDict *test_dict = qdict_new();
	331	QDict dict1, dict2;
7841c768	332	QInt *int1;
be331341 HR	333	QList *test_list;
	334
	335	/*
	336	* Test the split of
	337	*
	338	* {
	339	* "1.x": 0,
7841c768	340	* "4.y": 1,
be331341 HR	341	* "0.a": 42,
be331341 HR	342	* "o.o": 7,
7841c768 HR	343	* "0.b": 23,
7841c768 HR	344	* "2": 66
be331341 HR	345	* }
	346	*
	347	* to
	348	*
	349	* [
	350	* {
	351	* "a": 42,
	352	* "b": 23
	353	* },
	354	* {
	355	* "x": 0
7841c768 HR	356	* },
7841c768 HR	357	* 66
be331341 HR	358	* ]
	359	*
	360	* and
	361	*
	362	* {
7841c768	363	* "4.y": 1,
be331341 HR	364	* "o.o": 7
	365	* }
	366	*
	367	* (remaining in the old QDict)
	368	*
	369	* This example is given in the comment of qdict_array_split().
	370	*/
	371
	372	qdict_put(test_dict, "1.x", qint_from_int(0));
7841c768	373	qdict_put(test_dict, "4.y", qint_from_int(1));
be331341 HR	374	qdict_put(test_dict, "0.a", qint_from_int(42));
	375	qdict_put(test_dict, "o.o", qint_from_int(7));
	376	qdict_put(test_dict, "0.b", qint_from_int(23));
7841c768	377	qdict_put(test_dict, "2", qint_from_int(66));
be331341 HR	378
	379	qdict_array_split(test_dict, &test_list);
	380
	381	dict1 = qobject_to_qdict(qlist_pop(test_list));
	382	dict2 = qobject_to_qdict(qlist_pop(test_list));
7841c768	383	int1 = qobject_to_qint(qlist_pop(test_list));
be331341 HR	384
	385	g_assert(dict1);
	386	g_assert(dict2);
7841c768	387	g_assert(int1);
be331341 HR	388	g_assert(qlist_empty(test_list));
	389
	390	QDECREF(test_list);
	391
	392	g_assert(qdict_get_int(dict1, "a") == 42);
	393	g_assert(qdict_get_int(dict1, "b") == 23);
	394
	395	g_assert(qdict_size(dict1) == 2);
	396
	397	QDECREF(dict1);
	398
	399	g_assert(qdict_get_int(dict2, "x") == 0);
	400
	401	g_assert(qdict_size(dict2) == 1);
	402
	403	QDECREF(dict2);
	404
7841c768 HR	405	g_assert(qint_get_int(int1) == 66);
	406
	407	QDECREF(int1);
	408
	409	g_assert(qdict_get_int(test_dict, "4.y") == 1);
be331341 HR	410	g_assert(qdict_get_int(test_dict, "o.o") == 7);
	411
	412	g_assert(qdict_size(test_dict) == 2);
	413
	414	QDECREF(test_dict);
64757582 HR	415
	416
	417	/*
	418	* Test the split of
	419	*
	420	* {
	421	* "0": 42,
	422	* "1": 23,
	423	* "1.x": 84
	424	* }
	425	*
	426	* to
	427	*
	428	* [
	429	* 42
	430	* ]
	431	*
	432	* and
	433	*
	434	* {
	435	* "1": 23,
	436	* "1.x": 84
	437	* }
	438	*
	439	* That is, test whether splitting stops if there is both an entry with key
	440	* of "%u" and other entries with keys prefixed "%u." for the same index.
	441	*/
	442
	443	test_dict = qdict_new();
	444
	445	qdict_put(test_dict, "0", qint_from_int(42));
	446	qdict_put(test_dict, "1", qint_from_int(23));
	447	qdict_put(test_dict, "1.x", qint_from_int(84));
	448
	449	qdict_array_split(test_dict, &test_list);
	450
	451	int1 = qobject_to_qint(qlist_pop(test_list));
	452
	453	g_assert(int1);
	454	g_assert(qlist_empty(test_list));
	455
	456	QDECREF(test_list);
	457
	458	g_assert(qint_get_int(int1) == 42);
	459
	460	QDECREF(int1);
	461
	462	g_assert(qdict_get_int(test_dict, "1") == 23);
	463	g_assert(qdict_get_int(test_dict, "1.x") == 84);
	464
	465	g_assert(qdict_size(test_dict) == 2);
	466
	467	QDECREF(test_dict);
be331341 HR	468	}
be331341 HR	469
ef1919df KW	470	static void qdict_array_entries_test(void)
	471	{
	472	QDict *dict = qdict_new();
	473
	474	g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);
	475
	476	qdict_put(dict, "bar", qint_from_int(0));
	477	qdict_put(dict, "baz.0", qint_from_int(0));
	478	g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);
	479
	480	qdict_put(dict, "foo.1", qint_from_int(0));
	481	g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
	482	qdict_put(dict, "foo.0", qint_from_int(0));
	483	g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 2);
	484	qdict_put(dict, "foo.bar", qint_from_int(0));
	485	g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
	486	qdict_del(dict, "foo.bar");
	487
	488	qdict_put(dict, "foo.2.a", qint_from_int(0));
	489	qdict_put(dict, "foo.2.b", qint_from_int(0));
	490	qdict_put(dict, "foo.2.c", qint_from_int(0));
	491	g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 3);
	492	g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
	493
	494	QDECREF(dict);
	495
	496	dict = qdict_new();
	497	qdict_put(dict, "1", qint_from_int(0));
	498	g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
	499	qdict_put(dict, "0", qint_from_int(0));
	500	g_assert_cmpint(qdict_array_entries(dict, ""), ==, 2);
	501	qdict_put(dict, "bar", qint_from_int(0));
	502	g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
	503	qdict_del(dict, "bar");
	504
	505	qdict_put(dict, "2.a", qint_from_int(0));
	506	qdict_put(dict, "2.b", qint_from_int(0));
	507	qdict_put(dict, "2.c", qint_from_int(0));
	508	g_assert_cmpint(qdict_array_entries(dict, ""), ==, 3);
	509
	510	QDECREF(dict);
	511	}
	512
8a5eb36a HR	513	static void qdict_join_test(void)
	514	{
	515	QDict dict1, dict2;
	516	bool overwrite = false;
	517	int i;
	518
	519	dict1 = qdict_new();
	520	dict2 = qdict_new();
	521
	522
	523	/* Test everything once without overwrite and once with */
	524	do
	525	{
	526	/* Test empty dicts */
	527	qdict_join(dict1, dict2, overwrite);
	528
	529	g_assert(qdict_size(dict1) == 0);
	530	g_assert(qdict_size(dict2) == 0);
	531
	532
	533	/* First iteration: Test movement */
	534	/* Second iteration: Test empty source and non-empty destination */
	535	qdict_put(dict2, "foo", qint_from_int(42));
	536
	537	for (i = 0; i < 2; i++) {
	538	qdict_join(dict1, dict2, overwrite);
	539
	540	g_assert(qdict_size(dict1) == 1);
	541	g_assert(qdict_size(dict2) == 0);
	542
	543	g_assert(qdict_get_int(dict1, "foo") == 42);
	544	}
	545
	546
	547	/* Test non-empty source and destination without conflict */
	548	qdict_put(dict2, "bar", qint_from_int(23));
	549
	550	qdict_join(dict1, dict2, overwrite);
	551
	552	g_assert(qdict_size(dict1) == 2);
	553	g_assert(qdict_size(dict2) == 0);
	554
	555	g_assert(qdict_get_int(dict1, "foo") == 42);
	556	g_assert(qdict_get_int(dict1, "bar") == 23);
	557
	558
	559	/* Test conflict */
	560	qdict_put(dict2, "foo", qint_from_int(84));
	561
	562	qdict_join(dict1, dict2, overwrite);
	563
	564	g_assert(qdict_size(dict1) == 2);
	565	g_assert(qdict_size(dict2) == !overwrite);
	566
	567	g_assert(qdict_get_int(dict1, "foo") == overwrite ? 84 : 42);
	568	g_assert(qdict_get_int(dict1, "bar") == 23);
	569
	570	if (!overwrite) {
	571	g_assert(qdict_get_int(dict2, "foo") == 84);
	572	}
	573
	574
	575	/* Check the references */
	576	g_assert(qdict_get(dict1, "foo")->refcnt == 1);
577	g_assert(qdict_get(dict1, "bar")->refcnt == 1);
578
579	if (!overwrite) {
580	g_assert(qdict_get(dict2, "foo")->refcnt == 1);
581	}
582
583
584	/* Clean up */
585	qdict_del(dict1, "foo");
586	qdict_del(dict1, "bar");
587
588	if (!overwrite) {
589	qdict_del(dict2, "foo");
590	}
591	}
592	while (overwrite ^= true);
593
594
595	QDECREF(dict1);
596	QDECREF(dict2);
597	}
598
7b8c51ad LC	599	/*
	600	* Errors test-cases
	601	*/
	602
ac531cb6	603	static void qdict_put_exists_test(void)
7b8c51ad LC	604	{
	605	int value;
	606	const char *key = "exists";
ac531cb6	607	QDict *tests_dict = qdict_new();
7b8c51ad LC	608
	609	qdict_put(tests_dict, key, qint_from_int(1));
	610	qdict_put(tests_dict, key, qint_from_int(2));
	611
	612	value = qdict_get_int(tests_dict, key);
ac531cb6	613	g_assert(value == 2);
29ec3156	614
ac531cb6 AL	615	g_assert(qdict_size(tests_dict) == 1);
	616
	617	QDECREF(tests_dict);
7b8c51ad	618	}
7b8c51ad	619
ac531cb6	620	static void qdict_get_not_exists_test(void)
7b8c51ad	621	{
ac531cb6 AL	622	QDict *tests_dict = qdict_new();
	623	g_assert(qdict_get(tests_dict, "foo") == NULL);
	624
	625	QDECREF(tests_dict);
7b8c51ad	626	}
7b8c51ad LC	627
	628	/*
	629	* Stress test-case
	630	*
	631	* This is a lot big for a unit-test, but there is no other place
	632	* to have it.
	633	*/
	634
	635	static void remove_dots(char *string)
	636	{
	637	char *p = strchr(string, ':');
	638	if (p)
	639	*p = '\0';
	640	}
	641
	642	static QString read_line(FILE file, char *key)
	643	{
	644	char value[128];
	645
7464f058	646	if (fscanf(file, "%127s%127s", key, value) == EOF) {
7b8c51ad	647	return NULL;
7464f058	648	}
7b8c51ad LC	649	remove_dots(key);
	650	return qstring_from_str(value);
	651	}
	652
	653	#define reset_file(file) fseek(file, 0L, SEEK_SET)
	654
ac531cb6	655	static void qdict_stress_test(void)
7b8c51ad LC	656	{
	657	size_t lines;
	658	char key[128];
	659	FILE *test_file;
	660	QDict *qdict;
	661	QString *value;
	662	const char *test_file_path = "qdict-test-data.txt";
	663
	664	test_file = fopen(test_file_path, "r");
ac531cb6	665	g_assert(test_file != NULL);
7b8c51ad LC	666
	667	// Create the dict
	668	qdict = qdict_new();
ac531cb6	669	g_assert(qdict != NULL);
7b8c51ad LC	670
	671	// Add everything from the test file
	672	for (lines = 0;; lines++) {
	673	value = read_line(test_file, key);
	674	if (!value)
	675	break;
	676
	677	qdict_put(qdict, key, value);
	678	}
ac531cb6	679	g_assert(qdict_size(qdict) == lines);
7b8c51ad LC	680
	681	// Check if everything is really in there
	682	reset_file(test_file);
	683	for (;;) {
	684	const char str1, str2;
	685
	686	value = read_line(test_file, key);
	687	if (!value)
	688	break;
	689
	690	str1 = qstring_get_str(value);
	691
	692	str2 = qdict_get_str(qdict, key);
ac531cb6	693	g_assert(str2 != NULL);
7b8c51ad	694
ac531cb6	695	g_assert(strcmp(str1, str2) == 0);
7b8c51ad LC	696
	697	QDECREF(value);
	698	}
	699
	700	// Delete everything
	701	reset_file(test_file);
	702	for (;;) {
	703	value = read_line(test_file, key);
	704	if (!value)
	705	break;
	706
	707	qdict_del(qdict, key);
	708	QDECREF(value);
	709
ac531cb6	710	g_assert(qdict_haskey(qdict, key) == 0);
7b8c51ad LC	711	}
	712	fclose(test_file);
	713
ac531cb6	714	g_assert(qdict_size(qdict) == 0);
7b8c51ad LC	715	QDECREF(qdict);
7b8c51ad LC	716	}
7b8c51ad	717
ac531cb6	718	int main(int argc, char **argv)
7b8c51ad	719	{
ac531cb6	720	g_test_init(&argc, &argv, NULL);
7b8c51ad	721
ac531cb6 AL	722	g_test_add_func("/public/new", qdict_new_test);
	723	g_test_add_func("/public/put_obj", qdict_put_obj_test);
	724	g_test_add_func("/public/destroy_simple", qdict_destroy_simple_test);
7b8c51ad LC	725
7b8c51ad LC	726	/* Continue, but now with fixtures */
ac531cb6 AL	727	g_test_add_func("/public/get", qdict_get_test);
	728	g_test_add_func("/public/get_int", qdict_get_int_test);
	729	g_test_add_func("/public/get_try_int", qdict_get_try_int_test);
	730	g_test_add_func("/public/get_str", qdict_get_str_test);
	731	g_test_add_func("/public/get_try_str", qdict_get_try_str_test);
ef1919df	732	g_test_add_func("/public/defaults", qdict_defaults_test);
ac531cb6 AL	733	g_test_add_func("/public/haskey_not", qdict_haskey_not_test);
	734	g_test_add_func("/public/haskey", qdict_haskey_test);
	735	g_test_add_func("/public/del", qdict_del_test);
	736	g_test_add_func("/public/to_qdict", qobject_to_qdict_test);
	737	g_test_add_func("/public/iterapi", qdict_iterapi_test);
3fb11779	738	g_test_add_func("/public/flatten", qdict_flatten_test);
be331341	739	g_test_add_func("/public/array_split", qdict_array_split_test);
ef1919df	740	g_test_add_func("/public/array_entries", qdict_array_entries_test);
8a5eb36a	741	g_test_add_func("/public/join", qdict_join_test);
ac531cb6 AL	742
	743	g_test_add_func("/errors/put_exists", qdict_put_exists_test);
	744	g_test_add_func("/errors/get_not_exists", qdict_get_not_exists_test);
7b8c51ad LC	745
7b8c51ad LC	746	/* The Big one */
ac531cb6 AL	747	if (g_test_slow()) {
	748	g_test_add_func("/stress/test", qdict_stress_test);
	749	}
7b8c51ad	750
ac531cb6	751	return g_test_run();
7b8c51ad	752	}