1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport os
2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport sys
3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport antlr3
4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport testbase
5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport unittest
6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverclass t021hoist(testbase.ANTLRTest):
9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def setUp(self):
10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        self.compileGrammar()
11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testValid1(self):
14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        cStream = antlr3.StringStream('enum')
15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        lexer = self.getLexer(cStream)
16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        tStream = antlr3.CommonTokenStream(lexer)
17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        parser = self.getParser(tStream)
18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        parser.enableEnum = True
19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        enumIs = parser.stat()
20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        assert enumIs == 'keyword', repr(enumIs)
22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    def testValid2(self):
25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        cStream = antlr3.StringStream('enum')
26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        lexer = self.getLexer(cStream)
27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        tStream = antlr3.CommonTokenStream(lexer)
28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        parser = self.getParser(tStream)
29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        parser.enableEnum = False
30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        enumIs = parser.stat()
31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver        assert enumIs == 'ID', repr(enumIs)
33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverif __name__ == '__main__':
37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver    unittest.main()
38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver
39