check-qdict.c 25.2 KB
Newer Older
L
Luiz Capitulino 已提交
1 2 3 4 5 6 7
/*
 * QDict unit-tests.
 *
 * Copyright (C) 2009 Red Hat Inc.
 *
 * Authors:
 *  Luiz Capitulino <lcapitulino@redhat.com>
L
Luiz Capitulino 已提交
8 9 10
 *
 * 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.
L
Luiz Capitulino 已提交
11 12
 */

13
#include "qemu/osdep.h"
14
#include "qapi/qmp/qdict.h"
15
#include "qapi/qmp/qlist.h"
16
#include "qapi/qmp/qnum.h"
17
#include "qapi/qmp/qstring.h"
18
#include "qapi/error.h"
L
Luiz Capitulino 已提交
19 20 21 22 23 24 25 26
#include "qemu-common.h"

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

A
Anthony Liguori 已提交
27
static void qdict_new_test(void)
L
Luiz Capitulino 已提交
28 29 30 31
{
    QDict *qdict;

    qdict = qdict_new();
A
Anthony Liguori 已提交
32 33 34 35
    g_assert(qdict != NULL);
    g_assert(qdict_size(qdict) == 0);
    g_assert(qdict->base.refcnt == 1);
    g_assert(qobject_type(QOBJECT(qdict)) == QTYPE_QDICT);
L
Luiz Capitulino 已提交
36

37
    QDECREF(qdict);
L
Luiz Capitulino 已提交
38 39
}

A
Anthony Liguori 已提交
40
static void qdict_put_obj_test(void)
L
Luiz Capitulino 已提交
41
{
42
    QNum *qn;
L
Luiz Capitulino 已提交
43 44 45 46 47 48 49
    QDict *qdict;
    QDictEntry *ent;
    const int num = 42;

    qdict = qdict_new();

    // key "" will have tdb hash 12345
50
    qdict_put_int(qdict, "", num);
L
Luiz Capitulino 已提交
51

A
Anthony Liguori 已提交
52
    g_assert(qdict_size(qdict) == 1);
53
    ent = QLIST_FIRST(&qdict->table[12345 % QDICT_BUCKET_MAX]);
54
    qn = qobject_to(QNum, ent->value);
55
    g_assert_cmpint(qnum_get_int(qn), ==, num);
L
Luiz Capitulino 已提交
56

57
    QDECREF(qdict);
L
Luiz Capitulino 已提交
58 59
}

A
Anthony Liguori 已提交
60
static void qdict_destroy_simple_test(void)
L
Luiz Capitulino 已提交
61 62 63 64
{
    QDict *qdict;

    qdict = qdict_new();
65 66
    qdict_put_int(qdict, "num", 0);
    qdict_put_str(qdict, "str", "foo");
L
Luiz Capitulino 已提交
67 68 69 70

    QDECREF(qdict);
}

A
Anthony Liguori 已提交
71
static void qdict_get_test(void)
L
Luiz Capitulino 已提交
72
{
73
    QNum *qn;
L
Luiz Capitulino 已提交
74 75 76
    QObject *obj;
    const int value = -42;
    const char *key = "test";
A
Anthony Liguori 已提交
77
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
78

79
    qdict_put_int(tests_dict, key, value);
L
Luiz Capitulino 已提交
80 81

    obj = qdict_get(tests_dict, key);
A
Anthony Liguori 已提交
82
    g_assert(obj != NULL);
L
Luiz Capitulino 已提交
83

84
    qn = qobject_to(QNum, obj);
85
    g_assert_cmpint(qnum_get_int(qn), ==, value);
A
Anthony Liguori 已提交
86 87

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
88 89
}

A
Anthony Liguori 已提交
90
static void qdict_get_int_test(void)
L
Luiz Capitulino 已提交
91 92 93 94
{
    int ret;
    const int value = 100;
    const char *key = "int";
A
Anthony Liguori 已提交
95
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
96

97
    qdict_put_int(tests_dict, key, value);
L
Luiz Capitulino 已提交
98 99

    ret = qdict_get_int(tests_dict, key);
A
Anthony Liguori 已提交
100 101 102
    g_assert(ret == value);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
103 104
}

