atst/tests/utils/test_flash.py
2020-03-04 11:51:15 -05:00

11 lines
332 B
Python

from atat.utils.flash import formatted_flash as flash
from flask import get_flashed_messages
def test_flash_message():
flash("revoked_portfolio_access", member_name="Lando")
messages = get_flashed_messages()
message_info = messages[0]
assert "message" in message_info
assert "Lando" in message_info["message"]