def _create_lexer_mock(self, return_values, flag): return_values.append(EndOfTextToken()) lexer = Lexer(None) lexer.get_next_token = Mock() lexer.is_next_nonempty_char_an_open_of_tag = Mock() lexer.get_text_until_open_of_tag = Mock() lexer.get_comment = Mock() lexer.get_next_token.side_effect = return_values lexer.is_next_nonempty_char_an_open_of_tag.side_effect = [(flag, ""), (False, "")] lexer.get_text_until_open_of_tag.side_effect = ["text"] lexer.get_comment.side_effect = ["comment"] return lexer
def test_get_string_method(self): lexer = Lexer("> text <") self.assertIsInstance(lexer.get_next_token(), CloseOfTagToken) text = lexer.get_text_until_open_of_tag() self.assertEqual(text, " text ") self.assertIsInstance(lexer.get_next_token(), OpenOfTagToken)