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_check_whitespace_method_with_text(self):
lexer = Lexer("> text <")
self.assertIsInstance(lexer.get_next_token(), CloseOfTagToken)
flag, buffer = lexer.is_next_nonempty_char_an_open_of_tag()
self.assertEqual(flag, False)
self.assertEqual(buffer, " ")
