xref: /qemu/tests/qapi-schema/test-qapi.py (revision 41d0ad1d) 
1#!/usr/bin/env python3
2#
3# QAPI parser test harness
4#
5# Copyright (c) 2013 Red Hat Inc.
6#
7# Authors:
8#  Markus Armbruster <armbru@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
15import argparse
16import difflib
17import os
18import sys
19from io import StringIO
20
21from qapi.error import QAPIError
22from qapi.schema import QAPISchema, QAPISchemaVisitor
23
24
25class QAPISchemaTestVisitor(QAPISchemaVisitor):
26
27    def visit_module(self, name):
28        print('module %s' % name)
29
30    def visit_include(self, name, info):
31        print('include %s' % name)
32
33    def visit_enum_type(self, name, info, ifcond, features, members, prefix):
34        print('enum %s' % name)
35        if prefix:
36            print('    prefix %s' % prefix)
37        for m in members:
38            print('    member %s' % m.name)
39            self._print_if(m.ifcond, indent=8)
40            self._print_features(m.features, indent=8)
41        self._print_if(ifcond)
42        self._print_features(features)
43
44    def visit_array_type(self, name, info, ifcond, element_type):
45        if not info:
46            return              # suppress built-in arrays
47        print('array %s %s' % (name, element_type.name))
48        self._print_if(ifcond)
49
50    def visit_object_type(self, name, info, ifcond, features,
51                          base, members, branches):
52        print('object %s' % name)
53        if base:
54            print('    base %s' % base.name)
55        for m in members:
56            print('    member %s: %s optional=%s'
57                  % (m.name, m.type.name, m.optional))
58            self._print_if(m.ifcond, 8)
59            self._print_features(m.features, indent=8)
60        self._print_variants(branches)
61        self._print_if(ifcond)
62        self._print_features(features)
63
64    def visit_alternate_type(self, name, info, ifcond, features,
65                             alternatives):
66        print('alternate %s' % name)
67        self._print_variants(alternatives)
68        self._print_if(ifcond)
69        self._print_features(features)
70
71    def visit_command(self, name, info, ifcond, features,
72                      arg_type, ret_type, gen, success_response, boxed,
73                      allow_oob, allow_preconfig, coroutine):
74        print('command %s %s -> %s'
75              % (name, arg_type and arg_type.name,
76                 ret_type and ret_type.name))
77        print('    gen=%s success_response=%s boxed=%s oob=%s preconfig=%s%s'
78              % (gen, success_response, boxed, allow_oob, allow_preconfig,
79                 " coroutine=True" if coroutine else ""))
80        self._print_if(ifcond)
81        self._print_features(features)
82
83    def visit_event(self, name, info, ifcond, features, arg_type, boxed):
84        print('event %s %s' % (name, arg_type and arg_type.name))
85        print('    boxed=%s' % boxed)
86        self._print_if(ifcond)
87        self._print_features(features)
88
89    @staticmethod
90    def _print_variants(variants):
91        if variants:
92            print('    tag %s' % variants.tag_member.name)
93            for v in variants.variants:
94                print('    case %s: %s' % (v.name, v.type.name))
95                QAPISchemaTestVisitor._print_if(v.ifcond, indent=8)
96
97    @staticmethod
98    def _print_if(ifcond, indent=4):
99        # TODO Drop this hack after replacing OrderedDict by plain
100        # dict (requires Python 3.7)
101        def _massage(subcond):
102            if isinstance(subcond, str):
103                return subcond
104            if isinstance(subcond, list):
105                return [_massage(val) for val in subcond]
106            return {key: _massage(val) for key, val in subcond.items()}
107
108        if ifcond.is_present():
109            print('%sif %s' % (' ' * indent, _massage(ifcond.ifcond)))
110
111    @classmethod
112    def _print_features(cls, features, indent=4):
113        if features:
114            for f in features:
115                print('%sfeature %s' % (' ' * indent, f.name))
116                cls._print_if(f.ifcond, indent + 4)
117
118
119def test_frontend(fname):
120    schema = QAPISchema(fname)
121    schema.visit(QAPISchemaTestVisitor())
122
123    for doc in schema.docs:
124        if doc.symbol:
125            print('doc symbol=%s' % doc.symbol)
126        else:
127            print('doc freeform')
128        print('    body=\n%s' % doc.body.text)
129        for arg, section in doc.args.items():
130            print('    arg=%s\n%s' % (arg, section.text))
131        for feat, section in doc.features.items():
132            print('    feature=%s\n%s' % (feat, section.text))
133        for section in doc.sections:
134            print('    section=%s\n%s' % (section.tag, section.text))
135
136
137def open_test_result(dir_name, file_name, update):
138    mode = 'r+' if update else 'r'
139    try:
140        return open(os.path.join(dir_name, file_name), mode, encoding='utf-8')
141    except FileNotFoundError:
142        if not update:
143            raise
144    return open(os.path.join(dir_name, file_name), 'w+', encoding='utf-8')
145
146
147def test_and_diff(test_name, dir_name, update):
148    sys.stdout = StringIO()
149    try:
150        test_frontend(os.path.join(dir_name, test_name + '.json'))
151    except QAPIError as err:
152        errstr = str(err) + '\n'
153        if dir_name:
154            errstr = errstr.replace(dir_name + '/', '')
155        actual_err = errstr.splitlines(True)
156    else:
157        actual_err = []
158    finally:
159        actual_out = sys.stdout.getvalue().splitlines(True)
160        sys.stdout.close()
161        sys.stdout = sys.__stdout__
162
163    try:
164        outfp = open_test_result(dir_name, test_name + '.out', update)
165        errfp = open_test_result(dir_name, test_name + '.err', update)
166        expected_out = outfp.readlines()
167        expected_err = errfp.readlines()
168    except OSError as err:
169        print("%s: can't open '%s': %s"
170              % (sys.argv[0], err.filename, err.strerror),
171              file=sys.stderr)
172        return 2
173
174    if actual_out == expected_out and actual_err == expected_err:
175        return 0
176
177    print("%s %s" % (test_name, 'UPDATE' if update else 'FAIL'),
178          file=sys.stderr)
179    out_diff = difflib.unified_diff(expected_out, actual_out, outfp.name)
180    err_diff = difflib.unified_diff(expected_err, actual_err, errfp.name)
181    sys.stdout.writelines(out_diff)
182    sys.stdout.writelines(err_diff)
183
184    if not update:
185        return 1
186
187    try:
188        outfp.truncate(0)
189        outfp.seek(0)
190        outfp.writelines(actual_out)
191        errfp.truncate(0)
192        errfp.seek(0)
193        errfp.writelines(actual_err)
194    except OSError as err:
195        print("%s: can't write '%s': %s"
196              % (sys.argv[0], err.filename, err.strerror),
197              file=sys.stderr)
198        return 2
199
200    return 0
201
202
203def main(argv):
204    parser = argparse.ArgumentParser(
205        description='QAPI schema tester')
206    parser.add_argument('-d', '--dir', action='store', default='',
207                        help="directory containing tests")
208    parser.add_argument('-u', '--update', action='store_true',
209                        default='QAPI_TEST_UPDATE' in os.environ,
210                        help="update expected test results")
211    parser.add_argument('tests', nargs='*', metavar='TEST', action='store')
212    args = parser.parse_args()
213
214    status = 0
215    for t in args.tests:
216        (dir_name, base_name) = os.path.split(t)
217        dir_name = dir_name or args.dir
218        test_name = os.path.splitext(base_name)[0]
219        status |= test_and_diff(test_name, dir_name, args.update)
220
221    sys.exit(status)
222
223
224if __name__ == '__main__':
225    main(sys.argv)
226    sys.exit(0)
227