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