tests/test-qmp-commands.c

   1 #include <glib.h>
   2 #include "qemu-common.h"
   3 #include "qapi/qmp/types.h"
   4 #include "test-qmp-commands.h"
   5 #include "qapi/qmp/dispatch.h"
   6 #include "qemu/module.h"
   7 #include "qapi/qmp-input-visitor.h"
   8 #include "tests/test-qapi-types.h"
   9 #include "tests/test-qapi-visit.h"
  10
  11 void qmp_user_def_cmd(Error **errp)
  12 {
  13 }
  14
  15 void qmp_user_def_cmd1(UserDefOne * ud1, Error **errp)
  16 {
  17 }
  18
  19 UserDefTwo *qmp_user_def_cmd2(UserDefOne *ud1a,
  20                               bool has_udb1, UserDefOne *ud1b,
  21                               Error **errp)
  22 {
  23     UserDefTwo *ret;
  24     UserDefOne *ud1c = g_malloc0(sizeof(UserDefOne));
  25     UserDefOne *ud1d = g_malloc0(sizeof(UserDefOne));
  26
  27     ud1c->string = strdup(ud1a->string);
  28     ud1c->integer = ud1a->integer;
  29     ud1d->string = strdup(has_udb1 ? ud1b->string : "blah0");
  30     ud1d->integer = has_udb1 ? ud1b->integer : 0;
  31
  32     ret = g_new0(UserDefTwo, 1);
  33     ret->string0 = strdup("blah1");
  34     ret->dict1 = g_new0(UserDefTwoDict, 1);
  35     ret->dict1->string1 = strdup("blah2");
  36     ret->dict1->dict2 = g_new0(UserDefTwoDictDict, 1);
  37     ret->dict1->dict2->userdef = ud1c;
  38     ret->dict1->dict2->string = strdup("blah3");
  39     ret->dict1->dict3 = g_new0(UserDefTwoDictDict, 1);
  40     ret->dict1->has_dict3 = true;
  41     ret->dict1->dict3->userdef = ud1d;
  42     ret->dict1->dict3->string = strdup("blah4");
  43
  44     return ret;
  45 }
  46
  47 int64_t qmp_guest_get_time(int64_t a, bool has_b, int64_t b, Error **errp)
  48 {
  49     return a + (has_b ? b : 0);
  50 }
  51
  52 QObject *qmp_guest_sync(QObject *arg, Error **errp)
  53 {
  54     return arg;
  55 }
  56
  57 __org_qemu_x_Union1 *qmp___org_qemu_x_command(__org_qemu_x_EnumList *a,
  58                                               __org_qemu_x_StructList *b,
  59                                               __org_qemu_x_Union2 *c,
  60                                               __org_qemu_x_Alt *d,
  61                                               Error **errp)
  62 {
  63     __org_qemu_x_Union1 *ret = g_new0(__org_qemu_x_Union1, 1);
  64
  65     ret->type = ORG_QEMU_X_UNION1_KIND___ORG_QEMU_X_BRANCH;
  66     ret->u.__org_qemu_x_branch = strdup("blah1");
  67
  68     /* Also test that 'wchar-t' was munged to 'q_wchar_t' */
  69     if (b && b->value && !b->value->has_q_wchar_t) {
  70         b->value->q_wchar_t = 1;
  71     }
  72     return ret;
  73 }
  74
  75
  76 /* test commands with no input and no return value */
  77 static void test_dispatch_cmd(void)
  78 {
  79     QDict *req = qdict_new();
  80     QObject *resp;
  81
  82     qdict_put_obj(req, "execute", QOBJECT(qstring_from_str("user_def_cmd")));
  83
  84     resp = qmp_dispatch(QOBJECT(req));
  85     assert(resp != NULL);
  86     assert(!qdict_haskey(qobject_to_qdict(resp), "error"));
  87
  88     qobject_decref(resp);
  89     QDECREF(req);
  90 }
  91
  92 /* test commands that return an error due to invalid parameters */
  93 static void test_dispatch_cmd_error(void)
  94 {
  95     QDict *req = qdict_new();
  96     QObject *resp;
  97
  98     qdict_put_obj(req, "execute", QOBJECT(qstring_from_str("user_def_cmd2")));
  99
 100     resp = qmp_dispatch(QOBJECT(req));
 101     assert(resp != NULL);
 102     assert(qdict_haskey(qobject_to_qdict(resp), "error"));
 103
 104     qobject_decref(resp);
 105     QDECREF(req);
 106 }
 107
 108 static QObject *test_qmp_dispatch(QDict *req)
 109 {
 110     QObject *resp_obj;
 111     QDict *resp;
 112     QObject *ret;
 113
 114     resp_obj = qmp_dispatch(QOBJECT(req));
 115     assert(resp_obj);
 116     resp = qobject_to_qdict(resp_obj);
 117     assert(resp && !qdict_haskey(resp, "error"));
 118     ret = qdict_get(resp, "return");
 119     assert(ret);
 120     qobject_incref(ret);
 121     qobject_decref(resp_obj);
 122     return ret;
 123 }
 124
 125 /* test commands that involve both input parameters and return values */
 126 static void test_dispatch_cmd_io(void)
 127 {
 128     QDict *req = qdict_new();
 129     QDict *args = qdict_new();
 130     QDict *args3 = qdict_new();
 131     QDict *ud1a = qdict_new();
 132     QDict *ud1b = qdict_new();
 133     QDict *ret, *ret_dict, *ret_dict_dict, *ret_dict_dict_userdef;
 134     QDict *ret_dict_dict2, *ret_dict_dict2_userdef;
 135     QInt *ret3;
 136
 137     qdict_put_obj(ud1a, "integer", QOBJECT(qint_from_int(42)));
 138     qdict_put_obj(ud1a, "string", QOBJECT(qstring_from_str("hello")));
 139     qdict_put_obj(ud1b, "integer", QOBJECT(qint_from_int(422)));
 140     qdict_put_obj(ud1b, "string", QOBJECT(qstring_from_str("hello2")));
 141     qdict_put_obj(args, "ud1a", QOBJECT(ud1a));
 142     qdict_put_obj(args, "ud1b", QOBJECT(ud1b));
 143     qdict_put_obj(req, "arguments", QOBJECT(args));
 144     qdict_put_obj(req, "execute", QOBJECT(qstring_from_str("user_def_cmd2")));
 145
 146     ret = qobject_to_qdict(test_qmp_dispatch(req));
 147
 148     assert(!strcmp(qdict_get_str(ret, "string0"), "blah1"));
 149     ret_dict = qdict_get_qdict(ret, "dict1");
 150     assert(!strcmp(qdict_get_str(ret_dict, "string1"), "blah2"));
 151     ret_dict_dict = qdict_get_qdict(ret_dict, "dict2");
 152     ret_dict_dict_userdef = qdict_get_qdict(ret_dict_dict, "userdef");
 153     assert(qdict_get_int(ret_dict_dict_userdef, "integer") == 42);
 154     assert(!strcmp(qdict_get_str(ret_dict_dict_userdef, "string"), "hello"));
 155     assert(!strcmp(qdict_get_str(ret_dict_dict, "string"), "blah3"));
 156     ret_dict_dict2 = qdict_get_qdict(ret_dict, "dict3");
 157     ret_dict_dict2_userdef = qdict_get_qdict(ret_dict_dict2, "userdef");
 158     assert(qdict_get_int(ret_dict_dict2_userdef, "integer") == 422);
 159     assert(!strcmp(qdict_get_str(ret_dict_dict2_userdef, "string"), "hello2"));
 160     assert(!strcmp(qdict_get_str(ret_dict_dict2, "string"), "blah4"));
 161     QDECREF(ret);
 162
 163     qdict_put(args3, "a", qint_from_int(66));
 164     qdict_put(req, "arguments", args3);
 165     qdict_put(req, "execute", qstring_from_str("guest-get-time"));
 166
 167     ret3 = qobject_to_qint(test_qmp_dispatch(req));
 168     assert(qint_get_int(ret3) == 66);
 169     QDECREF(ret3);
 170
 171     QDECREF(req);
 172 }
 173
 174 /* test generated dealloc functions for generated types */
 175 static void test_dealloc_types(void)
 176 {
 177     UserDefOne *ud1test, *ud1a, *ud1b;
 178     UserDefOneList *ud1list;
 179
 180     ud1test = g_malloc0(sizeof(UserDefOne));
 181     ud1test->integer = 42;
 182     ud1test->string = g_strdup("hi there 42");
 183
 184     qapi_free_UserDefOne(ud1test);
 185
 186     ud1a = g_malloc0(sizeof(UserDefOne));
 187     ud1a->integer = 43;
 188     ud1a->string = g_strdup("hi there 43");
 189
 190     ud1b = g_malloc0(sizeof(UserDefOne));
 191     ud1b->integer = 44;
 192     ud1b->string = g_strdup("hi there 44");
 193
 194     ud1list = g_malloc0(sizeof(UserDefOneList));
 195     ud1list->value = ud1a;
 196     ud1list->next = g_malloc0(sizeof(UserDefOneList));
 197     ud1list->next->value = ud1b;
 198
 199     qapi_free_UserDefOneList(ud1list);
 200 }
 201
 202 /* test generated deallocation on an object whose construction was prematurely
 203  * terminated due to an error */
 204 static void test_dealloc_partial(void)
 205 {
 206     static const char text[] = "don't leak me";
 207
 208     UserDefTwo *ud2 = NULL;
 209     Error *err = NULL;
 210
 211     /* create partial object */
 212     {
 213         QDict *ud2_dict;
 214         QmpInputVisitor *qiv;
 215
 216         ud2_dict = qdict_new();
 217         qdict_put_obj(ud2_dict, "string0", QOBJECT(qstring_from_str(text)));
 218
 219         qiv = qmp_input_visitor_new(QOBJECT(ud2_dict));
 220         visit_type_UserDefTwo(qmp_input_get_visitor(qiv), &ud2, NULL, &err);
 221         qmp_input_visitor_cleanup(qiv);
 222         QDECREF(ud2_dict);
 223     }
 224
 225     /* verify partial success */
 226     assert(ud2 != NULL);
 227     assert(ud2->string0 != NULL);
 228     assert(strcmp(ud2->string0, text) == 0);
 229     assert(ud2->dict1 == NULL);
 230
 231     /* confirm & release construction error */
 232     error_free_or_abort(&err);
 233
 234     /* tear down partial object */
 235     qapi_free_UserDefTwo(ud2);
 236 }
 237
 238
 239 int main(int argc, char **argv)
 240 {
 241     g_test_init(&argc, &argv, NULL);
 242
 243     g_test_add_func("/0.15/dispatch_cmd", test_dispatch_cmd);
 244     g_test_add_func("/0.15/dispatch_cmd_error", test_dispatch_cmd_error);
 245     g_test_add_func("/0.15/dispatch_cmd_io", test_dispatch_cmd_io);
 246     g_test_add_func("/0.15/dealloc_types", test_dealloc_types);
 247     g_test_add_func("/0.15/dealloc_partial", test_dealloc_partial);
 248
 249     module_call_init(MODULE_INIT_QAPI);
 250     g_test_run();
 251
 252     return 0;
 253 }