create-emitter-tests.py 3.37 KB
Newer Older
Jesse Beder's avatar
Jesse Beder committed
1
2
3
4
5
6
7
import sys
import yaml
import hashlib

NS = 'Emitter'
EVENT_COUNT = 5

8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
def encode_stream(line):
    for c in line:
        if c == '\n':
            yield '\\n'
        elif c == '"':
            yield '\\"'
        elif c == '\t':
            yield '\\t'
        elif ord(c) < 0x20:
            yield '\\x' + hex(ord(c))
        else:
            yield c

def encode(line):
    return ''.join(encode_stream(line))

def doc_start(implicit=False):
    if implicit:
        return {'emit': '', 'handle': 'DOC_START()'}
    else:
28
        return {'emit': 'YAML::BeginDoc', 'handle': 'DOC_START()'}
29
30
31
32
33

def doc_end(implicit=False):
    if implicit:
        return {'emit': '', 'handle': 'DOC_END()'}
    else:
34
        return {'emit': 'YAML::EndDoc', 'handle': 'DOC_END()'}
35
36
37
38
39
40
41

def scalar(value, tag='', anchor='', anchor_id=0):
    emit = []
    if tag:
        emit += ['YAML::VerbatimTag("%s")' % encode(tag)]
    if anchor:
        emit += ['YAML::Anchor("%s")' % encode(anchor)]
42
43
44
    if tag:
        out_tag = encode(tag)
    else:
Jesse Beder's avatar
Jesse Beder committed
45
46
47
48
        if value == encode(value):
            out_tag = '?'
        else:
            out_tag = '!'
49
    emit += ['"%s"' % encode(value)]
50
    return {'emit': emit, 'handle': 'SCALAR("%s", %s, "%s")' % (out_tag, anchor_id, encode(value))}
51

52
53
54
55
def comment(value):
    return {'emit': 'YAML::Comment("%s")' % value, 'handle': ''}

def gen_templates():
Jesse Beder's avatar
Jesse Beder committed
56
    yield [[doc_start(), doc_start(True)], [scalar('foo'), scalar('foo\n')], [doc_end(), doc_end(True)]]
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

def expand(template):
    if len(template) == 0:
        pass
    elif len(template) == 1:
        for item in template[0]:
            if isinstance(item, list):
                yield item
            else:
                yield [item]
    else:
        for car in expand(template[:1]):
            for cdr in expand(template[1:]):
                yield car + cdr
            
Jesse Beder's avatar
Jesse Beder committed
72
73

def gen_events():
74
75
    for template in gen_templates():
        for events in expand(template):
Jesse Beder's avatar
Jesse Beder committed
76
77
78
79
80
            base = list(events)
            for i in range(0, len(base)+1):
                cpy = list(base)
                cpy.insert(i, comment('comment'))
                yield cpy
Jesse Beder's avatar
Jesse Beder committed
81
82
83
84
85
86
87
88
89
90

def gen_tests():
    for events in gen_events():
        name = 'test' + hashlib.sha1(''.join(yaml.dump(event) for event in events)).hexdigest()[:20]
        yield {'name': name, 'events': events}
        

def create_emitter_tests(out):
    out.write('namespace %s {\n' % NS)

91
92
93
94
    tests = list(gen_tests())

    for test in tests:
        out.write('TEST %s(YAML::Emitter& out)\n' % test['name'])
Jesse Beder's avatar
Jesse Beder committed
95
96
97
        out.write('{\n')
        for event in test['events']:
            emit = event['emit']
98
99
100
101
            if isinstance(emit, list):
                for e in emit:
                    out.write('    out << %s;\n' % e)
            elif emit:
Jesse Beder's avatar
Jesse Beder committed
102
103
104
105
106
107
108
109
110
111
112
113
                out.write('    out << %s;\n' % emit)
        out.write('\n')
        out.write('    HANDLE(out.c_str());\n')
        for event in test['events']:
            handle = event['handle']
            if handle:
                out.write('    EXPECT_%s;\n' % handle)
        out.write('    DONE();\n')
        out.write('}\n')

    out.write('}\n')

114
115
116
    out.write('void RunGenEmitterTests(int& passed, int& total)\n')
    out.write('{\n')
    for test in tests:
117
        out.write('    RunGenEmitterTest(&Emitter::%s, "%s", passed, total);\n' % (test['name'], encode(test['name'])))
118
119
    out.write('}\n')

Jesse Beder's avatar
Jesse Beder committed
120
121
if __name__ == '__main__':
    create_emitter_tests(sys.stdout)