Archipelago » qemu

/*

 * QMP Input Visitor unit-tests (strict mode).

 *

 * Copyright (C) 2011-2012 Red Hat Inc.

 *

 * Authors:

 *  Luiz Capitulino <lcapitulino@redhat.com>

 *  Paolo Bonzini <pbonzini@redhat.com>

 *

 * This work is licensed under the terms of the GNU GPL, version 2 or later.

 * See the COPYING file in the top-level directory.

 */

#include <glib.h>

#include <stdarg.h>

#include "qemu-common.h"

#include "qapi/qmp-input-visitor.h"

#include "test-qapi-types.h"

#include "test-qapi-visit.h"

#include "qapi/qmp/types.h"

typedef struct TestInputVisitorData {

    QObject *obj;

    QmpInputVisitor *qiv;

} TestInputVisitorData;

static void validate_teardown(TestInputVisitorData *data,

                               const void *unused)

{

    qobject_decref(data->obj);

    data->obj = NULL;

    if (data->qiv) {

        qmp_input_visitor_cleanup(data->qiv);

        data->qiv = NULL;

    }

}

/* This is provided instead of a test setup function so that the JSON

   string used by the tests are kept in the test functions (and not

   int main()) */

static GCC_FMT_ATTR(2, 3)

Visitor *validate_test_init(TestInputVisitorData *data,

                             const char *json_string, ...)

{

    Visitor *v;

    va_list ap;

    va_start(ap, json_string);

    data->obj = qobject_from_jsonv(json_string, &ap);

    va_end(ap);

    g_assert(data->obj != NULL);

    data->qiv = qmp_input_visitor_new_strict(data->obj);

    g_assert(data->qiv != NULL);

    v = qmp_input_get_visitor(data->qiv);

    g_assert(v != NULL);

    return v;

}

typedef struct TestStruct

{

    int64_t integer;

    bool boolean;

    char *string;

} TestStruct;

static void visit_type_TestStruct(Visitor *v, TestStruct **obj,

                                  const char *name, Error **errp)

{

    visit_start_struct(v, (void **)obj, "TestStruct", name, sizeof(TestStruct),

                       errp);

    visit_type_int(v, &(*obj)->integer, "integer", errp);

    visit_type_bool(v, &(*obj)->boolean, "boolean", errp);

    visit_type_str(v, &(*obj)->string, "string", errp);

    visit_end_struct(v, errp);

}

static void test_validate_struct(TestInputVisitorData *data,

                                  const void *unused)

{

    TestStruct *p = NULL;

    Error *errp = NULL;

    Visitor *v;

    v = validate_test_init(data, "{ 'integer': -42, 'boolean': true, 'string': 'foo' }");

    visit_type_TestStruct(v, &p, NULL, &errp);

    g_assert(!error_is_set(&errp));

    g_free(p->string);

    g_free(p);

}

static void test_validate_struct_nested(TestInputVisitorData *data,

                                         const void *unused)

{

    UserDefNested *udp = NULL;

    Error *errp = NULL;

    Visitor *v;

    v = validate_test_init(data, "{ 'string0': 'string0', 'dict1': { 'string1': 'string1', 'dict2': { 'userdef1': { 'integer': 42, 'string': 'string' }, 'string2': 'string2'}}}");

    visit_type_UserDefNested(v, &udp, NULL, &errp);

    g_assert(!error_is_set(&errp));

    qapi_free_UserDefNested(udp);

}

static void test_validate_list(TestInputVisitorData *data,

                                const void *unused)

{

    UserDefOneList *head = NULL;

    Error *errp = NULL;

    Visitor *v;

    v = validate_test_init(data, "[ { 'string': 'string0', 'integer': 42 }, { 'string': 'string1', 'integer': 43 }, { 'string': 'string2', 'integer': 44 } ]");

    visit_type_UserDefOneList(v, &head, NULL, &errp);

    g_assert(!error_is_set(&errp));

    qapi_free_UserDefOneList(head);

}

