xref: /aosp_15_r20/external/antlr/runtime/Python/tests/t037rulePropertyRef.py (revision 16467b971bd3e2009fad32dd79016f2c7e421deb)
1*16467b97STreehugger Robotimport antlr3
2*16467b97STreehugger Robotimport testbase
3*16467b97STreehugger Robotimport unittest
4*16467b97STreehugger Robot
5*16467b97STreehugger Robot
6*16467b97STreehugger Robotclass t037rulePropertyRef(testbase.ANTLRTest):
7*16467b97STreehugger Robot    def setUp(self):
8*16467b97STreehugger Robot        self.compileGrammar()
9*16467b97STreehugger Robot
10*16467b97STreehugger Robot
11*16467b97STreehugger Robot    def lexerClass(self, base):
12*16467b97STreehugger Robot        class TLexer(base):
13*16467b97STreehugger Robot            def recover(self, input, re):
14*16467b97STreehugger Robot                # no error recovery yet, just crash!
15*16467b97STreehugger Robot                raise
16*16467b97STreehugger Robot
17*16467b97STreehugger Robot        return TLexer
18*16467b97STreehugger Robot
19*16467b97STreehugger Robot
20*16467b97STreehugger Robot    def parserClass(self, base):
21*16467b97STreehugger Robot        class TParser(base):
22*16467b97STreehugger Robot            def recover(self, input, re):
23*16467b97STreehugger Robot                # no error recovery yet, just crash!
24*16467b97STreehugger Robot                raise
25*16467b97STreehugger Robot
26*16467b97STreehugger Robot        return TParser
27*16467b97STreehugger Robot
28*16467b97STreehugger Robot
29*16467b97STreehugger Robot    def testValid1(self):
30*16467b97STreehugger Robot        cStream = antlr3.StringStream('   a a a a  ')
31*16467b97STreehugger Robot
32*16467b97STreehugger Robot        lexer = self.getLexer(cStream)
33*16467b97STreehugger Robot        tStream = antlr3.CommonTokenStream(lexer)
34*16467b97STreehugger Robot        parser = self.getParser(tStream)
35*16467b97STreehugger Robot        start, stop, text = parser.a().bla
36*16467b97STreehugger Robot
37*16467b97STreehugger Robot        # first token of rule b is the 2nd token (counting hidden tokens)
38*16467b97STreehugger Robot        assert start.index == 1, start
39*16467b97STreehugger Robot
40*16467b97STreehugger Robot        # first token of rule b is the 7th token (counting hidden tokens)
41*16467b97STreehugger Robot        assert stop.index == 7, stop
42*16467b97STreehugger Robot
43*16467b97STreehugger Robot        assert text == "a a a a", text
44*16467b97STreehugger Robot
45*16467b97STreehugger Robot
46*16467b97STreehugger Robotif __name__ == '__main__':
47*16467b97STreehugger Robot    unittest.main()
48