A
Anthony Liguori 已提交
105
static void qdict_get_try_int_test(void)
L
Luiz Capitulino 已提交
106 107 108 109
{
    int ret;
    const int value = 100;
    const char *key = "int";
A
Anthony Liguori 已提交
110
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
111

112
    qdict_put_int(tests_dict, key, value);
113
    qdict_put_str(tests_dict, "string", "test");
L
Luiz Capitulino 已提交
114 115

    ret = qdict_get_try_int(tests_dict, key, 0);
A
Anthony Liguori 已提交
116 117
    g_assert(ret == value);

118 119 120 121 122 123
    ret = qdict_get_try_int(tests_dict, "missing", -42);
    g_assert_cmpuint(ret, ==, -42);

    ret = qdict_get_try_int(tests_dict, "string", -42);
    g_assert_cmpuint(ret, ==, -42);

A
Anthony Liguori 已提交
124
    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
125 126
}

A
Anthony Liguori 已提交
127
static void qdict_get_str_test(void)
L
Luiz Capitulino 已提交
128 129 130 131
{
    const char *p;
    const char *key = "key";
    const char *str = "string";
A
Anthony Liguori 已提交
132
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
133

134
    qdict_put_str(tests_dict, key, str);
L
Luiz Capitulino 已提交
135 136

    p = qdict_get_str(tests_dict, key);
A
Anthony Liguori 已提交
137 138 139 140
    g_assert(p != NULL);
    g_assert(strcmp(p, str) == 0);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
141 142
}

A
Anthony Liguori 已提交
143
static void qdict_get_try_str_test(void)
L
Luiz Capitulino 已提交
144 145 146 147
{
    const char *p;
    const char *key = "key";
    const char *str = "string";
A
Anthony Liguori 已提交
148
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
149

150
    qdict_put_str(tests_dict, key, str);
L
Luiz Capitulino 已提交
151 152

    p = qdict_get_try_str(tests_dict, key);
A
Anthony Liguori 已提交
153 154 155 156
    g_assert(p != NULL);
    g_assert(strcmp(p, str) == 0);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
157 158
}

159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180
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);
}

A
Anthony Liguori 已提交
181
static void qdict_haskey_not_test(void)
L
Luiz Capitulino 已提交
182
{
A
Anthony Liguori 已提交
183 184 185 186
    QDict *tests_dict = qdict_new();
    g_assert(qdict_haskey(tests_dict, "test") == 0);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
187 188
}

A
Anthony Liguori 已提交
189
static void qdict_haskey_test(void)
L
Luiz Capitulino 已提交
190 191
{
    const char *key = "test";
A
Anthony Liguori 已提交
192
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
193

194
    qdict_put_int(tests_dict, key, 0);
A
Anthony Liguori 已提交
195 196 197
    g_assert(qdict_haskey(tests_dict, key) == 1);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
198 199
}

A
Anthony Liguori 已提交
200
static void qdict_del_test(void)
L
Luiz Capitulino 已提交
201 202
{
    const char *key = "key test";
A
Anthony Liguori 已提交
203
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
204

205
    qdict_put_str(tests_dict, key, "foo");
A
Anthony Liguori 已提交
206
    g_assert(qdict_size(tests_dict) == 1);
L
Luiz Capitulino 已提交
207 208 209

    qdict_del(tests_dict, key);

A
Anthony Liguori 已提交
210 211 212 213
    g_assert(qdict_size(tests_dict) == 0);
    g_assert(qdict_haskey(tests_dict, key) == 0);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
214 215
}

A
Anthony Liguori 已提交
216
static void qobject_to_qdict_test(void)
L
Luiz Capitulino 已提交
217
{
A
Anthony Liguori 已提交
218
    QDict *tests_dict = qdict_new();
219
    g_assert(qobject_to(QDict, QOBJECT(tests_dict)) == tests_dict);
A
Anthony Liguori 已提交
220 221

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
222 223
}

A
Anthony Liguori 已提交
224
static void qdict_iterapi_test(void)
225 226 227
{
    int count;
    const QDictEntry *ent;
A
Anthony Liguori 已提交
228
    QDict *tests_dict = qdict_new();
229

A
Anthony Liguori 已提交
230
    g_assert(qdict_first(tests_dict) == NULL);
231

232 233 234
    qdict_put_int(tests_dict, "key1", 1);
    qdict_put_int(tests_dict, "key2", 2);
    qdict_put_int(tests_dict, "key3", 3);
235 236 237

    count = 0;
    for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
A
Anthony Liguori 已提交
238
        g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
239 240 241
        count++;
    }

