1*16467b97STreehugger Robotimport antlr3 2*16467b97STreehugger Robotimport testbase 3*16467b97STreehugger Robotimport unittest 4*16467b97STreehugger Robot 5*16467b97STreehugger Robotclass t014parser(testbase.ANTLRTest): 6*16467b97STreehugger Robot def setUp(self): 7*16467b97STreehugger Robot self.compileGrammar() 8*16467b97STreehugger Robot 9*16467b97STreehugger Robot 10*16467b97STreehugger Robot def testValid(self): 11*16467b97STreehugger Robot cStream = antlr3.StringStream('var foobar; gnarz(); var blupp; flupp ( ) ;') 12*16467b97STreehugger Robot lexer = self.getLexer(cStream) 13*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 14*16467b97STreehugger Robot parser = self.getParser(tStream) 15*16467b97STreehugger Robot parser.document() 16*16467b97STreehugger Robot 17*16467b97STreehugger Robot self.assertEqual(parser.reportedErrors, []) 18*16467b97STreehugger Robot self.assertEqual(parser.events, 19*16467b97STreehugger Robot [('decl', 'foobar'), ('call', 'gnarz'), 20*16467b97STreehugger Robot ('decl', 'blupp'), ('call', 'flupp')]) 21*16467b97STreehugger Robot 22*16467b97STreehugger Robot 23*16467b97STreehugger Robot def testMalformedInput1(self): 24*16467b97STreehugger Robot cStream = antlr3.StringStream('var; foo();') 25*16467b97STreehugger Robot lexer = self.getLexer(cStream) 26*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 27*16467b97STreehugger Robot parser = self.getParser(tStream) 28*16467b97STreehugger Robot 29*16467b97STreehugger Robot parser.document() 30*16467b97STreehugger Robot 31*16467b97STreehugger Robot # FIXME: currently strings with formatted errors are collected 32*16467b97STreehugger Robot # can't check error locations yet 33*16467b97STreehugger Robot self.assertEqual(len(parser.reportedErrors), 1, parser.reportedErrors) 34*16467b97STreehugger Robot self.assertEqual(parser.events, []) 35*16467b97STreehugger Robot 36*16467b97STreehugger Robot 37*16467b97STreehugger Robot def testMalformedInput2(self): 38*16467b97STreehugger Robot cStream = antlr3.StringStream('var foobar(); gnarz();') 39*16467b97STreehugger Robot lexer = self.getLexer(cStream) 40*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 41*16467b97STreehugger Robot parser = self.getParser(tStream) 42*16467b97STreehugger Robot 43*16467b97STreehugger Robot parser.document() 44*16467b97STreehugger Robot 45*16467b97STreehugger Robot # FIXME: currently strings with formatted errors are collected 46*16467b97STreehugger Robot # can't check error locations yet 47*16467b97STreehugger Robot self.assertEqual(len(parser.reportedErrors), 1, parser.reportedErrors) 48*16467b97STreehugger Robot self.assertEqual(parser.events, [('call', 'gnarz')]) 49*16467b97STreehugger Robot 50*16467b97STreehugger Robot 51*16467b97STreehugger Robot def testMalformedInput3(self): 52*16467b97STreehugger Robot cStream = antlr3.StringStream('gnarz(; flupp();') 53*16467b97STreehugger Robot lexer = self.getLexer(cStream) 54*16467b97STreehugger Robot tStream = antlr3.CommonTokenStream(lexer) 55*16467b97STreehugger Robot parser = self.getParser(tStream) 56*16467b97STreehugger Robot 57*16467b97STreehugger Robot parser.document() 58*16467b97STreehugger Robot 59*16467b97STreehugger Robot # FIXME: currently strings with formatted errors are collected 60*16467b97STreehugger Robot # can't check error locations yet 61*16467b97STreehugger Robot self.assertEqual(len(parser.reportedErrors), 1, parser.reportedErrors) 62*16467b97STreehugger Robot self.assertEqual(parser.events, [('call', 'gnarz'), ('call', 'flupp')]) 63*16467b97STreehugger Robot 64*16467b97STreehugger Robot 65*16467b97STreehugger Robotif __name__ == '__main__': 66*16467b97STreehugger Robot unittest.main() 67