4 * Copyright (C) 2009 Red Hat Inc.
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.
12 #include "qemu/osdep.h"
14 #include "qapi/qmp/qint.h"
15 #include "qapi/qmp/qdict.h"
16 #include "qapi/qmp/qstring.h"
17 #include "qapi/error.h"
18 #include "qemu-common.h"
21 * Public Interface test-cases
23 * (with some violations to access 'private' data)
26 static void qdict_new_test(void)
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);
36 // destroy doesn't exit yet
40 static void qdict_put_obj_test(void)
49 // key "" will have tdb hash 12345
50 qdict_put(qdict, "", qint_from_int(num));
52 g_assert(qdict_size(qdict) == 1);
53 ent = QLIST_FIRST(&qdict->table[12345 % QDICT_BUCKET_MAX]);
54 qi = qobject_to_qint(ent->value);
55 g_assert(qint_get_int(qi) == num);
57 // destroy doesn't exit yet
64 static void qdict_destroy_simple_test(void)
69 qdict_put(qdict, "num", qint_from_int(0));
70 qdict_put(qdict, "str", qstring_from_str("foo"));
75 static void qdict_get_test(void)
79 const int value = -42;
80 const char *key = "test";
81 QDict *tests_dict = qdict_new();
83 qdict_put(tests_dict, key, qint_from_int(value));
85 obj = qdict_get(tests_dict, key);
86 g_assert(obj != NULL);
88 qi = qobject_to_qint(obj);
89 g_assert(qint_get_int(qi) == value);
94 static void qdict_get_int_test(void)
97 const int value = 100;
98 const char *key = "int";
99 QDict *tests_dict = qdict_new();
101 qdict_put(tests_dict, key, qint_from_int(value));
103 ret = qdict_get_int(tests_dict, key);
104 g_assert(ret == value);
109 static void qdict_get_try_int_test(void)
112 const int value = 100;
113 const char *key = "int";
114 QDict *tests_dict = qdict_new();
116 qdict_put(tests_dict, key, qint_from_int(value));
118 ret = qdict_get_try_int(tests_dict, key, 0);
119 g_assert(ret == value);
124 static void qdict_get_str_test(void)
127 const char *key = "key";
128 const char *str = "string";
129 QDict *tests_dict = qdict_new();
131 qdict_put(tests_dict, key, qstring_from_str(str));
133 p = qdict_get_str(tests_dict, key);
135 g_assert(strcmp(p, str) == 0);
140 static void qdict_get_try_str_test(void)
143 const char *key = "key";
144 const char *str = "string";
145 QDict *tests_dict = qdict_new();
147 qdict_put(tests_dict, key, qstring_from_str(str));
149 p = qdict_get_try_str(tests_dict, key);
151 g_assert(strcmp(p, str) == 0);
156 static void qdict_defaults_test(void)
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");
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");
178 static void qdict_haskey_not_test(void)
180 QDict *tests_dict = qdict_new();
181 g_assert(qdict_haskey(tests_dict, "test") == 0);
186 static void qdict_haskey_test(void)
188 const char *key = "test";
189 QDict *tests_dict = qdict_new();
191 qdict_put(tests_dict, key, qint_from_int(0));
192 g_assert(qdict_haskey(tests_dict, key) == 1);
197 static void qdict_del_test(void)
199 const char *key = "key test";
200 QDict *tests_dict = qdict_new();
202 qdict_put(tests_dict, key, qstring_from_str("foo"));
203 g_assert(qdict_size(tests_dict) == 1);
205 qdict_del(tests_dict, key);
207 g_assert(qdict_size(tests_dict) == 0);
208 g_assert(qdict_haskey(tests_dict, key) == 0);
213 static void qobject_to_qdict_test(void)
215 QDict *tests_dict = qdict_new();
216 g_assert(qobject_to_qdict(QOBJECT(tests_dict)) == tests_dict);
221 static void qdict_iterapi_test(void)
224 const QDictEntry *ent;
225 QDict *tests_dict = qdict_new();
227 g_assert(qdict_first(tests_dict) == NULL);
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));
234 for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
235 g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
239 g_assert(count == qdict_size(tests_dict));
241 /* Do it again to test restarting */
243 for (ent = qdict_first(tests_dict); ent; ent = qdict_next(tests_dict, ent)){
244 g_assert(qdict_haskey(tests_dict, qdict_entry_key(ent)) == 1);
248 g_assert(count == qdict_size(tests_dict));
253 static void qdict_flatten_test(void)
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();
262 * Test the flattening of
297 qdict_put(dict1, "a", qint_from_int(0));
298 qdict_put(dict1, "b", qint_from_int(1));
300 qlist_append(list1, qint_from_int(23));
301 qlist_append(list1, qint_from_int(66));
302 qlist_append(list1, dict1);
303 qlist_append(list2, qint_from_int(42));
304 qlist_append(list2, list1);
306 qdict_put(dict2, "c", qint_from_int(2));
307 qdict_put(dict2, "d", qint_from_int(3));
308 qdict_put(dict3, "e", list2);
309 qdict_put(dict3, "f", dict2);
310 qdict_put(dict3, "g", qint_from_int(4));
312 qdict_flatten(dict3);
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);
323 g_assert(qdict_size(dict3) == 8);
328 static void qdict_array_split_test(void)
330 QDict *test_dict = qdict_new();
331 QDict *dict1, *dict2;
367 * (remaining in the old QDict)
369 * This example is given in the comment of qdict_array_split().
372 qdict_put(test_dict, "1.x", qint_from_int(0));
373 qdict_put(test_dict, "4.y", qint_from_int(1));
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));
377 qdict_put(test_dict, "2", qint_from_int(66));
379 qdict_array_split(test_dict, &test_list);
381 dict1 = qobject_to_qdict(qlist_pop(test_list));
382 dict2 = qobject_to_qdict(qlist_pop(test_list));
383 int1 = qobject_to_qint(qlist_pop(test_list));
388 g_assert(qlist_empty(test_list));
392 g_assert(qdict_get_int(dict1, "a") == 42);
393 g_assert(qdict_get_int(dict1, "b") == 23);
395 g_assert(qdict_size(dict1) == 2);
399 g_assert(qdict_get_int(dict2, "x") == 0);
401 g_assert(qdict_size(dict2) == 1);
405 g_assert(qint_get_int(int1) == 66);
409 g_assert(qdict_get_int(test_dict, "4.y") == 1);
410 g_assert(qdict_get_int(test_dict, "o.o") == 7);
412 g_assert(qdict_size(test_dict) == 2);
438 * That is, test whether splitting stops if there is both an entry with key
439 * of "%u" and other entries with keys prefixed "%u." for the same index.
442 test_dict = qdict_new();
444 qdict_put(test_dict, "0", qint_from_int(42));
445 qdict_put(test_dict, "1", qint_from_int(23));
446 qdict_put(test_dict, "1.x", qint_from_int(84));
448 qdict_array_split(test_dict, &test_list);
450 int1 = qobject_to_qint(qlist_pop(test_list));
453 g_assert(qlist_empty(test_list));
457 g_assert(qint_get_int(int1) == 42);
461 g_assert(qdict_get_int(test_dict, "1") == 23);
462 g_assert(qdict_get_int(test_dict, "1.x") == 84);
464 g_assert(qdict_size(test_dict) == 2);
469 static void qdict_array_entries_test(void)
471 QDict *dict = qdict_new();
473 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);
475 qdict_put(dict, "bar", qint_from_int(0));
476 qdict_put(dict, "baz.0", qint_from_int(0));
477 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 0);
479 qdict_put(dict, "foo.1", qint_from_int(0));
480 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
481 qdict_put(dict, "foo.0", qint_from_int(0));
482 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 2);
483 qdict_put(dict, "foo.bar", qint_from_int(0));
484 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, -EINVAL);
485 qdict_del(dict, "foo.bar");
487 qdict_put(dict, "foo.2.a", qint_from_int(0));
488 qdict_put(dict, "foo.2.b", qint_from_int(0));
489 qdict_put(dict, "foo.2.c", qint_from_int(0));
490 g_assert_cmpint(qdict_array_entries(dict, "foo."), ==, 3);
491 g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
496 qdict_put(dict, "1", qint_from_int(0));
497 g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
498 qdict_put(dict, "0", qint_from_int(0));
499 g_assert_cmpint(qdict_array_entries(dict, ""), ==, 2);
500 qdict_put(dict, "bar", qint_from_int(0));
501 g_assert_cmpint(qdict_array_entries(dict, ""), ==, -EINVAL);
502 qdict_del(dict, "bar");
504 qdict_put(dict, "2.a", qint_from_int(0));
505 qdict_put(dict, "2.b", qint_from_int(0));
506 qdict_put(dict, "2.c", qint_from_int(0));
507 g_assert_cmpint(qdict_array_entries(dict, ""), ==, 3);
512 static void qdict_join_test(void)
514 QDict *dict1, *dict2;
515 bool overwrite = false;
521 /* Test everything once without overwrite and once with */
524 /* Test empty dicts */
525 qdict_join(dict1, dict2, overwrite);
527 g_assert(qdict_size(dict1) == 0);
528 g_assert(qdict_size(dict2) == 0);
530 /* First iteration: Test movement */
531 /* Second iteration: Test empty source and non-empty destination */
532 qdict_put(dict2, "foo", qint_from_int(42));
534 for (i = 0; i < 2; i++) {
535 qdict_join(dict1, dict2, overwrite);
537 g_assert(qdict_size(dict1) == 1);
538 g_assert(qdict_size(dict2) == 0);
540 g_assert(qdict_get_int(dict1, "foo") == 42);
543 /* Test non-empty source and destination without conflict */
544 qdict_put(dict2, "bar", qint_from_int(23));
546 qdict_join(dict1, dict2, overwrite);
548 g_assert(qdict_size(dict1) == 2);
549 g_assert(qdict_size(dict2) == 0);
551 g_assert(qdict_get_int(dict1, "foo") == 42);
552 g_assert(qdict_get_int(dict1, "bar") == 23);
555 qdict_put(dict2, "foo", qint_from_int(84));
557 qdict_join(dict1, dict2, overwrite);
559 g_assert(qdict_size(dict1) == 2);
560 g_assert(qdict_size(dict2) == !overwrite);
562 g_assert(qdict_get_int(dict1, "foo") == (overwrite ? 84 : 42));
563 g_assert(qdict_get_int(dict1, "bar") == 23);
566 g_assert(qdict_get_int(dict2, "foo") == 84);
569 /* Check the references */
570 g_assert(qdict_get(dict1, "foo")->refcnt == 1);
571 g_assert(qdict_get(dict1, "bar")->refcnt == 1);
574 g_assert(qdict_get(dict2, "foo")->refcnt == 1);
578 qdict_del(dict1, "foo");
579 qdict_del(dict1, "bar");
582 qdict_del(dict2, "foo");
585 while (overwrite ^= true);
591 static void qdict_crumple_test_recursive(void)
593 QDict *src, *dst, *rule, *vnc, *acl, *listen;
597 qdict_put(src, "vnc.listen.addr", qstring_from_str("127.0.0.1"));
598 qdict_put(src, "vnc.listen.port", qstring_from_str("5901"));
599 qdict_put(src, "vnc.acl.rules.0.match", qstring_from_str("fred"));
600 qdict_put(src, "vnc.acl.rules.0.policy", qstring_from_str("allow"));
601 qdict_put(src, "vnc.acl.rules.1.match", qstring_from_str("bob"));
602 qdict_put(src, "vnc.acl.rules.1.policy", qstring_from_str("deny"));
603 qdict_put(src, "vnc.acl.default", qstring_from_str("deny"));
604 qdict_put(src, "vnc.acl..name", qstring_from_str("acl0"));
605 qdict_put(src, "vnc.acl.rule..name", qstring_from_str("acl0"));
607 dst = qobject_to_qdict(qdict_crumple(src, &error_abort));
609 g_assert_cmpint(qdict_size(dst), ==, 1);
611 vnc = qdict_get_qdict(dst, "vnc");
613 g_assert_cmpint(qdict_size(vnc), ==, 3);
615 listen = qdict_get_qdict(vnc, "listen");
617 g_assert_cmpint(qdict_size(listen), ==, 2);
618 g_assert_cmpstr("127.0.0.1", ==, qdict_get_str(listen, "addr"));
619 g_assert_cmpstr("5901", ==, qdict_get_str(listen, "port"));
621 acl = qdict_get_qdict(vnc, "acl");
623 g_assert_cmpint(qdict_size(acl), ==, 3);
625 rules = qdict_get_qlist(acl, "rules");
627 g_assert_cmpint(qlist_size(rules), ==, 2);
629 rule = qobject_to_qdict(qlist_pop(rules));
631 g_assert_cmpint(qdict_size(rule), ==, 2);
632 g_assert_cmpstr("fred", ==, qdict_get_str(rule, "match"));
633 g_assert_cmpstr("allow", ==, qdict_get_str(rule, "policy"));
636 rule = qobject_to_qdict(qlist_pop(rules));
638 g_assert_cmpint(qdict_size(rule), ==, 2);
639 g_assert_cmpstr("bob", ==, qdict_get_str(rule, "match"));
640 g_assert_cmpstr("deny", ==, qdict_get_str(rule, "policy"));
643 /* With recursive crumpling, we should see all names unescaped */
644 g_assert_cmpstr("acl0", ==, qdict_get_str(vnc, "acl.name"));
645 g_assert_cmpstr("acl0", ==, qdict_get_str(acl, "rule.name"));
651 static void qdict_crumple_test_empty(void)
657 dst = (QDict *)qdict_crumple(src, &error_abort);
659 g_assert_cmpint(qdict_size(dst), ==, 0);
665 static void qdict_crumple_test_bad_inputs(void)
671 /* rule.0 can't be both a string and a dict */
672 qdict_put(src, "rule.0", qstring_from_str("fred"));
673 qdict_put(src, "rule.0.policy", qstring_from_str("allow"));
675 g_assert(qdict_crumple(src, &error) == NULL);
676 g_assert(error != NULL);
682 /* rule can't be both a list and a dict */
683 qdict_put(src, "rule.0", qstring_from_str("fred"));
684 qdict_put(src, "rule.a", qstring_from_str("allow"));
686 g_assert(qdict_crumple(src, &error) == NULL);
687 g_assert(error != NULL);
693 /* The input should be flat, ie no dicts or lists */
694 qdict_put(src, "rule.a", qdict_new());
695 qdict_put(src, "rule.b", qstring_from_str("allow"));
697 g_assert(qdict_crumple(src, &error) == NULL);
698 g_assert(error != NULL);
704 /* List indexes must not have gaps */
705 qdict_put(src, "rule.0", qstring_from_str("deny"));
706 qdict_put(src, "rule.3", qstring_from_str("allow"));
708 g_assert(qdict_crumple(src, &error) == NULL);
709 g_assert(error != NULL);
715 /* List indexes must be in %zu format */
716 qdict_put(src, "rule.0", qstring_from_str("deny"));
717 qdict_put(src, "rule.+1", qstring_from_str("allow"));
719 g_assert(qdict_crumple(src, &error) == NULL);
720 g_assert(error != NULL);
730 static void qdict_put_exists_test(void)
733 const char *key = "exists";
734 QDict *tests_dict = qdict_new();
736 qdict_put(tests_dict, key, qint_from_int(1));
737 qdict_put(tests_dict, key, qint_from_int(2));
739 value = qdict_get_int(tests_dict, key);
740 g_assert(value == 2);
742 g_assert(qdict_size(tests_dict) == 1);
747 static void qdict_get_not_exists_test(void)
749 QDict *tests_dict = qdict_new();
750 g_assert(qdict_get(tests_dict, "foo") == NULL);
758 * This is a lot big for a unit-test, but there is no other place
762 static void remove_dots(char *string)
764 char *p = strchr(string, ':');
769 static QString *read_line(FILE *file, char *key)
773 if (fscanf(file, "%127s%127s", key, value) == EOF) {
777 return qstring_from_str(value);
780 #define reset_file(file) fseek(file, 0L, SEEK_SET)
782 static void qdict_stress_test(void)
789 const char *test_file_path = "qdict-test-data.txt";
791 test_file = fopen(test_file_path, "r");
792 g_assert(test_file != NULL);
796 g_assert(qdict != NULL);
798 // Add everything from the test file
799 for (lines = 0;; lines++) {
800 value = read_line(test_file, key);
804 qdict_put(qdict, key, value);
806 g_assert(qdict_size(qdict) == lines);
808 // Check if everything is really in there
809 reset_file(test_file);
811 const char *str1, *str2;
813 value = read_line(test_file, key);
817 str1 = qstring_get_str(value);
819 str2 = qdict_get_str(qdict, key);
820 g_assert(str2 != NULL);
822 g_assert(strcmp(str1, str2) == 0);
828 reset_file(test_file);
830 value = read_line(test_file, key);
834 qdict_del(qdict, key);
837 g_assert(qdict_haskey(qdict, key) == 0);
841 g_assert(qdict_size(qdict) == 0);
845 int main(int argc, char **argv)
847 g_test_init(&argc, &argv, NULL);
849 g_test_add_func("/public/new", qdict_new_test);
850 g_test_add_func("/public/put_obj", qdict_put_obj_test);
851 g_test_add_func("/public/destroy_simple", qdict_destroy_simple_test);
853 /* Continue, but now with fixtures */
854 g_test_add_func("/public/get", qdict_get_test);
855 g_test_add_func("/public/get_int", qdict_get_int_test);
856 g_test_add_func("/public/get_try_int", qdict_get_try_int_test);
857 g_test_add_func("/public/get_str", qdict_get_str_test);
858 g_test_add_func("/public/get_try_str", qdict_get_try_str_test);
859 g_test_add_func("/public/defaults", qdict_defaults_test);
860 g_test_add_func("/public/haskey_not", qdict_haskey_not_test);
861 g_test_add_func("/public/haskey", qdict_haskey_test);
862 g_test_add_func("/public/del", qdict_del_test);
863 g_test_add_func("/public/to_qdict", qobject_to_qdict_test);
864 g_test_add_func("/public/iterapi", qdict_iterapi_test);
865 g_test_add_func("/public/flatten", qdict_flatten_test);
866 g_test_add_func("/public/array_split", qdict_array_split_test);
867 g_test_add_func("/public/array_entries", qdict_array_entries_test);
868 g_test_add_func("/public/join", qdict_join_test);
870 g_test_add_func("/errors/put_exists", qdict_put_exists_test);
871 g_test_add_func("/errors/get_not_exists", qdict_get_not_exists_test);
873 g_test_add_func("/public/crumple/recursive",
874 qdict_crumple_test_recursive);
875 g_test_add_func("/public/crumple/empty",
876 qdict_crumple_test_empty);
877 g_test_add_func("/public/crumple/bad_inputs",
878 qdict_crumple_test_bad_inputs);
882 g_test_add_func("/stress/test", qdict_stress_test);