A
Anthony Liguori 已提交
242
    g_assert(count == qdict_size(tests_dict));
243 244 245 246

    /* Do it again to test restarting */
    count = 0;
    for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
A
Anthony Liguori 已提交
247
        g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
248 249 250
        count++;
    }

A
Anthony Liguori 已提交
251 252 253
    g_assert(count == qdict_size(tests_dict));

    QDECREF(tests_dict);
254 255
}

M
Max Reitz 已提交
256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299
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
     * }
     */

300 301
    qdict_put_int(dict1, "a", 0);
    qdict_put_int(dict1, "b", 1);
M
Max Reitz 已提交
302

303 304
    qlist_append_int(list1, 23);
    qlist_append_int(list1, 66);
E
Eric Blake 已提交
305
    qlist_append(list1, dict1);
306
    qlist_append_int(list2, 42);
E
Eric Blake 已提交
307
    qlist_append(list2, list1);
M
Max Reitz 已提交
308

309 310
    qdict_put_int(dict2, "c", 2);
    qdict_put_int(dict2, "d", 3);
E
Eric Blake 已提交
311 312
    qdict_put(dict3, "e", list2);
    qdict_put(dict3, "f", dict2);
313
    qdict_put_int(dict3, "g", 4);
M
Max Reitz 已提交
314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330

    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);
}

331 332 333 334
static void qdict_array_split_test(void)
{
    QDict *test_dict = qdict_new();
    QDict *dict1, *dict2;
335
    QNum *int1;
336 337 338 339 340 341 342
    QList *test_list;

    /*
     * Test the split of
     *
     * {
     *     "1.x": 0,
343
     *     "4.y": 1,
344 345
     *     "0.a": 42,
     *     "o.o": 7,
346 347
     *     "0.b": 23,
     *     "2": 66
348 349 350 351 352 353 354 355 356 357 358
     * }
     *
     * to
     *
     * [
     *     {
     *         "a": 42,
     *         "b": 23
     *     },
     *     {
     *         "x": 0
359 360
     *     },
     *     66
361 362 363 364 365
     * ]
     *
     * and
     *
     * {
366
     *     "4.y": 1,
367 368 369 370 371 372 373 374
     *     "o.o": 7
     * }
     *
     * (remaining in the old QDict)
     *
     * This example is given in the comment of qdict_array_split().
     */

375 376 377 378 379 380
    qdict_put_int(test_dict, "1.x", 0);
    qdict_put_int(test_dict, "4.y", 1);
    qdict_put_int(test_dict, "0.a", 42);
    qdict_put_int(test_dict, "o.o", 7);
    qdict_put_int(test_dict, "0.b", 23);
    qdict_put_int(test_dict, "2", 66);
381 382 383

    qdict_array_split(test_dict, &test_list);

384 385 386
    dict1 = qobject_to(QDict, qlist_pop(test_list));
    dict2 = qobject_to(QDict, qlist_pop(test_list));
    int1 = qobject_to(QNum, qlist_pop(test_list));
387 388 389

    g_assert(dict1);
    g_assert(dict2);
390
    g_assert(int1);
391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407
    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);

408
    g_assert_cmpint(qnum_get_int(int1), ==, 66);
409 410 411 412

    QDECREF(int1);

    g_assert(qdict_get_int(test_dict, "4.y") == 1);
413 414 415 416 417
    g_assert(qdict_get_int(test_dict, "o.o") == 7);

    g_assert(qdict_size(test_dict) == 2);

    QDECREF(test_dict);
418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446

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

447 448 449
    qdict_put_int(test_dict, "0", 42);
    qdict_put_int(test_dict, "1", 23);
    qdict_put_int(test_dict, "1.x", 84);
450 451 452

    qdict_array_split(test_dict, &test_list);

453
    int1 = qobject_to(QNum, qlist_pop(test_list));
454 455 456 457 458 459

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

    QDECREF(test_list);

460
    g_assert_cmpint(qnum_get_int(int1), ==, 42);
461 462 463 464 465 466 467 468 469

    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);
470 471
}

472 473 474 475 476 477
static void qdict_array_entries_test(void)
{
    QDict *dict = qdict_new();

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

478 479
    qdict_put_int(dict, "bar", 0);
    qdict_put_int(dict, "baz.0", 0);
480 481
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);

482
    qdict_put_int(dict, "foo.1", 0);
