qemu/tests/test-string-input-visitor.c
<<
>>
Prefs
   1/*
   2 * String Input Visitor unit-tests.
   3 *
   4 * Copyright (C) 2012 Red Hat Inc.
   5 *
   6 * Authors:
   7 *  Paolo Bonzini <pbonzini@redhat.com> (based on test-qmp-input-visitor)
   8 *
   9 * This work is licensed under the terms of the GNU GPL, version 2 or later.
  10 * See the COPYING file in the top-level directory.
  11 */
  12
  13#include "qemu/osdep.h"
  14#include <glib.h>
  15
  16#include "qemu-common.h"
  17#include "qapi/error.h"
  18#include "qapi/string-input-visitor.h"
  19#include "test-qapi-types.h"
  20#include "test-qapi-visit.h"
  21#include "qapi/qmp/types.h"
  22
  23typedef struct TestInputVisitorData {
  24    StringInputVisitor *siv;
  25} TestInputVisitorData;
  26
  27static void visitor_input_teardown(TestInputVisitorData *data,
  28                                   const void *unused)
  29{
  30    if (data->siv) {
  31        string_input_visitor_cleanup(data->siv);
  32        data->siv = NULL;
  33    }
  34}
  35
  36/* This is provided instead of a test setup function so that the JSON
  37   string used by the tests are kept in the test functions (and not
  38   int main()) */
  39static
  40Visitor *visitor_input_test_init(TestInputVisitorData *data,
  41                                 const char *string)
  42{
  43    Visitor *v;
  44
  45    data->siv = string_input_visitor_new(string);
  46    g_assert(data->siv != NULL);
  47
  48    v = string_input_get_visitor(data->siv);
  49    g_assert(v != NULL);
  50
  51    return v;
  52}
  53
  54static void test_visitor_in_int(TestInputVisitorData *data,
  55                                const void *unused)
  56{
  57    int64_t res = 0, value = -42;
  58    Error *err = NULL;
  59    Visitor *v;
  60
  61    v = visitor_input_test_init(data, "-42");
  62
  63    visit_type_int(v, NULL, &res, &err);
  64    g_assert(!err);
  65    g_assert_cmpint(res, ==, value);
  66}
  67
  68static void test_visitor_in_intList(TestInputVisitorData *data,
  69                                    const void *unused)
  70{
  71    int64_t value[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 20};
  72    int16List *res = NULL, *tmp;
  73    Visitor *v;
  74    int i = 0;
  75
  76    v = visitor_input_test_init(data, "1,2,0,2-4,20,5-9,1-8");
  77
  78    visit_type_int16List(v, NULL, &res, &error_abort);
  79    tmp = res;
  80    while (i < sizeof(value) / sizeof(value[0])) {
  81        g_assert(tmp);
  82        g_assert_cmpint(tmp->value, ==, value[i++]);
  83        tmp = tmp->next;
  84    }
  85    g_assert(!tmp);
  86
  87    tmp = res;
  88    while (tmp) {
  89        res = res->next;
  90        g_free(tmp);
  91        tmp = res;
  92    }
  93}
  94
  95static void test_visitor_in_bool(TestInputVisitorData *data,
  96                                 const void *unused)
  97{
  98    Error *err = NULL;
  99    bool res = false;
 100    Visitor *v;
 101
 102    v = visitor_input_test_init(data, "true");
 103
 104    visit_type_bool(v, NULL, &res, &err);
 105    g_assert(!err);
 106    g_assert_cmpint(res, ==, true);
 107    visitor_input_teardown(data, unused);
 108
 109    v = visitor_input_test_init(data, "yes");
 110
 111    visit_type_bool(v, NULL, &res, &err);
 112    g_assert(!err);
 113    g_assert_cmpint(res, ==, true);
 114    visitor_input_teardown(data, unused);
 115
 116    v = visitor_input_test_init(data, "on");
 117
 118    visit_type_bool(v, NULL, &res, &err);
 119    g_assert(!err);
 120    g_assert_cmpint(res, ==, true);
 121    visitor_input_teardown(data, unused);
 122
 123    v = visitor_input_test_init(data, "false");
 124
 125    visit_type_bool(v, NULL, &res, &err);
 126    g_assert(!err);
 127    g_assert_cmpint(res, ==, false);
 128    visitor_input_teardown(data, unused);
 129
 130    v = visitor_input_test_init(data, "no");
 131
 132    visit_type_bool(v, NULL, &res, &err);
 133    g_assert(!err);
 134    g_assert_cmpint(res, ==, false);
 135    visitor_input_teardown(data, unused);
 136
 137    v = visitor_input_test_init(data, "off");
 138
 139    visit_type_bool(v, NULL, &res, &err);
 140    g_assert(!err);
 141    g_assert_cmpint(res, ==, false);
 142}
 143
 144static void test_visitor_in_number(TestInputVisitorData *data,
 145                                   const void *unused)
 146{
 147    double res = 0, value = 3.14;
 148    Error *err = NULL;
 149    Visitor *v;
 150
 151    v = visitor_input_test_init(data, "3.14");
 152
 153    visit_type_number(v, NULL, &res, &err);
 154    g_assert(!err);
 155    g_assert_cmpfloat(res, ==, value);
 156}
 157
 158static void test_visitor_in_string(TestInputVisitorData *data,
 159                                   const void *unused)
 160{
 161    char *res = NULL, *value = (char *) "Q E M U";
 162    Error *err = NULL;
 163    Visitor *v;
 164
 165    v = visitor_input_test_init(data, value);
 166
 167    visit_type_str(v, NULL, &res, &err);
 168    g_assert(!err);
 169    g_assert_cmpstr(res, ==, value);
 170
 171    g_free(res);
 172}
 173
 174static void test_visitor_in_enum(TestInputVisitorData *data,
 175                                 const void *unused)
 176{
 177    Error *err = NULL;
 178    Visitor *v;
 179    EnumOne i;
 180
 181    for (i = 0; EnumOne_lookup[i]; i++) {
 182        EnumOne res = -1;
 183
 184        v = visitor_input_test_init(data, EnumOne_lookup[i]);
 185
 186        visit_type_EnumOne(v, NULL, &res, &err);
 187        g_assert(!err);
 188        g_assert_cmpint(i, ==, res);
 189
 190        visitor_input_teardown(data, NULL);
 191    }
 192
 193    data->siv = NULL;
 194}
 195
 196/* Try to crash the visitors */
 197static void test_visitor_in_fuzz(TestInputVisitorData *data,
 198                                 const void *unused)
 199{
 200    int64_t ires;
 201    intList *ilres;
 202    bool bres;
 203    double nres;
 204    char *sres;
 205    EnumOne eres;
 206    Visitor *v;
 207    unsigned int i;
 208    char buf[10000];
 209
 210    for (i = 0; i < 100; i++) {
 211        unsigned int j;
 212
 213        j = g_test_rand_int_range(0, sizeof(buf) - 1);
 214
 215        buf[j] = '\0';
 216
 217        if (j != 0) {
 218            for (j--; j != 0; j--) {
 219                buf[j - 1] = (char)g_test_rand_int_range(0, 256);
 220            }
 221        }
 222
 223        v = visitor_input_test_init(data, buf);
 224        visit_type_int(v, NULL, &ires, NULL);
 225        visitor_input_teardown(data, NULL);
 226
 227        v = visitor_input_test_init(data, buf);
 228        visit_type_intList(v, NULL, &ilres, NULL);
 229        visitor_input_teardown(data, NULL);
 230
 231        v = visitor_input_test_init(data, buf);
 232        visit_type_bool(v, NULL, &bres, NULL);
 233        visitor_input_teardown(data, NULL);
 234
 235        v = visitor_input_test_init(data, buf);
 236        visit_type_number(v, NULL, &nres, NULL);
 237        visitor_input_teardown(data, NULL);
 238
 239        v = visitor_input_test_init(data, buf);
 240        sres = NULL;
 241        visit_type_str(v, NULL, &sres, NULL);
 242        g_free(sres);
 243        visitor_input_teardown(data, NULL);
 244
 245        v = visitor_input_test_init(data, buf);
 246        visit_type_EnumOne(v, NULL, &eres, NULL);
 247        visitor_input_teardown(data, NULL);
 248    }
 249}
 250
 251static void input_visitor_test_add(const char *testpath,
 252                                   TestInputVisitorData *data,
 253                                   void (*test_func)(TestInputVisitorData *data, const void *user_data))
 254{
 255    g_test_add(testpath, TestInputVisitorData, data, NULL, test_func,
 256               visitor_input_teardown);
 257}
 258
 259int main(int argc, char **argv)
 260{
 261    TestInputVisitorData in_visitor_data;
 262
 263    g_test_init(&argc, &argv, NULL);
 264
 265    input_visitor_test_add("/string-visitor/input/int",
 266                           &in_visitor_data, test_visitor_in_int);
 267    input_visitor_test_add("/string-visitor/input/intList",
 268                           &in_visitor_data, test_visitor_in_intList);
 269    input_visitor_test_add("/string-visitor/input/bool",
 270                           &in_visitor_data, test_visitor_in_bool);
 271    input_visitor_test_add("/string-visitor/input/number",
 272                           &in_visitor_data, test_visitor_in_number);
 273    input_visitor_test_add("/string-visitor/input/string",
 274                            &in_visitor_data, test_visitor_in_string);
 275    input_visitor_test_add("/string-visitor/input/enum",
 276                            &in_visitor_data, test_visitor_in_enum);
 277    input_visitor_test_add("/string-visitor/input/fuzz",
 278                            &in_visitor_data, test_visitor_in_fuzz);
 279
 280    g_test_run();
 281
 282    return 0;
 283}
 284