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