483
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
484
    qdict_put_int(dict, "foo.0", 0);
485
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 2);
486
    qdict_put_int(dict, "foo.bar", 0);
487 488 489
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
    qdict_del(dict, "foo.bar");

490 491 492
    qdict_put_int(dict, "foo.2.a", 0);
    qdict_put_int(dict, "foo.2.b", 0);
    qdict_put_int(dict, "foo.2.c", 0);
493 494 495 496 497 498
    g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 3);
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);

    QDECREF(dict);

    dict = qdict_new();
499
    qdict_put_int(dict, "1", 0);
500
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
501
    qdict_put_int(dict, "0", 0);
502
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, 2);
503
    qdict_put_int(dict, "bar", 0);
504 505 506
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
    qdict_del(dict, "bar");

507 508 509
    qdict_put_int(dict, "2.a", 0);
    qdict_put_int(dict, "2.b", 0);
    qdict_put_int(dict, "2.c", 0);
510 511 512 513 514
    g_assert_cmpint(qdict_array_entries(dict, ""), ==, 3);

    QDECREF(dict);
}

515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534
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 */
535
        qdict_put_int(dict2, "foo", 42);
536 537 538 539 540 541 542 543 544 545 546

        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 */
547
        qdict_put_int(dict2, "bar", 23);
548 549 550 551 552 553 554 555 556 557

        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 */
558
        qdict_put_int(dict2, "foo", 84);
559 560 561 562 563 564

        qdict_join(dict1, dict2, overwrite);

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

565
        g_assert(qdict_get_int(dict1, "foo") == (overwrite ? 84 : 42));
566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593
        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);
}

594 595 596 597 598 599
static void qdict_crumple_test_recursive(void)
{
    QDict *src, *dst, *rule, *vnc, *acl, *listen;
    QList *rules;

    src = qdict_new();
600 601 602 603 604 605 606 607 608
    qdict_put_str(src, "vnc.listen.addr", "127.0.0.1");
    qdict_put_str(src, "vnc.listen.port", "5901");
    qdict_put_str(src, "vnc.acl.rules.0.match", "fred");
    qdict_put_str(src, "vnc.acl.rules.0.policy", "allow");
    qdict_put_str(src, "vnc.acl.rules.1.match", "bob");
    qdict_put_str(src, "vnc.acl.rules.1.policy", "deny");
    qdict_put_str(src, "vnc.acl.default", "deny");
    qdict_put_str(src, "vnc.acl..name", "acl0");
    qdict_put_str(src, "vnc.acl.rule..name", "acl0");
609

610
    dst = qobject_to(QDict, qdict_crumple(src, &error_abort));
611
    g_assert(dst);
612 613
    g_assert_cmpint(qdict_size(dst), ==, 1);

614 615
    vnc = qdict_get_qdict(dst, "vnc");
    g_assert(vnc);
616
    g_assert_cmpint(qdict_size(vnc), ==, 3);
617

618 619
    listen = qdict_get_qdict(vnc, "listen");
    g_assert(listen);
620
    g_assert_cmpint(qdict_size(listen), ==, 2);
621 622 623
    g_assert_cmpstr("127.0.0.1", ==, qdict_get_str(listen, "addr"));
    g_assert_cmpstr("5901", ==, qdict_get_str(listen, "port"));

624 625
    acl = qdict_get_qdict(vnc, "acl");
    g_assert(acl);
626
    g_assert_cmpint(qdict_size(acl), ==, 3);
627

628 629
    rules = qdict_get_qlist(acl, "rules");
    g_assert(rules);
630 631
    g_assert_cmpint(qlist_size(rules), ==, 2);

632
    rule = qobject_to(QDict, qlist_pop(rules));
633
    g_assert(rule);
634 635 636 637 638
    g_assert_cmpint(qdict_size(rule), ==, 2);
    g_assert_cmpstr("fred", ==, qdict_get_str(rule, "match"));
    g_assert_cmpstr("allow", ==, qdict_get_str(rule, "policy"));
    QDECREF(rule);

639
    rule = qobject_to(QDict, qlist_pop(rules));
640
    g_assert(rule);
641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659
    g_assert_cmpint(qdict_size(rule), ==, 2);
    g_assert_cmpstr("bob", ==, qdict_get_str(rule, "match"));
    g_assert_cmpstr("deny", ==, qdict_get_str(rule, "policy"));
    QDECREF(rule);

    /* With recursive crumpling, we should see all names unescaped */
    g_assert_cmpstr("acl0", ==, qdict_get_str(vnc, "acl.name"));
    g_assert_cmpstr("acl0", ==, qdict_get_str(acl, "rule.name"));

    QDECREF(src);
    QDECREF(dst);
}

