tests/test-qmp-input-strict.c

   1 /*
   2  * QMP Input Visitor unit-tests (strict mode).
   3  *
   4  * Copyright (C) 2011-2012 Red Hat Inc.
   5  *
   6  * Authors:
   7  *  Luiz Capitulino <lcapitulino@redhat.com>
   8  *  Paolo Bonzini <pbonzini@redhat.com>
   9  *
  10  * This work is licensed under the terms of the GNU GPL, version 2 or later.
  11  * See the COPYING file in the top-level directory.
  12  */
  13
  14 #include <glib.h>
  15 #include <stdarg.h>
  16
  17 #include "qemu-common.h"
  18 #include "qapi/qmp-input-visitor.h"
  19 #include "test-qapi-types.h"
  20 #include "test-qapi-visit.h"
  21 #include "qapi/qmp/types.h"
  22
  23 typedef struct TestInputVisitorData {
  24     QObject *obj;
  25     QmpInputVisitor *qiv;
  26 } TestInputVisitorData;
  27
  28 static void validate_teardown(TestInputVisitorData *data,
  29                                const void *unused)
  30 {
  31     qobject_decref(data->obj);
  32     data->obj = NULL;
  33
  34     if (data->qiv) {
  35         qmp_input_visitor_cleanup(data->qiv);
  36         data->qiv = NULL;
  37     }
  38 }
  39
  40 /* This is provided instead of a test setup function so that the JSON
  41    string used by the tests are kept in the test functions (and not
  42    int main()) */
  43 static GCC_FMT_ATTR(2, 3)
  44 Visitor *validate_test_init(TestInputVisitorData *data,
  45                              const char *json_string, ...)
  46 {
  47     Visitor *v;
  48     va_list ap;
  49
  50     va_start(ap, json_string);
  51     data->obj = qobject_from_jsonv(json_string, &ap);
  52     va_end(ap);
  53
  54     g_assert(data->obj != NULL);
  55
  56     data->qiv = qmp_input_visitor_new_strict(data->obj);
  57     g_assert(data->qiv != NULL);
  58
  59     v = qmp_input_get_visitor(data->qiv);
  60     g_assert(v != NULL);
  61
  62     return v;
  63 }
  64
  65 typedef struct TestStruct
  66 {
  67     int64_t integer;
  68     bool boolean;
  69     char *string;
  70 } TestStruct;
  71
  72 static void visit_type_TestStruct(Visitor *v, TestStruct **obj,
  73                                   const char *name, Error **errp)
  74 {
  75     visit_start_struct(v, (void **)obj, "TestStruct", name, sizeof(TestStruct),
  76                        errp);
  77
  78     visit_type_int(v, &(*obj)->integer, "integer", errp);
  79     visit_type_bool(v, &(*obj)->boolean, "boolean", errp);
  80     visit_type_str(v, &(*obj)->string, "string", errp);
  81
  82     visit_end_struct(v, errp);
  83 }
  84
  85 static void test_validate_struct(TestInputVisitorData *data,
  86                                   const void *unused)
  87 {
  88     TestStruct *p = NULL;
  89     Error *errp = NULL;
  90     Visitor *v;
  91
  92     v = validate_test_init(data, "{ 'integer': -42, 'boolean': true, 'string': 'foo' }");
  93
  94     visit_type_TestStruct(v, &p, NULL, &errp);
  95     g_assert(!errp);
  96     g_free(p->string);
  97     g_free(p);
  98 }
  99
 100 static void test_validate_struct_nested(TestInputVisitorData *data,
 101                                          const void *unused)
 102 {
 103     UserDefNested *udp = NULL;
 104     Error *errp = NULL;
 105     Visitor *v;
 106
 107     v = validate_test_init(data, "{ 'string0': 'string0', 'dict1': { 'string1': 'string1', 'dict2': { 'userdef1': { 'integer': 42, 'string': 'string' }, 'string2': 'string2'}}}");
 108
 109     visit_type_UserDefNested(v, &udp, NULL, &errp);
 110     g_assert(!errp);
 111     qapi_free_UserDefNested(udp);
 112 }
 113
 114 static void test_validate_list(TestInputVisitorData *data,
 115                                 const void *unused)
 116 {
 117     UserDefOneList *head = NULL;
 118     Error *errp = NULL;
 119     Visitor *v;
 120
 121     v = validate_test_init(data, "[ { 'string': 'string0', 'integer': 42 }, { 'string': 'string1', 'integer': 43 }, { 'string': 'string2', 'integer': 44 } ]");
 122
 123     visit_type_UserDefOneList(v, &head, NULL, &errp);
 124     g_assert(!errp);
 125     qapi_free_UserDefOneList(head);
 126 }
 127
 128 static void test_validate_union(TestInputVisitorData *data,
 129                                  const void *unused)
 130 {
 131     UserDefUnion *tmp = NULL;
 132     Visitor *v;
 133     Error *errp = NULL;
 134
 135     v = validate_test_init(data, "{ 'type': 'b', 'integer': 41, 'data' : { 'integer': 42 } }");
 136
 137     visit_type_UserDefUnion(v, &tmp, NULL, &errp);
 138     g_assert(!errp);
 139     qapi_free_UserDefUnion(tmp);
 140 }
 141
 142 static void test_validate_union_flat(TestInputVisitorData *data,
 143                                      const void *unused)
 144 {
 145     UserDefFlatUnion *tmp = NULL;
 146     Visitor *v;
 147     Error *errp = NULL;
 148
 149     v = validate_test_init(data, "{ 'string': 'a', 'boolean': true }");
 150     /* TODO when generator bug is fixed, add 'integer': 41 */
 151
 152     visit_type_UserDefFlatUnion(v, &tmp, NULL, &errp);
 153     g_assert(!error_is_set(&errp));
 154     qapi_free_UserDefFlatUnion(tmp);
 155 }
 156
 157 static void test_validate_union_anon(TestInputVisitorData *data,
 158                                      const void *unused)
 159 {
 160     UserDefAnonUnion *tmp = NULL;
 161     Visitor *v;
 162     Error *errp = NULL;
 163
 164     v = validate_test_init(data, "42");
 165
 166     visit_type_UserDefAnonUnion(v, &tmp, NULL, &errp);
 167     g_assert(!error_is_set(&errp));
 168     qapi_free_UserDefAnonUnion(tmp);
 169 }
 170
 171 static void test_validate_fail_struct(TestInputVisitorData *data,
 172                                        const void *unused)
 173 {
 174     TestStruct *p = NULL;
 175     Error *errp = NULL;
 176     Visitor *v;
 177
 178     v = validate_test_init(data, "{ 'integer': -42, 'boolean': true, 'string': 'foo', 'extra': 42 }");
 179
 180     visit_type_TestStruct(v, &p, NULL, &errp);
 181     g_assert(errp);
 182     if (p) {
 183         g_free(p->string);
 184     }
 185     g_free(p);
 186 }
 187
 188 static void test_validate_fail_struct_nested(TestInputVisitorData *data,
 189                                               const void *unused)
 190 {
 191     UserDefNested *udp = NULL;
 192     Error *errp = NULL;
 193     Visitor *v;
 194
 195     v = validate_test_init(data, "{ 'string0': 'string0', 'dict1': { 'string1': 'string1', 'dict2': { 'userdef1': { 'integer': 42, 'string': 'string', 'extra': [42, 23, {'foo':'bar'}] }, 'string2': 'string2'}}}");
 196
 197     visit_type_UserDefNested(v, &udp, NULL, &errp);
 198     g_assert(errp);
 199     qapi_free_UserDefNested(udp);
 200 }
 201
 202 static void test_validate_fail_list(TestInputVisitorData *data,
 203                                      const void *unused)
 204 {
 205     UserDefOneList *head = NULL;
 206     Error *errp = NULL;
 207     Visitor *v;
 208
 209     v = validate_test_init(data, "[ { 'string': 'string0', 'integer': 42 }, { 'string': 'string1', 'integer': 43 }, { 'string': 'string2', 'integer': 44, 'extra': 'ggg' } ]");
 210
 211     visit_type_UserDefOneList(v, &head, NULL, &errp);
 212     g_assert(errp);
 213     qapi_free_UserDefOneList(head);
 214 }
 215
 216 static void test_validate_fail_union(TestInputVisitorData *data,
 217                                       const void *unused)
 218 {
 219     UserDefUnion *tmp = NULL;
 220     Error *errp = NULL;
 221     Visitor *v;
 222
 223     v = validate_test_init(data, "{ 'type': 'b', 'data' : { 'integer': 42 } }");
 224
 225     visit_type_UserDefUnion(v, &tmp, NULL, &errp);
 226     g_assert(errp);
 227     qapi_free_UserDefUnion(tmp);
 228 }
 229
 230 static void test_validate_fail_union_flat(TestInputVisitorData *data,
 231                                           const void *unused)
 232 {
 233     UserDefFlatUnion *tmp = NULL;
 234     Error *errp = NULL;
 235     Visitor *v;
 236
 237     v = validate_test_init(data, "{ 'string': 'c', 'integer': 41, 'boolean': true }");
 238
 239     visit_type_UserDefFlatUnion(v, &tmp, NULL, &errp);
 240     g_assert(error_is_set(&errp));
 241     qapi_free_UserDefFlatUnion(tmp);
 242 }
 243
 244 static void test_validate_fail_union_anon(TestInputVisitorData *data,
 245                                           const void *unused)
 246 {
 247     UserDefAnonUnion *tmp = NULL;
 248     Visitor *v;
 249     Error *errp = NULL;
 250
 251     v = validate_test_init(data, "3.14");
 252
 253     visit_type_UserDefAnonUnion(v, &tmp, NULL, &errp);
 254     g_assert(error_is_set(&errp));
 255     qapi_free_UserDefAnonUnion(tmp);
 256 }
 257
 258 static void validate_test_add(const char *testpath,
 259                                TestInputVisitorData *data,
 260                                void (*test_func)(TestInputVisitorData *data, const void *user_data))
 261 {
 262     g_test_add(testpath, TestInputVisitorData, data, NULL, test_func,
 263                validate_teardown);
 264 }
 265
 266 int main(int argc, char **argv)
 267 {
 268     TestInputVisitorData testdata;
 269
 270     g_test_init(&argc, &argv, NULL);
 271
 272     validate_test_add("/visitor/input-strict/pass/struct",
 273                        &testdata, test_validate_struct);
 274     validate_test_add("/visitor/input-strict/pass/struct-nested",
 275                        &testdata, test_validate_struct_nested);
 276     validate_test_add("/visitor/input-strict/pass/list",
 277                        &testdata, test_validate_list);
 278     validate_test_add("/visitor/input-strict/pass/union",
 279                        &testdata, test_validate_union);
 280     validate_test_add("/visitor/input-strict/pass/union-flat",
 281                        &testdata, test_validate_union_flat);
 282     validate_test_add("/visitor/input-strict/pass/union-anon",
 283                        &testdata, test_validate_union_anon);
 284     validate_test_add("/visitor/input-strict/fail/struct",
 285                        &testdata, test_validate_fail_struct);
 286     validate_test_add("/visitor/input-strict/fail/struct-nested",
 287                        &testdata, test_validate_fail_struct_nested);
 288     validate_test_add("/visitor/input-strict/fail/list",
 289                        &testdata, test_validate_fail_list);
 290     validate_test_add("/visitor/input-strict/fail/union",
 291                        &testdata, test_validate_fail_union);
 292     validate_test_add("/visitor/input-strict/fail/union-flat",
 293                        &testdata, test_validate_fail_union_flat);
 294     validate_test_add("/visitor/input-strict/fail/union-anon",
 295                        &testdata, test_validate_fail_union_anon);
 296
 297     g_test_run();
 298
 299     return 0;
 300 }