migration: do cleanup operation after completion
[kvmfornfv.git] / qemu / tests / test-qmp-event.c
1 /*
2  * qapi event unit-tests.
3  *
4  * Copyright (c) 2014 Wenchao Xia
5  *
6  * Authors:
7  *  Wenchao Xia   <wenchaoqemu@gmail.com>
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.
11  *
12  */
13
14 #include <glib.h>
15 #include <stdarg.h>
16
17 #include "qemu-common.h"
18 #include "test-qapi-types.h"
19 #include "test-qapi-visit.h"
20 #include "test-qapi-event.h"
21 #include "qapi/qmp/types.h"
22 #include "qapi/qmp/qint.h"
23 #include "qapi/qmp/qobject.h"
24 #include "qapi/qmp-event.h"
25
26 typedef struct TestEventData {
27     QDict *expect;
28 } TestEventData;
29
30 typedef struct QDictCmpData {
31     QDict *expect;
32     bool result;
33 } QDictCmpData;
34
35 TestEventData *test_event_data;
36 static CompatGMutex test_event_lock;
37
38 /* Only compares bool, int, string */
39 static
40 void qdict_cmp_do_simple(const char *key, QObject *obj1, void *opaque)
41
42 {
43     QObject *obj2;
44     QDictCmpData d_new, *d = opaque;
45
46     if (!d->result) {
47         return;
48     }
49
50     obj2 = qdict_get(d->expect, key);
51     if (!obj2) {
52         d->result = false;
53         return;
54     }
55
56     if (qobject_type(obj1) != qobject_type(obj2)) {
57         d->result = false;
58         return;
59     }
60
61     switch (qobject_type(obj1)) {
62     case QTYPE_QBOOL:
63         d->result = (qbool_get_bool(qobject_to_qbool(obj1)) ==
64                      qbool_get_bool(qobject_to_qbool(obj2)));
65         return;
66     case QTYPE_QINT:
67         d->result = (qint_get_int(qobject_to_qint(obj1)) ==
68                      qint_get_int(qobject_to_qint(obj2)));
69         return;
70     case QTYPE_QSTRING:
71         d->result = g_strcmp0(qstring_get_str(qobject_to_qstring(obj1)),
72                               qstring_get_str(qobject_to_qstring(obj2))) == 0;
73         return;
74     case QTYPE_QDICT:
75         d_new.expect = qobject_to_qdict(obj2);
76         d_new.result = true;
77         qdict_iter(qobject_to_qdict(obj1), qdict_cmp_do_simple, &d_new);
78         d->result = d_new.result;
79         return;
80     default:
81         abort();
82     }
83 }
84
85 static bool qdict_cmp_simple(QDict *a, QDict *b)
86 {
87     QDictCmpData d;
88
89     d.expect = b;
90     d.result = true;
91     qdict_iter(a, qdict_cmp_do_simple, &d);
92     return d.result;
93 }
94
95 /* This function is hooked as final emit function, which can verify the
96    correctness. */
97 static void event_test_emit(TEST_QAPIEvent event, QDict *d, Error **errp)
98 {
99     QObject *obj;
100     QDict *t;
101     int64_t s, ms;
102
103     /* Verify that we have timestamp, then remove it to compare other fields */
104     obj = qdict_get(d, "timestamp");
105     g_assert(obj);
106     t = qobject_to_qdict(obj);
107     g_assert(t);
108     obj = qdict_get(t, "seconds");
109     g_assert(obj && qobject_type(obj) == QTYPE_QINT);
110     s = qint_get_int(qobject_to_qint(obj));
111     obj = qdict_get(t, "microseconds");
112     g_assert(obj && qobject_type(obj) == QTYPE_QINT);
113     ms = qint_get_int(qobject_to_qint(obj));
114     if (s == -1) {
115         g_assert(ms == -1);
116     } else {
117         g_assert(ms >= 0 && ms <= 999999);
118     }
119     g_assert(qdict_size(t) == 2);
120
121     qdict_del(d, "timestamp");
122
123     g_assert(qdict_cmp_simple(d, test_event_data->expect));
124
125 }
126
127 static void event_prepare(TestEventData *data,
128                           const void *unused)
129 {
130     /* Global variable test_event_data was used to pass the expectation, so
131        test cases can't be executed at same time. */
132     g_mutex_lock(&test_event_lock);
133
134     data->expect = qdict_new();
135     test_event_data = data;
136 }
137
138 static void event_teardown(TestEventData *data,
139                            const void *unused)
140 {
141     QDECREF(data->expect);
142     test_event_data = NULL;
143
144     g_mutex_unlock(&test_event_lock);
145 }
146
147 static void event_test_add(const char *testpath,
148                            void (*test_func)(TestEventData *data,
149                                              const void *user_data))
150 {
151     g_test_add(testpath, TestEventData, NULL, event_prepare, test_func,
152                event_teardown);
153 }
154
155
156 /* Test cases */
157
158 static void test_event_a(TestEventData *data,
159                          const void *unused)
160 {
161     QDict *d;
162     d = data->expect;
163     qdict_put(d, "event", qstring_from_str("EVENT_A"));
164     qapi_event_send_event_a(&error_abort);
165 }
166
167 static void test_event_b(TestEventData *data,
168                          const void *unused)
169 {
170     QDict *d;
171     d = data->expect;
172     qdict_put(d, "event", qstring_from_str("EVENT_B"));
173     qapi_event_send_event_b(&error_abort);
174 }
175
176 static void test_event_c(TestEventData *data,
177                          const void *unused)
178 {
179     QDict *d, *d_data, *d_b;
180
181     UserDefOne b;
182     UserDefZero z;
183     z.integer = 2;
184     b.base = &z;
185     b.string = g_strdup("test1");
186     b.has_enum1 = false;
187
188     d_b = qdict_new();
189     qdict_put(d_b, "integer", qint_from_int(2));
190     qdict_put(d_b, "string", qstring_from_str("test1"));
191
192     d_data = qdict_new();
193     qdict_put(d_data, "a", qint_from_int(1));
194     qdict_put(d_data, "b", d_b);
195     qdict_put(d_data, "c", qstring_from_str("test2"));
196
197     d = data->expect;
198     qdict_put(d, "event", qstring_from_str("EVENT_C"));
199     qdict_put(d, "data", d_data);
200
201     qapi_event_send_event_c(true, 1, true, &b, "test2", &error_abort);
202
203     g_free(b.string);
204 }
205
206 /* Complex type */
207 static void test_event_d(TestEventData *data,
208                          const void *unused)
209 {
210     UserDefOne struct1;
211     EventStructOne a;
212     UserDefZero z;
213     QDict *d, *d_data, *d_a, *d_struct1;
214
215     z.integer = 2;
216     struct1.base = &z;
217     struct1.string = g_strdup("test1");
218     struct1.has_enum1 = true;
219     struct1.enum1 = ENUM_ONE_VALUE1;
220
221     a.struct1 = &struct1;
222     a.string = g_strdup("test2");
223     a.has_enum2 = true;
224     a.enum2 = ENUM_ONE_VALUE2;
225
226     d_struct1 = qdict_new();
227     qdict_put(d_struct1, "integer", qint_from_int(2));
228     qdict_put(d_struct1, "string", qstring_from_str("test1"));
229     qdict_put(d_struct1, "enum1", qstring_from_str("value1"));
230
231     d_a = qdict_new();
232     qdict_put(d_a, "struct1", d_struct1);
233     qdict_put(d_a, "string", qstring_from_str("test2"));
234     qdict_put(d_a, "enum2", qstring_from_str("value2"));
235
236     d_data = qdict_new();
237     qdict_put(d_data, "a", d_a);
238     qdict_put(d_data, "b", qstring_from_str("test3"));
239     qdict_put(d_data, "enum3", qstring_from_str("value3"));
240
241     d = data->expect;
242     qdict_put(d, "event", qstring_from_str("EVENT_D"));
243     qdict_put(d, "data", d_data);
244
245     qapi_event_send_event_d(&a, "test3", false, NULL, true, ENUM_ONE_VALUE3,
246                            &error_abort);
247
248     g_free(struct1.string);
249     g_free(a.string);
250 }
251
252 int main(int argc, char **argv)
253 {
254 #if !GLIB_CHECK_VERSION(2, 31, 0)
255     if (!g_thread_supported()) {
256        g_thread_init(NULL);
257     }
258 #endif
259
260     qmp_event_set_func_emit(event_test_emit);
261
262     g_test_init(&argc, &argv, NULL);
263
264     event_test_add("/event/event_a", test_event_a);
265     event_test_add("/event/event_b", test_event_b);
266     event_test_add("/event/event_c", test_event_c);
267     event_test_add("/event/event_d", test_event_d);
268     g_test_run();
269
270     return 0;
271 }