static void qdict_crumple_test_empty(void)
{
    QDict *src, *dst;

    src = qdict_new();

660
    dst = qobject_to(QDict, qdict_crumple(src, &error_abort));
661 662 663 664 665 666 667

    g_assert_cmpint(qdict_size(dst), ==, 0);

    QDECREF(src);
    QDECREF(dst);
}

668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794
static int qdict_count_entries(QDict *dict)
{
    const QDictEntry *e;
    int count = 0;

    for (e = qdict_first(dict); e; e = qdict_next(dict, e)) {
        count++;
    }

    return count;
}

static void qdict_rename_keys_test(void)
{
    QDict *dict = qdict_new();
    QDict *copy;
    QDictRenames *renames;
    Error *local_err = NULL;

    qdict_put_str(dict, "abc", "foo");
    qdict_put_str(dict, "abcdef", "bar");
    qdict_put_int(dict, "number", 42);
    qdict_put_bool(dict, "flag", true);
    qdict_put_null(dict, "nothing");

    /* Empty rename list */
    renames = (QDictRenames[]) {
        { NULL, "this can be anything" }
    };
    copy = qdict_clone_shallow(dict);
    qdict_rename_keys(copy, renames, &error_abort);

    g_assert_cmpstr(qdict_get_str(copy, "abc"), ==, "foo");
    g_assert_cmpstr(qdict_get_str(copy, "abcdef"), ==, "bar");
    g_assert_cmpint(qdict_get_int(copy, "number"), ==, 42);
    g_assert_cmpint(qdict_get_bool(copy, "flag"), ==, true);
    g_assert(qobject_type(qdict_get(copy, "nothing")) == QTYPE_QNULL);
    g_assert_cmpint(qdict_count_entries(copy), ==, 5);

    QDECREF(copy);

    /* Simple rename of all entries */
    renames = (QDictRenames[]) {
        { "abc",        "str1" },
        { "abcdef",     "str2" },
        { "number",     "int" },
        { "flag",       "bool" },
        { "nothing",    "null" },
        { NULL , NULL }
    };
    copy = qdict_clone_shallow(dict);
    qdict_rename_keys(copy, renames, &error_abort);

    g_assert(!qdict_haskey(copy, "abc"));
    g_assert(!qdict_haskey(copy, "abcdef"));
    g_assert(!qdict_haskey(copy, "number"));
    g_assert(!qdict_haskey(copy, "flag"));
    g_assert(!qdict_haskey(copy, "nothing"));

    g_assert_cmpstr(qdict_get_str(copy, "str1"), ==, "foo");
    g_assert_cmpstr(qdict_get_str(copy, "str2"), ==, "bar");
    g_assert_cmpint(qdict_get_int(copy, "int"), ==, 42);
    g_assert_cmpint(qdict_get_bool(copy, "bool"), ==, true);
    g_assert(qobject_type(qdict_get(copy, "null")) == QTYPE_QNULL);
    g_assert_cmpint(qdict_count_entries(copy), ==, 5);

    QDECREF(copy);

    /* Renames are processed top to bottom */
    renames = (QDictRenames[]) {
        { "abc",        "tmp" },
        { "abcdef",     "abc" },
        { "number",     "abcdef" },
        { "flag",       "number" },
        { "nothing",    "flag" },
        { "tmp",        "nothing" },
        { NULL , NULL }
    };
    copy = qdict_clone_shallow(dict);
    qdict_rename_keys(copy, renames, &error_abort);

    g_assert_cmpstr(qdict_get_str(copy, "nothing"), ==, "foo");
    g_assert_cmpstr(qdict_get_str(copy, "abc"), ==, "bar");
    g_assert_cmpint(qdict_get_int(copy, "abcdef"), ==, 42);
    g_assert_cmpint(qdict_get_bool(copy, "number"), ==, true);
    g_assert(qobject_type(qdict_get(copy, "flag")) == QTYPE_QNULL);
    g_assert(!qdict_haskey(copy, "tmp"));
    g_assert_cmpint(qdict_count_entries(copy), ==, 5);

    QDECREF(copy);

    /* Conflicting rename */
    renames = (QDictRenames[]) {
        { "abcdef",     "abc" },
        { NULL , NULL }
    };
    copy = qdict_clone_shallow(dict);
    qdict_rename_keys(copy, renames, &local_err);

    g_assert(local_err != NULL);
    error_free(local_err);
    local_err = NULL;

    g_assert_cmpstr(qdict_get_str(copy, "abc"), ==, "foo");
    g_assert_cmpstr(qdict_get_str(copy, "abcdef"), ==, "bar");
    g_assert_cmpint(qdict_get_int(copy, "number"), ==, 42);
    g_assert_cmpint(qdict_get_bool(copy, "flag"), ==, true);
    g_assert(qobject_type(qdict_get(copy, "nothing")) == QTYPE_QNULL);
    g_assert_cmpint(qdict_count_entries(copy), ==, 5);

    QDECREF(copy);

    /* Renames in an empty dict */
    renames = (QDictRenames[]) {
        { "abcdef",     "abc" },
        { NULL , NULL }
    };

    QDECREF(dict);
    dict = qdict_new();

    qdict_rename_keys(dict, renames, &error_abort);
    g_assert(qdict_first(dict) == NULL);

    QDECREF(dict);
}