static void test_validate_union(TestInputVisitorData *data,

                                 const void *unused)

{

    UserDefUnion *tmp = NULL;

    Visitor *v;

    Error *errp = NULL;

    v = validate_test_init(data, "{ 'type': 'b', 'data' : { 'integer': 42 } }");

    visit_type_UserDefUnion(v, &tmp, NULL, &errp);

    g_assert(!error_is_set(&errp));

    qapi_free_UserDefUnion(tmp);

}

static void test_validate_fail_struct(TestInputVisitorData *data,

                                       const void *unused)

{

    TestStruct *p = NULL;

    Error *errp = NULL;

    Visitor *v;

    v = validate_test_init(data, "{ 'integer': -42, 'boolean': true, 'string': 'foo', 'extra': 42 }");

    visit_type_TestStruct(v, &p, NULL, &errp);

    g_assert(error_is_set(&errp));

    if (p) {

        g_free(p->string);

    }

    g_free(p);

}

static void test_validate_fail_struct_nested(TestInputVisitorData *data,

                                              const void *unused)

{

    UserDefNested *udp = NULL;

    Error *errp = NULL;

    Visitor *v;

    v = validate_test_init(data, "{ 'string0': 'string0', 'dict1': { 'string1': 'string1', 'dict2': { 'userdef1': { 'integer': 42, 'string': 'string', 'extra': [42, 23, {'foo':'bar'}] }, 'string2': 'string2'}}}");

    visit_type_UserDefNested(v, &udp, NULL, &errp);

    g_assert(error_is_set(&errp));

    qapi_free_UserDefNested(udp);

}

static void test_validate_fail_list(TestInputVisitorData *data,

                                     const void *unused)

{

    UserDefOneList *head = NULL;

    Error *errp = NULL;

    Visitor *v;

    v = validate_test_init(data, "[ { 'string': 'string0', 'integer': 42 }, { 'string': 'string1', 'integer': 43 }, { 'string': 'string2', 'integer': 44, 'extra': 'ggg' } ]");

    visit_type_UserDefOneList(v, &head, NULL, &errp);

    g_assert(error_is_set(&errp));

    qapi_free_UserDefOneList(head);

}

static void test_validate_fail_union(TestInputVisitorData *data,

                                      const void *unused)

{

    UserDefUnion *tmp = NULL;

    Error *errp = NULL;

    Visitor *v;

    v = validate_test_init(data, "{ 'type': 'b', 'data' : { 'integer': 42 }, 'extra': 'yyy' }");

    visit_type_UserDefUnion(v, &tmp, NULL, &errp);

    g_assert(error_is_set(&errp));

    qapi_free_UserDefUnion(tmp);

}

static void validate_test_add(const char *testpath,

                               TestInputVisitorData *data,

                               void (*test_func)(TestInputVisitorData *data, const void *user_data))

{

    g_test_add(testpath, TestInputVisitorData, data, NULL, test_func,

               validate_teardown);

}

int main(int argc, char **argv)

{

    TestInputVisitorData testdata;

    g_test_init(&argc, &argv, NULL);

    validate_test_add("/visitor/input-strict/pass/struct",

                       &testdata, test_validate_struct);

    validate_test_add("/visitor/input-strict/pass/struct-nested",

                       &testdata, test_validate_struct_nested);

    validate_test_add("/visitor/input-strict/pass/list",

                       &testdata, test_validate_list);

    validate_test_add("/visitor/input-strict/pass/union",

                       &testdata, test_validate_union);

    validate_test_add("/visitor/input-strict/fail/struct",

                       &testdata, test_validate_fail_struct);

    validate_test_add("/visitor/input-strict/fail/struct-nested",

                       &testdata, test_validate_fail_struct_nested);

    validate_test_add("/visitor/input-strict/fail/list",

                       &testdata, test_validate_fail_list);

    validate_test_add("/visitor/input-strict/fail/union",

                       &testdata, test_validate_fail_union);

    g_test_run();

    return 0;

}