def test_last_flashed_messages_none(self, mockflash): """Test last_flashed_message returns the none.""" messages = [] mockflash.return_value = messages msg = util.last_flashed_message() err_msg = "It should be None" assert msg is None, err_msg
def test_last_flashed_messages(self, mockflash): """Test last_flashed_message returns the last one.""" messages = ['foo', 'bar'] mockflash.return_value = messages msg = util.last_flashed_message() err_msg = "It should be the last message" assert msg == messages[-1], err_msg