795 796 797 798 799 800 801
static void qdict_crumple_test_bad_inputs(void)
{
    QDict *src;
    Error *error = NULL;

    src = qdict_new();
    /* rule.0 can't be both a string and a dict */
802 803
    qdict_put_str(src, "rule.0", "fred");
    qdict_put_str(src, "rule.0.policy", "allow");
804 805 806 807 808 809 810 811 812

    g_assert(qdict_crumple(src, &error) == NULL);
    g_assert(error != NULL);
    error_free(error);
    error = NULL;
    QDECREF(src);

    src = qdict_new();
    /* rule can't be both a list and a dict */
813 814
    qdict_put_str(src, "rule.0", "fred");
    qdict_put_str(src, "rule.a", "allow");
815 816 817 818 819 820 821 822 823 824

    g_assert(qdict_crumple(src, &error) == NULL);
    g_assert(error != NULL);
    error_free(error);
    error = NULL;
    QDECREF(src);

    src = qdict_new();
    /* The input should be flat, ie no dicts or lists */
    qdict_put(src, "rule.a", qdict_new());
825
    qdict_put_str(src, "rule.b", "allow");
826 827 828 829 830 831 832 833 834

    g_assert(qdict_crumple(src, &error) == NULL);
    g_assert(error != NULL);
    error_free(error);
    error = NULL;
    QDECREF(src);

    src = qdict_new();
    /* List indexes must not have gaps */
835 836
    qdict_put_str(src, "rule.0", "deny");
    qdict_put_str(src, "rule.3", "allow");
837 838 839 840 841 842 843 844 845

    g_assert(qdict_crumple(src, &error) == NULL);
    g_assert(error != NULL);
    error_free(error);
    error = NULL;
    QDECREF(src);

    src = qdict_new();
    /* List indexes must be in %zu format */
846 847
    qdict_put_str(src, "rule.0", "deny");
    qdict_put_str(src, "rule.+1", "allow");
848 849 850 851 852 853 854 855

    g_assert(qdict_crumple(src, &error) == NULL);
    g_assert(error != NULL);
    error_free(error);
    error = NULL;
    QDECREF(src);
}

L
Luiz Capitulino 已提交
856 857 858 859
/*
 * Errors test-cases
 */

A
Anthony Liguori 已提交
860
static void qdict_put_exists_test(void)
L
Luiz Capitulino 已提交
861 862 863
{
    int value;
    const char *key = "exists";
A
Anthony Liguori 已提交
864
    QDict *tests_dict = qdict_new();
L
Luiz Capitulino 已提交
865

866 867
    qdict_put_int(tests_dict, key, 1);
    qdict_put_int(tests_dict, key, 2);
L
Luiz Capitulino 已提交
868 869

    value = qdict_get_int(tests_dict, key);
A
Anthony Liguori 已提交
870
    g_assert(value == 2);
L
Luiz Capitulino 已提交
871

A
Anthony Liguori 已提交
872 873 874
    g_assert(qdict_size(tests_dict) == 1);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
875 876
}

