1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport unittest
2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport textwrap
3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport antlr3
4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport antlr3.tree
5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport antlr3.debug
6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport testbase
7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport sys
8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport threading
9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport socket
10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport errno
11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport time
12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverclass Debugger(threading.Thread):
14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def __init__(self, port):
15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        super(Debugger, self).__init__()
16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.events = []
17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.success = False
18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.port = port
19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def run(self):
21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        # create listening socket
22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        s = None
23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        tstart = time.time()
24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        while time.time() - tstart < 10:
25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            try:
26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                s.connect(('127.0.0.1', self.port))
28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                break
29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            except socket.error, exc:
30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                if exc.args[0] != errno.ECONNREFUSED:
31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    raise
32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                time.sleep(0.1)
33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        if s is None:
35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            self.events.append(['nosocket'])
36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            return
37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        s.setblocking(1)
39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        s.settimeout(10.0)
40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        output = s.makefile('w', 0)
42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        input = s.makefile('r', 0)
43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        try:
45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            # handshake
46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            l = input.readline().strip()
47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            assert l == 'ANTLR 2'
48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            l = input.readline().strip()
49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            assert l.startswith('grammar "')
50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            output.write('ACK\n')
52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            output.flush()
53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            while True:
55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                event = input.readline().strip()
56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                self.events.append(event.split('\t'))
57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                output.write('ACK\n')
59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                output.flush()
60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                if event == 'terminate':
62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    self.success = True
63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    break
64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        except socket.timeout:
66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            self.events.append(['timeout'])
67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        except socket.error, exc:
68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            self.events.append(['socketerror', exc.args])
69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        s.close()
71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverclass T(testbase.ANTLRTest):
74324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def execParser(self, grammar, grammarEntry, input, listener,
75324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                   parser_args={}):
76324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        if listener is None:
77324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            port = 49100
78324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            debugger = Debugger(port)
79324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            debugger.start()
80324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            # TODO(pink): install alarm, so it doesn't hang forever in case of a bug
81324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
82324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        else:
83324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            port = None
84324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
85324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        try:
86324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            lexerCls, parserCls = self.compileInlineGrammar(
87324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                grammar, options='-debug')
88324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
89324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            cStream = antlr3.StringStream(input)
90324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            lexer = lexerCls(cStream)
91324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            tStream = antlr3.CommonTokenStream(lexer)
92324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            parser = parserCls(tStream, dbg=listener, port=port, **parser_args)
93324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            getattr(parser, grammarEntry)()
94324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
95324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        finally:
96324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            if listener is None:
97324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                debugger.join()
98324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                return debugger
99324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
100324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testBasicParser(self):
101324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
102324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
103324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
104324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
105324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
106324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
107324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID EOF;
108324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
109324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
110324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
111324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
112324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        listener = antlr3.debug.RecordDebugEventListener()
113324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
114324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.execParser(
115324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
116324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a",
117324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=listener)
118324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
119324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        # We only check that some LT events are present. How many is subject
120324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        # to change (at the time of writing there are two, which is one too
121324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        # many).
122324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        lt_events = [event for event in listener.events
123324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     if event.startswith("LT ")]
124324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertNotEqual(lt_events, [])
125324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
126324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        # For the rest, filter out LT events to get a reliable test.
127324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = ["enterRule a",
128324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    "location 6:1",
129324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    "location 6:5",
130324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    "location 6:8",
131324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    "location 6:11",
132324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    "exitRule a"]
133324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        found = [event for event in listener.events
134324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                 if not event.startswith("LT ")]
135324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(found, expected)
136324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
137324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testSocketProxy(self):
138324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
139324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
140324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
141324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
142324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
143324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
144324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID EOF;
145324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
146324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
147324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
148324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
149324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
150324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
151324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a",
152324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
153324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
154324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
155324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
156324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
157324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
158324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
159324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
160324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
161324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
162324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
163324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
164324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
165324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '-1', '-1', '0', '1', '1', '"<EOF>'],
166324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '11'],
167324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
168324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
169324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
170324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
171324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
172324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testRecognitionException(self):
173324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
174324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
175324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
176324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
177324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
178324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
179324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID EOF;
180324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
181324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
182324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
183324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
184324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
185324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
186324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a b",
187324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
188324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
189324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
190324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
191324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
192324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
193324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
194324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
195324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
196324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
197324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '1', '5', '99', '1', '1', '"'],
198324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
199324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '4', '0', '1', '2', '"b'],
200324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '4', '0', '1', '2', '"b'],
201324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '2', '-1', '-1', '0', '1', '3', '"<EOF>'],
202324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '4', '0', '1', '2', '"b'],
203324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '4', '0', '1', '2', '"b'],
204324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['beginResync'],
205324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '2', '4', '0', '1', '2', '"b'],
206324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['endResync'],
207324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exception', 'UnwantedTokenException', '2', '1', '2'],
208324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '3', '"<EOF>'],
209324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '-1', '-1', '0', '1', '3', '"<EOF>'],
210324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '11'],
211324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
212324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
213324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
214324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
215324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
216324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
217324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testSemPred(self):
218324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
219324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
220324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
221324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
222324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
223324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
224324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : {True}? ID EOF;
225324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
226324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
227324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
228324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
229324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
230324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
231324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a",
232324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
233324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
234324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
235324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
236324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
237324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
238324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
239324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['semanticPredicate', '1', 'True'],
240324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '13'],
241324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
242324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
243324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
244324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '16'],
245324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
246324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
247324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '-1', '-1', '0', '1', '1', '"<EOF>'],
248324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '19'],
249324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
250324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
251324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
252324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
253324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
254324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
255324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testPositiveClosureBlock(self):
256324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
257324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
258324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
259324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
260324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
261324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
262324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID ( ID | INT )+ EOF;
263324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
264324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
265324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
266324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
267324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
268324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
269324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
270324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a 1 b c 3",
271324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
272324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
273324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
274324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
275324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
276324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
277324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
278324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
279324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
280324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
281324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '1', '6', '99', '1', '1', '"'],
282324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
283324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterSubRule', '1'],
284324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
285324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '5', '0', '1', '2', '"1'],
286324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
287324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
288324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
289324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '5', '0', '1', '2', '"1'],
290324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '2', '5', '0', '1', '2', '"1'],
291324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '3', '6', '99', '1', '3', '"'],
292324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
293324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '4', '4', '0', '1', '4', '"b'],
294324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
295324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
296324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
297324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '4', '4', '0', '1', '4', '"b'],
298324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '4', '4', '0', '1', '4', '"b'],
299324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '5', '6', '99', '1', '5', '"'],
300324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
301324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '6', '4', '0', '1', '6', '"c'],
302324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
303324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
304324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
305324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '6', '4', '0', '1', '6', '"c'],
306324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '6', '4', '0', '1', '6', '"c'],
307324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '7', '6', '99', '1', '7', '"'],
308324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
309324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '8', '5', '0', '1', '8', '"3'],
310324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
311324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
312324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
313324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '8', '5', '0', '1', '8', '"3'],
314324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '8', '5', '0', '1', '8', '"3'],
315324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
316324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '9', '"<EOF>'],
317324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
318324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitSubRule', '1'],
319324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '22'],
320324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '9', '"<EOF>'],
321324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '9', '"<EOF>'],
322324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '-1', '-1', '0', '1', '9', '"<EOF>'],
323324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '25'],
324324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
325324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
326324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
327324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
328324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
329324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
330324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testClosureBlock(self):
331324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
332324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
333324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
334324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
335324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
336324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
337324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID ( ID | INT )* EOF;
338324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
339324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
340324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
341324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
342324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
343324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
344324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
345324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a 1 b c 3",
346324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
347324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
348324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
349324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
350324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
351324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
352324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
353324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
354324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
355324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
356324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '1', '6', '99', '1', '1', '"'],
357324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
358324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterSubRule', '1'],
359324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
360324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '5', '0', '1', '2', '"1'],
361324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
362324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
363324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
364324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '2', '5', '0', '1', '2', '"1'],
365324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '2', '5', '0', '1', '2', '"1'],
366324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '3', '6', '99', '1', '3', '"'],
367324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
368324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '4', '4', '0', '1', '4', '"b'],
369324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
370324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
371324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
372324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '4', '4', '0', '1', '4', '"b'],
373324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '4', '4', '0', '1', '4', '"b'],
374324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '5', '6', '99', '1', '5', '"'],
375324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
376324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '6', '4', '0', '1', '6', '"c'],
377324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
378324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
379324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
380324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '6', '4', '0', '1', '6', '"c'],
381324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '6', '4', '0', '1', '6', '"c'],
382324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeHiddenToken', '7', '6', '99', '1', '7', '"'],
383324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
384324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '8', '5', '0', '1', '8', '"3'],
385324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
386324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
387324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
388324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '8', '5', '0', '1', '8', '"3'],
389324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '8', '5', '0', '1', '8', '"3'],
390324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
391324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '9', '"<EOF>'],
392324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
393324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitSubRule', '1'],
394324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '22'],
395324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '9', '"<EOF>'],
396324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '9', '"<EOF>'],
397324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '-1', '-1', '0', '1', '9', '"<EOF>'],
398324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '25'],
399324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
400324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
401324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
402324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
403324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
404324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
405324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testMismatchedSetException(self):
406324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
407324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
408324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
409324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
410324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
411324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
412324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID ( ID | INT ) EOF;
413324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
414324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
415324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
416324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
417324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
418324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
419324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
420324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a",
421324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
422324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
423324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
424324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
425324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
426324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
427324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
428324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
429324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
430324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
431324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '8'],
432324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
433324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
434324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
435324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exception', 'MismatchedSetException', '1', '1', '1'],
436324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exception', 'MismatchedSetException', '1', '1', '1'],
437324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['beginResync'],
438324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '1', '"<EOF>'],
439324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['endResync'],
440324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '24'],
441324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
442324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
443324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
444324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
445324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
446324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
447324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testBlock(self):
448324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
449324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
450324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
451324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
452324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
453324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
454324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID ( b | c ) EOF;
455324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        b : ID;
456324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        c : INT;
457324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
458324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
459324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
460324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
461324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
462324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
463324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
464324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a 1",
465324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
466324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
467324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
468324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected =  [['enterRule', 'T.g', 'a'],
469324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '1'],
470324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterAlt', '1'],
471324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '5'],
472324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '0', '4', '0', '1', '0', '"a'],
473324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '0', '4', '0', '1', '0', '"a'],
474324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['consumeToken', '0', '4', '0', '1', '0', '"a'],
475324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['consumeHiddenToken', '1', '6', '99', '1', '1', '"'],
476324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '8'],
477324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterSubRule', '1'],
478324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterDecision', '1', '0'],
479324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '2', '5', '0', '1', '2', '"1'],
480324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exitDecision', '1'],
481324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterAlt', '2'],
482324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '14'],
483324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterRule', 'T.g', 'c'],
484324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '8', '1'],
485324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterAlt', '1'],
486324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '8', '5'],
487324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '2', '5', '0', '1', '2', '"1'],
488324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '2', '5', '0', '1', '2', '"1'],
489324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['consumeToken', '2', '5', '0', '1', '2', '"1'],
490324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '8', '8'],
491324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exitRule', 'T.g', 'c'],
492324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exitSubRule', '1'],
493324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '18'],
494324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '-1', '-1', '0', '1', '3', '"<EOF>'],
495324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '-1', '-1', '0', '1', '3', '"<EOF>'],
496324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['consumeToken', '-1', '-1', '0', '1', '3', '"<EOF>'],
497324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '21'],
498324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exitRule', 'T.g', 'a'],
499324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['terminate']]
500324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
501324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
502324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
503324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
504324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testNoViableAlt(self):
505324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
506324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
507324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
508324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
509324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
510324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
511324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ID ( b | c ) EOF;
512324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        b : ID;
513324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        c : INT;
514324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
515324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
516324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        BANG : '!' ;
517324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
518324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
519324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
520324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
521324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
522324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a !",
523324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
524324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
525324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
526324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected =  [['enterRule', 'T.g', 'a'],
527324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '1'],
528324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterAlt', '1'],
529324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '5'],
530324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '0', '5', '0', '1', '0', '"a'],
531324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '0', '5', '0', '1', '0', '"a'],
532324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['consumeToken', '0', '5', '0', '1', '0', '"a'],
533324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['consumeHiddenToken', '1', '7', '99', '1', '1', '"'],
534324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '8'],
535324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterSubRule', '1'],
536324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['enterDecision', '1', '0'],
537324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '2', '4', '0', '1', '2', '"!'],
538324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '2', '4', '0', '1', '2', '"!'],
539324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '2', '4', '0', '1', '2', '"!'],
540324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exception', 'NoViableAltException', '2', '1', '2'],
541324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exitDecision', '1'],
542324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exitSubRule', '1'],
543324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exception', 'NoViableAltException', '2', '1', '2'],
544324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['beginResync'],
545324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '2', '4', '0', '1', '2', '"!'],
546324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['consumeToken', '2', '4', '0', '1', '2', '"!'],
547324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['LT', '1', '-1', '-1', '0', '1', '3', '"<EOF>'],
548324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['endResync'],
549324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['location', '6', '21'],
550324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['exitRule', 'T.g', 'a'],
551324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                     ['terminate']]
552324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
553324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
554324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
555324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
556324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testRuleBlock(self):
557324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
558324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
559324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
560324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
561324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
562324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
563324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : b | c;
564324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        b : ID;
565324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        c : INT;
566324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
567324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
568324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
569324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
570324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
571324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
572324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
573324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="1",
574324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
575324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
576324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
577324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
578324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
579324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
580324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '5', '0', '1', '0', '"1'],
581324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
582324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '2'],
583324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '9'],
584324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterRule', 'T.g', 'c'],
585324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '1'],
586324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
587324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '5'],
588324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '5', '0', '1', '0', '"1'],
589324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '5', '0', '1', '0', '"1'],
590324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '5', '0', '1', '0', '"1'],
591324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '8'],
592324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'c'],
593324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '10'],
594324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
595324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
596324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
597324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
598324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
599324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
600324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testRuleBlockSingleAlt(self):
601324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
602324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
603324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
604324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
605324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
606324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
607324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : b;
608324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        b : ID;
609324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
610324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
611324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
612324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
613324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
614324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
615324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
616324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a",
617324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
618324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
619324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
620324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
621324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
622324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
623324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
624324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterRule', 'T.g', 'b'],
625324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '7', '1'],
626324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
627324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '7', '5'],
628324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
629324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
630324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
631324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '7', '7'],
632324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'b'],
633324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '6'],
634324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
635324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
636324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
637324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
638324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
639324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
640324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testBlockSingleAlt(self):
641324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
642324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
643324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
644324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
645324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
646324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
647324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ( b );
648324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        b : ID;
649324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
650324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
651324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
652324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
653324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
654324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
655324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
656324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a",
657324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
658324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
659324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
660324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
661324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
662324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
663324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
664324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
665324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '7'],
666324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterRule', 'T.g', 'b'],
667324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '7', '1'],
668324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
669324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '7', '5'],
670324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
671324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '4', '0', '1', '0', '"a'],
672324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '4', '0', '1', '0', '"a'],
673324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '7', '7'],
674324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'b'],
675324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '10'],
676324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
677324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
678324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
679324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
680324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
681324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
682324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testDFA(self):
683324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
684324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
685324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
686324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
687324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
688324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
689324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ( b | c ) EOF;
690324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        b : ID* INT;
691324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        c : ID+ BANG;
692324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
693324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
694324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        BANG : '!';
695324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
696324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
697324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
698324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        debugger = self.execParser(
699324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
700324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a!",
701324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=None)
702324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
703324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertTrue(debugger.success)
704324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        expected = [['enterRule', 'T.g', 'a'],
705324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '1'],
706324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
707324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '5'],
708324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterSubRule', '1'],
709324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '1', '0'],
710324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['mark', '0'],
711324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '5', '0', '1', '0', '"a'],
712324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '5', '0', '1', '0', '"a'],
713324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '1', '4', '0', '1', '1', '"!'],
714324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '1', '4', '0', '1', '1', '"!'],
715324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['rewind', '0'],
716324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '1'],
717324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '2'],
718324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '11'],
719324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterRule', 'T.g', 'c'],
720324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '1'],
721324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
722324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '5'],
723324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterSubRule', '3'],
724324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '3', '0'],
725324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '5', '0', '1', '0', '"a'],
726324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '3'],
727324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterAlt', '1'],
728324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '5'],
729324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '5', '0', '1', '0', '"a'],
730324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '0', '5', '0', '1', '0', '"a'],
731324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '0', '5', '0', '1', '0', '"a'],
732324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['enterDecision', '3', '0'],
733324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '1', '4', '0', '1', '1', '"!'],
734324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitDecision', '3'],
735324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitSubRule', '3'],
736324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '9'],
737324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '1', '4', '0', '1', '1', '"!'],
738324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '1', '4', '0', '1', '1', '"!'],
739324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '1', '4', '0', '1', '1', '"!'],
740324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '8', '13'],
741324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'c'],
742324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitSubRule', '1'],
743324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '15'],
744324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '2', '"<EOF>'],
745324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['LT', '1', '-1', '-1', '0', '1', '2', '"<EOF>'],
746324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['consumeToken', '-1', '-1', '0', '1', '2', '"<EOF>'],
747324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['location', '6', '18'],
748324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['exitRule', 'T.g', 'a'],
749324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                    ['terminate']]
750324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
751324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.assertListEqual(debugger.events, expected)
752324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
753324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
754324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testBasicAST(self):
755324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar = textwrap.dedent(
756324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        r'''
757324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        grammar T;
758324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        options {
759324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            language=Python;
760324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            output=AST;
761324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        }
762324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        a : ( b | c ) EOF!;
763324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        b : ID* INT -> ^(INT ID*);
764324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        c : ID+ BANG -> ^(BANG ID+);
765324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ID : 'a'..'z'+ ;
766324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        INT : '0'..'9'+ ;
767324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        BANG : '!';
768324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        WS : (' '|'\n') {$channel=HIDDEN;} ;
769324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        ''')
770324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
771324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        listener = antlr3.debug.RecordDebugEventListener()
772324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
773324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.execParser(
774324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            grammar, 'a',
775324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            input="a!",
776324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            listener=listener)
777324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
778324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        # don't check output for now (too dynamic), I'm satisfied if it
779324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        # doesn't crash
780324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
781324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
782324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverif __name__ == '__main__':
783324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    unittest.main()
784