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 } 272