A
Anthony Liguori 已提交
877
static void qdict_get_not_exists_test(void)
L
Luiz Capitulino 已提交
878
{
A
Anthony Liguori 已提交
879 880 881 882
    QDict *tests_dict = qdict_new();
    g_assert(qdict_get(tests_dict, "foo") == NULL);

    QDECREF(tests_dict);
L
Luiz Capitulino 已提交
883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902
}

/*
 * 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];

S
Stefan Weil 已提交
903
    if (fscanf(file, "%127s%127s", key, value) == EOF) {
L
Luiz Capitulino 已提交
904
        return NULL;
S
Stefan Weil 已提交
905
    }
L
Luiz Capitulino 已提交
906 907 908 909 910 911
    remove_dots(key);
    return qstring_from_str(value);
}

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

A
Anthony Liguori 已提交
912
static void qdict_stress_test(void)
L
Luiz Capitulino 已提交
913 914 915 916 917 918 919 920 921
{
    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");
A
Anthony Liguori 已提交
922
    g_assert(test_file != NULL);
L
Luiz Capitulino 已提交
923 924 925

    // Create the dict
    qdict = qdict_new();
A
Anthony Liguori 已提交
926
    g_assert(qdict != NULL);
L
Luiz Capitulino 已提交
927 928 929 930 931 932 933 934 935

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

        qdict_put(qdict, key, value);
    }
A
Anthony Liguori 已提交
936
    g_assert(qdict_size(qdict) == lines);
L
Luiz Capitulino 已提交
937 938 939 940 941 942 943 944 945 946 947 948 949

    // 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);
A
Anthony Liguori 已提交
950
        g_assert(str2 != NULL);
L
Luiz Capitulino 已提交
951

A
Anthony Liguori 已提交
952
        g_assert(strcmp(str1, str2) == 0);
L
Luiz Capitulino 已提交
953 954 955 956 957 958 959 960 961 962 963 964 965 966

        QDECREF(value);
    }

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

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

A
Anthony Liguori 已提交
967
        g_assert(qdict_haskey(qdict, key) == 0);
L
Luiz Capitulino 已提交
968 969 970
    }
    fclose(test_file);

A
Anthony Liguori 已提交
971
    g_assert(qdict_size(qdict) == 0);
L
Luiz Capitulino 已提交
972 973 974
    QDECREF(qdict);
}

A
Anthony Liguori 已提交
975
int main(int argc, char **argv)
L
Luiz Capitulino 已提交
976
{
A
Anthony Liguori 已提交
977
    g_test_init(&argc, &argv, NULL);
L
Luiz Capitulino 已提交
978

A
Anthony Liguori 已提交
979 980 981
    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);
L
Luiz Capitulino 已提交
982 983

    /* Continue, but now with fixtures */
A
Anthony Liguori 已提交
984 985 986 987 988
    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);
989
    g_test_add_func("/public/defaults", qdict_defaults_test);
A
Anthony Liguori 已提交
990 991 992 993 994
    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);
M
Max Reitz 已提交
995
    g_test_add_func("/public/flatten", qdict_flatten_test);
996
    g_test_add_func("/public/array_split", qdict_array_split_test);
997
    g_test_add_func("/public/array_entries", qdict_array_entries_test);
998
    g_test_add_func("/public/join", qdict_join_test);
A
Anthony Liguori 已提交
999 1000 1001

    g_test_add_func("/errors/put_exists", qdict_put_exists_test);
    g_test_add_func("/errors/get_not_exists", qdict_get_not_exists_test);
L
Luiz Capitulino 已提交
1002

1003 1004 1005 1006 1007 1008 1009
    g_test_add_func("/public/crumple/recursive",
                    qdict_crumple_test_recursive);
    g_test_add_func("/public/crumple/empty",
                    qdict_crumple_test_empty);
    g_test_add_func("/public/crumple/bad_inputs",
                    qdict_crumple_test_bad_inputs);

1010 1011
    g_test_add_func("/public/rename_keys", qdict_rename_keys_test);

L
Luiz Capitulino 已提交
1012
    /* The Big one */
A
Anthony Liguori 已提交
1013 1014 1015
    if (g_test_slow()) {
        g_test_add_func("/stress/test", qdict_stress_test);
    }
L
Luiz Capitulino 已提交
1016

A
Anthony Liguori 已提交
1017
    return g_test_run();
L
Luiz Capitulino 已提交
1018
}