def test_parse_statements(self):
"""
(letStatement | ifStatement | whileStatement | doStatement | returnStatement)*
"""
tokens = (
Token('keyword', 'let'),
Token('keyword', 'if'),
Token('keyword', 'while'),
Token('keyword', 'do'),
Token('keyword', 'return'),
Token('symbol', '}') # Expect unparsed
)
expected = Token('statements', [
Token('dummy', 'dummy'),
Token('dummy', 'dummy'),
Token('dummy', 'dummy'),
Token('dummy', 'dummy'),
Token('dummy', 'dummy'),
])
parser = Parser(tokens)
parser.parse_let_statement = self._mock_parse(parser)
parser.parse_if_statement = self._mock_parse(parser)
parser.parse_while_statement = self._mock_parse(parser)
parser.parse_do_statement = self._mock_parse(parser)
parser.parse_return_statement = self._mock_parse(parser)
actual = parser.parse_statements()
self.assertEqual(expected, actual)
def test_parse_return_statement_empty(self):
"""
'return' expression? ';'
"""
tokens = (
Token('keyword', 'return'),
Token('symbol', ';'),
)
expected = Token('returnStatement', [
Token('keyword', 'return'),
Token('symbol', ';'),
])
parser = Parser(tokens)
actual = parser.parse_return_statement()
self.assertEqual(expected, actual)
def test_parse_return_statement(self):
"""
'return' expression? ';'
"""
tokens = (
Token('keyword', 'return'),
Token('dummy', 'dummy'), # Dummy expression
Token('symbol', ';'),
)
expected = Token('returnStatement', [
Token('keyword', 'return'),
Token('dummy', 'dummy'), # Dummy expression
Token('symbol', ';'),
])
parser = Parser(tokens)
parser.parse_expression = self._mock_parse(parser)
actual = parser.parse_return_statement()
self.assertEqual(expected, actual)
