1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport antlr3
2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport testbase
3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport unittest
4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverclass t032subrulePredict(testbase.ANTLRTest):
7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def setUp(self):
8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.compileGrammar()
9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def parserClass(self, base):
12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        class TParser(base):
13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            def recover(self, input, re):
14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                # no error recovery yet, just crash!
15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver                raise
16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        return TParser
18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testValid1(self):
21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        cStream = antlr3.StringStream(
22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            'BEGIN A END'
23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            )
24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        lexer = self.getLexer(cStream)
26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        tStream = antlr3.CommonTokenStream(lexer)
27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        parser = self.getParser(tStream)
28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        events = parser.a()
29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    @testbase.broken("DFA tries to look beyond end of rule b", Exception)
32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testValid2(self):
33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        cStream = antlr3.StringStream(
34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            ' A'
35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver            )
36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        lexer = self.getLexer(cStream)
38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        tStream = antlr3.CommonTokenStream(lexer)
39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        parser = self.getParser(tStream)
40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        events = parser.b()
41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverif __name__ == '__main__':
44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    unittest.main()
45