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