]>
Commit | Line | Data |
---|---|---|
f294f82a LC |
1 | /* |
2 | * QMP Output Visitor unit-tests. | |
3 | * | |
4 | * Copyright (C) 2011 Red Hat Inc. | |
5 | * | |
6 | * Authors: | |
7 | * Luiz Capitulino <[email protected]> | |
8 | * | |
9 | * This work is licensed under the terms of the GNU GPL, version 2 or later. | |
10 | * See the COPYING file in the top-level directory. | |
11 | */ | |
12 | ||
13 | #include <glib.h> | |
14 | ||
15 | #include "qapi/qmp-output-visitor.h" | |
16 | #include "test-qapi-types.h" | |
17 | #include "test-qapi-visit.h" | |
18 | #include "qemu-objects.h" | |
19 | ||
20 | typedef struct TestOutputVisitorData { | |
21 | QmpOutputVisitor *qov; | |
22 | Visitor *ov; | |
23 | } TestOutputVisitorData; | |
24 | ||
25 | static void visitor_output_setup(TestOutputVisitorData *data, | |
26 | const void *unused) | |
27 | { | |
28 | data->qov = qmp_output_visitor_new(); | |
29 | g_assert(data->qov != NULL); | |
30 | ||
31 | data->ov = qmp_output_get_visitor(data->qov); | |
32 | g_assert(data->ov != NULL); | |
33 | } | |
34 | ||
35 | static void visitor_output_teardown(TestOutputVisitorData *data, | |
36 | const void *unused) | |
37 | { | |
38 | qmp_output_visitor_cleanup(data->qov); | |
39 | data->qov = NULL; | |
40 | data->ov = NULL; | |
41 | } | |
42 | ||
43 | static void test_visitor_out_int(TestOutputVisitorData *data, | |
44 | const void *unused) | |
45 | { | |
46 | int64_t value = -42; | |
47 | Error *errp = NULL; | |
48 | QObject *obj; | |
49 | ||
50 | visit_type_int(data->ov, &value, NULL, &errp); | |
51 | g_assert(error_is_set(&errp) == 0); | |
52 | ||
53 | obj = qmp_output_get_qobject(data->qov); | |
54 | g_assert(obj != NULL); | |
55 | g_assert(qobject_type(obj) == QTYPE_QINT); | |
56 | g_assert_cmpint(qint_get_int(qobject_to_qint(obj)), ==, value); | |
57 | ||
58 | qobject_decref(obj); | |
59 | } | |
60 | ||
61 | static void test_visitor_out_bool(TestOutputVisitorData *data, | |
62 | const void *unused) | |
63 | { | |
64 | Error *errp = NULL; | |
65 | bool value = true; | |
66 | QObject *obj; | |
67 | ||
68 | visit_type_bool(data->ov, &value, NULL, &errp); | |
69 | g_assert(error_is_set(&errp) == 0); | |
70 | ||
71 | obj = qmp_output_get_qobject(data->qov); | |
72 | g_assert(obj != NULL); | |
73 | g_assert(qobject_type(obj) == QTYPE_QBOOL); | |
74 | g_assert(qbool_get_int(qobject_to_qbool(obj)) == value); | |
75 | ||
76 | qobject_decref(obj); | |
77 | } | |
78 | ||
79 | static void test_visitor_out_number(TestOutputVisitorData *data, | |
80 | const void *unused) | |
81 | { | |
82 | double value = 3.14; | |
83 | Error *errp = NULL; | |
84 | QObject *obj; | |
85 | ||
86 | visit_type_number(data->ov, &value, NULL, &errp); | |
87 | g_assert(error_is_set(&errp) == 0); | |
88 | ||
89 | obj = qmp_output_get_qobject(data->qov); | |
90 | g_assert(obj != NULL); | |
91 | g_assert(qobject_type(obj) == QTYPE_QFLOAT); | |
92 | g_assert(qfloat_get_double(qobject_to_qfloat(obj)) == value); | |
93 | ||
94 | qobject_decref(obj); | |
95 | } | |
96 | ||
97 | static void test_visitor_out_string(TestOutputVisitorData *data, | |
98 | const void *unused) | |
99 | { | |
100 | char *string = (char *) "Q E M U"; | |
101 | Error *errp = NULL; | |
102 | QObject *obj; | |
103 | ||
104 | visit_type_str(data->ov, &string, NULL, &errp); | |
105 | g_assert(error_is_set(&errp) == 0); | |
106 | ||
107 | obj = qmp_output_get_qobject(data->qov); | |
108 | g_assert(obj != NULL); | |
109 | g_assert(qobject_type(obj) == QTYPE_QSTRING); | |
110 | g_assert_cmpstr(qstring_get_str(qobject_to_qstring(obj)), ==, string); | |
111 | ||
112 | qobject_decref(obj); | |
113 | } | |
114 | ||
115 | static void test_visitor_out_no_string(TestOutputVisitorData *data, | |
116 | const void *unused) | |
117 | { | |
118 | char *string = NULL; | |
119 | Error *errp = NULL; | |
120 | QObject *obj; | |
121 | ||
122 | /* A null string should return "" */ | |
123 | visit_type_str(data->ov, &string, NULL, &errp); | |
124 | g_assert(error_is_set(&errp) == 0); | |
125 | ||
126 | obj = qmp_output_get_qobject(data->qov); | |
127 | g_assert(obj != NULL); | |
128 | g_assert(qobject_type(obj) == QTYPE_QSTRING); | |
129 | g_assert_cmpstr(qstring_get_str(qobject_to_qstring(obj)), ==, ""); | |
130 | ||
131 | qobject_decref(obj); | |
132 | } | |
133 | ||
134 | static void test_visitor_out_enum(TestOutputVisitorData *data, | |
135 | const void *unused) | |
136 | { | |
137 | Error *errp = NULL; | |
138 | QObject *obj; | |
139 | EnumOne i; | |
140 | ||
141 | for (i = 0; i < ENUM_ONE_MAX; i++) { | |
142 | visit_type_EnumOne(data->ov, &i, "unused", &errp); | |
143 | g_assert(!error_is_set(&errp)); | |
144 | ||
145 | obj = qmp_output_get_qobject(data->qov); | |
146 | g_assert(obj != NULL); | |
147 | g_assert(qobject_type(obj) == QTYPE_QSTRING); | |
148 | g_assert_cmpstr(qstring_get_str(qobject_to_qstring(obj)), ==, | |
149 | EnumOne_lookup[i]); | |
150 | qobject_decref(obj); | |
151 | } | |
152 | } | |
153 | ||
154 | static void test_visitor_out_enum_errors(TestOutputVisitorData *data, | |
155 | const void *unused) | |
156 | { | |
157 | EnumOne i, bad_values[] = { ENUM_ONE_MAX, -1 }; | |
158 | Error *errp; | |
159 | ||
160 | for (i = 0; i < ARRAY_SIZE(bad_values) ; i++) { | |
161 | errp = NULL; | |
162 | visit_type_EnumOne(data->ov, &bad_values[i], "unused", &errp); | |
163 | g_assert(error_is_set(&errp) == true); | |
164 | error_free(errp); | |
165 | } | |
166 | } | |
167 | ||
168 | typedef struct TestStruct | |
169 | { | |
170 | int64_t integer; | |
171 | bool boolean; | |
172 | char *string; | |
173 | } TestStruct; | |
174 | ||
175 | static void visit_type_TestStruct(Visitor *v, TestStruct **obj, | |
176 | const char *name, Error **errp) | |
177 | { | |
178 | visit_start_struct(v, (void **)obj, "TestStruct", name, sizeof(TestStruct), | |
179 | errp); | |
180 | ||
181 | visit_type_int(v, &(*obj)->integer, "integer", errp); | |
182 | visit_type_bool(v, &(*obj)->boolean, "boolean", errp); | |
183 | visit_type_str(v, &(*obj)->string, "string", errp); | |
184 | ||
185 | visit_end_struct(v, errp); | |
186 | } | |
187 | ||
188 | static void test_visitor_out_struct(TestOutputVisitorData *data, | |
189 | const void *unused) | |
190 | { | |
191 | TestStruct test_struct = { .integer = 42, | |
192 | .boolean = false, | |
193 | .string = (char *) "foo"}; | |
194 | TestStruct *p = &test_struct; | |
195 | Error *errp = NULL; | |
196 | QObject *obj; | |
197 | QDict *qdict; | |
198 | ||
199 | visit_type_TestStruct(data->ov, &p, NULL, &errp); | |
200 | g_assert(!error_is_set(&errp)); | |
201 | ||
202 | obj = qmp_output_get_qobject(data->qov); | |
203 | g_assert(obj != NULL); | |
204 | g_assert(qobject_type(obj) == QTYPE_QDICT); | |
205 | ||
206 | qdict = qobject_to_qdict(obj); | |
207 | g_assert_cmpint(qdict_size(qdict), ==, 3); | |
208 | g_assert_cmpint(qdict_get_int(qdict, "integer"), ==, 42); | |
209 | g_assert_cmpint(qdict_get_bool(qdict, "boolean"), ==, 0); | |
210 | g_assert_cmpstr(qdict_get_str(qdict, "string"), ==, "foo"); | |
211 | ||
212 | QDECREF(qdict); | |
213 | } | |
214 | ||
215 | static void test_visitor_out_struct_nested(TestOutputVisitorData *data, | |
216 | const void *unused) | |
217 | { | |
218 | int64_t value = 42; | |
219 | Error *errp = NULL; | |
220 | UserDefNested *ud2; | |
221 | QObject *obj; | |
222 | QDict *qdict, *dict1, *dict2, *dict3, *userdef; | |
223 | const char *string = "user def string"; | |
224 | const char *strings[] = { "fourty two", "fourty three", "fourty four", | |
225 | "fourty five" }; | |
226 | ||
227 | ud2 = g_malloc0(sizeof(*ud2)); | |
228 | ud2->string0 = g_strdup(strings[0]); | |
229 | ||
230 | ud2->dict1.string1 = g_strdup(strings[1]); | |
231 | ud2->dict1.dict2.userdef1 = g_malloc0(sizeof(UserDefOne)); | |
232 | ud2->dict1.dict2.userdef1->string = g_strdup(string); | |
233 | ud2->dict1.dict2.userdef1->integer = value; | |
234 | ud2->dict1.dict2.string2 = g_strdup(strings[2]); | |
235 | ||
236 | ud2->dict1.has_dict3 = true; | |
237 | ud2->dict1.dict3.userdef2 = g_malloc0(sizeof(UserDefOne)); | |
238 | ud2->dict1.dict3.userdef2->string = g_strdup(string); | |
239 | ud2->dict1.dict3.userdef2->integer = value; | |
240 | ud2->dict1.dict3.string3 = g_strdup(strings[3]); | |
241 | ||
242 | visit_type_UserDefNested(data->ov, &ud2, "unused", &errp); | |
243 | g_assert(!error_is_set(&errp)); | |
244 | ||
245 | obj = qmp_output_get_qobject(data->qov); | |
246 | g_assert(obj != NULL); | |
247 | g_assert(qobject_type(obj) == QTYPE_QDICT); | |
248 | ||
249 | qdict = qobject_to_qdict(obj); | |
250 | g_assert_cmpint(qdict_size(qdict), ==, 2); | |
251 | g_assert_cmpstr(qdict_get_str(qdict, "string0"), ==, strings[0]); | |
252 | ||
253 | dict1 = qdict_get_qdict(qdict, "dict1"); | |
254 | g_assert_cmpint(qdict_size(dict1), ==, 3); | |
255 | g_assert_cmpstr(qdict_get_str(dict1, "string1"), ==, strings[1]); | |
256 | ||
257 | dict2 = qdict_get_qdict(dict1, "dict2"); | |
258 | g_assert_cmpint(qdict_size(dict2), ==, 2); | |
259 | g_assert_cmpstr(qdict_get_str(dict2, "string2"), ==, strings[2]); | |
260 | userdef = qdict_get_qdict(dict2, "userdef1"); | |
261 | g_assert_cmpint(qdict_size(userdef), ==, 2); | |
262 | g_assert_cmpint(qdict_get_int(userdef, "integer"), ==, value); | |
263 | g_assert_cmpstr(qdict_get_str(userdef, "string"), ==, string); | |
264 | ||
265 | dict3 = qdict_get_qdict(dict1, "dict3"); | |
266 | g_assert_cmpint(qdict_size(dict3), ==, 2); | |
267 | g_assert_cmpstr(qdict_get_str(dict3, "string3"), ==, strings[3]); | |
268 | userdef = qdict_get_qdict(dict3, "userdef2"); | |
269 | g_assert_cmpint(qdict_size(userdef), ==, 2); | |
270 | g_assert_cmpint(qdict_get_int(userdef, "integer"), ==, value); | |
271 | g_assert_cmpstr(qdict_get_str(userdef, "string"), ==, string); | |
272 | ||
273 | QDECREF(qdict); | |
274 | qapi_free_UserDefNested(ud2); | |
275 | } | |
276 | ||
277 | typedef struct TestStructList | |
278 | { | |
279 | TestStruct *value; | |
280 | struct TestStructList *next; | |
281 | } TestStructList; | |
282 | ||
283 | static void visit_type_TestStructList(Visitor *v, TestStructList **obj, | |
284 | const char *name, Error **errp) | |
285 | { | |
286 | GenericList *i, **head = (GenericList **)obj; | |
287 | ||
288 | visit_start_list(v, name, errp); | |
289 | ||
290 | for (*head = i = visit_next_list(v, head, errp); i; i = visit_next_list(v, &i, errp)) { | |
291 | TestStructList *native_i = (TestStructList *)i; | |
292 | visit_type_TestStruct(v, &native_i->value, NULL, errp); | |
293 | } | |
294 | ||
295 | visit_end_list(v, errp); | |
296 | } | |
297 | ||
298 | static void test_visitor_out_list(TestOutputVisitorData *data, | |
299 | const void *unused) | |
300 | { | |
301 | char *value_str = (char *) "list value"; | |
302 | TestStructList *p, *head = NULL; | |
303 | const int max_items = 10; | |
304 | bool value_bool = true; | |
305 | int value_int = 10; | |
306 | Error *errp = NULL; | |
307 | QListEntry *entry; | |
308 | QObject *obj; | |
309 | QList *qlist; | |
310 | int i; | |
311 | ||
312 | for (i = 0; i < max_items; i++) { | |
313 | p = g_malloc0(sizeof(*p)); | |
314 | p->value = g_malloc0(sizeof(*p->value)); | |
315 | p->value->integer = value_int; | |
316 | p->value->boolean = value_bool; | |
317 | p->value->string = value_str; | |
318 | ||
319 | p->next = head; | |
320 | head = p; | |
321 | } | |
322 | ||
323 | visit_type_TestStructList(data->ov, &head, NULL, &errp); | |
324 | g_assert(!error_is_set(&errp)); | |
325 | ||
326 | obj = qmp_output_get_qobject(data->qov); | |
327 | g_assert(obj != NULL); | |
328 | g_assert(qobject_type(obj) == QTYPE_QLIST); | |
329 | ||
330 | qlist = qobject_to_qlist(obj); | |
331 | g_assert(!qlist_empty(qlist)); | |
332 | ||
333 | i = 0; | |
334 | QLIST_FOREACH_ENTRY(qlist, entry) { | |
335 | QDict *qdict; | |
336 | ||
337 | g_assert(qobject_type(entry->value) == QTYPE_QDICT); | |
338 | qdict = qobject_to_qdict(entry->value); | |
339 | g_assert_cmpint(qdict_size(qdict), ==, 3); | |
340 | g_assert_cmpint(qdict_get_int(qdict, "integer"), ==, value_int); | |
341 | g_assert_cmpint(qdict_get_bool(qdict, "boolean"), ==, value_bool); | |
342 | g_assert_cmpstr(qdict_get_str(qdict, "string"), ==, value_str); | |
343 | i++; | |
344 | } | |
345 | g_assert_cmpint(i, ==, max_items); | |
346 | ||
347 | QDECREF(qlist); | |
348 | ||
349 | for (p = head; p;) { | |
350 | TestStructList *tmp = p->next; | |
351 | g_free(p->value); | |
352 | g_free(p); | |
353 | p = tmp; | |
354 | } | |
355 | } | |
356 | ||
357 | static void test_visitor_out_list_qapi_free(TestOutputVisitorData *data, | |
358 | const void *unused) | |
359 | { | |
360 | UserDefNestedList *p, *head = NULL; | |
361 | const char string[] = "foo bar"; | |
362 | int i, max_count = 1024; | |
363 | ||
364 | for (i = 0; i < max_count; i++) { | |
365 | p = g_malloc0(sizeof(*p)); | |
366 | p->value = g_malloc0(sizeof(*p->value)); | |
367 | ||
368 | p->value->string0 = g_strdup(string); | |
369 | p->value->dict1.string1 = g_strdup(string); | |
370 | p->value->dict1.dict2.userdef1 = g_malloc0(sizeof(UserDefOne)); | |
371 | p->value->dict1.dict2.userdef1->string = g_strdup(string); | |
372 | p->value->dict1.dict2.userdef1->integer = 42; | |
373 | p->value->dict1.dict2.string2 = g_strdup(string); | |
374 | p->value->dict1.has_dict3 = false; | |
375 | ||
376 | p->next = head; | |
377 | head = p; | |
378 | } | |
379 | ||
380 | qapi_free_UserDefNestedList(head); | |
381 | } | |
382 | ||
383 | static void output_visitor_test_add(const char *testpath, | |
384 | TestOutputVisitorData *data, | |
385 | void (*test_func)(TestOutputVisitorData *data, const void *user_data)) | |
386 | { | |
387 | g_test_add(testpath, TestOutputVisitorData, data, visitor_output_setup, | |
388 | test_func, visitor_output_teardown); | |
389 | } | |
390 | ||
391 | int main(int argc, char **argv) | |
392 | { | |
393 | TestOutputVisitorData out_visitor_data; | |
394 | ||
395 | g_test_init(&argc, &argv, NULL); | |
396 | ||
397 | output_visitor_test_add("/visitor/output/int", | |
398 | &out_visitor_data, test_visitor_out_int); | |
399 | output_visitor_test_add("/visitor/output/bool", | |
400 | &out_visitor_data, test_visitor_out_bool); | |
401 | output_visitor_test_add("/visitor/output/number", | |
402 | &out_visitor_data, test_visitor_out_number); | |
403 | output_visitor_test_add("/visitor/output/string", | |
404 | &out_visitor_data, test_visitor_out_string); | |
405 | output_visitor_test_add("/visitor/output/no-string", | |
406 | &out_visitor_data, test_visitor_out_no_string); | |
407 | output_visitor_test_add("/visitor/output/enum", | |
408 | &out_visitor_data, test_visitor_out_enum); | |
409 | output_visitor_test_add("/visitor/output/enum-errors", | |
410 | &out_visitor_data, test_visitor_out_enum_errors); | |
411 | output_visitor_test_add("/visitor/output/struct", | |
412 | &out_visitor_data, test_visitor_out_struct); | |
413 | output_visitor_test_add("/visitor/output/struct-nested", | |
414 | &out_visitor_data, test_visitor_out_struct_nested); | |
415 | output_visitor_test_add("/visitor/output/list", | |
416 | &out_visitor_data, test_visitor_out_list); | |
417 | output_visitor_test_add("/visitor/output/list-qapi-free", | |
418 | &out_visitor_data, test_visitor_out_list_qapi_free); | |
419 | ||
420 | g_test_run(); | |
421 | ||
422 | return 0; | |
423 | } |