update test

This commit is contained in:
sheldygg 2022-10-31 18:32:59 +02:00
parent 294210e44a
commit a14a07a6c6

View file

@ -20,31 +20,33 @@ class TestUser:
assert user.full_name == result
@pytest.mark.parametrize(
"first,last,result",
"first,last,name",
[
["User", None, "User"],
["", None, ""],
[" ", None, " "],
["User", "Name", "User Name"],
["User", " ", "User "],
[" ", " ", " "],
["User", None, "bebra"],
["", None, "telegram"],
[" ", None, "queue🤬"],
["User", "Name", "Alex"],
["User", " ", "aslo "],
[" ", " ", "telebot"],
],
)
def test_get_mention_markdown(self, first: str, last: str, result: bool):
def test_get_mention_markdown(self, first: str, last: str, name: str):
user = User(id=42, is_bot=False, first_name=first, last_name=last)
assert user.mention_markdown()
assert user.mention_markdown() == f"[{user.full_name}](tg://user?id=42)"
assert user.mention_markdown(name=name) == f"[{name}](tg://user?id=42)"
@pytest.mark.parametrize(
"first,last,result",
"first,last,name",
[
["User", None, "User"],
["", None, ""],
[" ", None, " "],
["User", "Name", "User Name"],
["User", " ", "User "],
[" ", " ", " "],
["User", None, "bebra"],
["", None, "telegram"],
[" ", None, "queue🤬"],
["User", "Name", "Alex"],
["User", " ", "aslo "],
[" ", " ", "telebot"],
],
)
def test_get_mention_html(self, first: str, last: str, result: bool):
def test_get_mention_html(self, first: str, last: str, name: str):
user = User(id=42, is_bot=False, first_name=first, last_name=last)
assert user.mention_html()
assert user.mention_html() == f'<a href="tg://user?id=42">{user.full_name}</a>'
assert user.mention_html(name=name) == f'<a href="tg://user?id=42">{name}</a>'