disagreement/tests/test_member.py

23 lines
620 B
Python

from disagreement.models import Member
def _make_member(member_id: str, username: str, nick: str | None):
data = {
"user": {"id": member_id, "username": username, "discriminator": "0001"},
"joined_at": "t",
"roles": [],
}
if nick is not None:
data["nick"] = nick
return Member(data, client_instance=None)
def test_display_name_prefers_nick():
member = _make_member("1", "u", "nickname")
assert member.display_name == "nickname"
def test_display_name_falls_back_to_username():
member = _make_member("2", "u2", None)
assert member.display_name == "u2"