python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Code format with black

Browse source
This commit is contained in:
Groosha 2021-05-18 23:26:03 +03:00
parent 1bdc5d40e9
commit 2f462ae6a8
1 changed files with 35 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

50
tests/test_api/test_methods/test_get_url.py
View file

@ -9,8 +9,10 @@ class TestGetMessageUrl:
fake_chat_pm = Chat(id=123456, username="username", type="private")
fake_message = Message(
message_id=fake_message_id, date=datetime.datetime.now(),
text="test", chat=fake_chat_pm,
message_id=fake_message_id,
date=datetime.datetime.now(),
text="test",
chat=fake_chat_pm,
)
# get_url() in PM must return None
@ -19,36 +21,49 @@ class TestGetMessageUrl:
# get_url() in basic group must return None
fake_chat_group = Chat(id=-123456, username="username", type="group")
fake_message = Message(
message_id=fake_message_id, date=datetime.datetime.now(),
text="test", chat=fake_chat_group,
message_id=fake_message_id,
date=datetime.datetime.now(),
text="test",
chat=fake_chat_group,
)
assert fake_message.get_url() is None
# get_url() in supergroup without username must return private link
fake_chat_supergroup_no_username = Chat(id=-1001234567890, type="supergroup")
fake_message = Message(
message_id=fake_message_id, date=datetime.datetime.now(),
text="test", chat=fake_chat_supergroup_no_username,
message_id=fake_message_id,
date=datetime.datetime.now(),
text="test",
chat=fake_chat_supergroup_no_username,
)
assert fake_message.chat.shifted_id == 1234567890
assert fake_message.get_url() == f"https://t.me/c/1234567890/{fake_message_id}"
# get_url() in supergroup with username must return link with username or private link
# if force_private set to True
fake_chat_supergroup_username = Chat(id=-1001234567890, username="username", type="supergroup")
fake_chat_supergroup_username = Chat(
id=-1001234567890, username="username", type="supergroup"
)
fake_message = Message(
message_id=fake_message_id, date=datetime.datetime.now(),
text="test", chat=fake_chat_supergroup_username,
message_id=fake_message_id,
date=datetime.datetime.now(),
text="test",
chat=fake_chat_supergroup_username,
)
assert fake_message.chat.shifted_id == 1234567890
assert fake_message.get_url() == f"https://t.me/username/{fake_message_id}"
assert fake_message.get_url(force_private=True) == f"https://t.me/c/1234567890/{fake_message_id}"
assert (
fake_message.get_url(force_private=True)
== f"https://t.me/c/1234567890/{fake_message_id}"
)
# get_url() in channel without username must return private link
fake_chat_channel_no_username = Chat(id=-1001234567890, type="channel")
fake_message = Message(
message_id=fake_message_id, date=datetime.datetime.now(),
text="test", chat=fake_chat_channel_no_username,
message_id=fake_message_id,
date=datetime.datetime.now(),
text="test",
chat=fake_chat_channel_no_username,
)
assert fake_message.chat.shifted_id == 1234567890
assert fake_message.get_url() == f"https://t.me/c/1234567890/{fake_message_id}"
@ -57,9 +72,14 @@ class TestGetMessageUrl:
# if force_private set to True
fake_chat_channel_username = Chat(id=-1001234567890, username="username", type="channel")
fake_message = Message(
message_id=fake_message_id, date=datetime.datetime.now(),
text="test", chat=fake_chat_channel_username,
message_id=fake_message_id,
date=datetime.datetime.now(),
text="test",
chat=fake_chat_channel_username,
)
assert fake_message.chat.shifted_id == 1234567890
assert fake_message.get_url() == f"https://t.me/username/{fake_message_id}"
assert fake_message.get_url(force_private=True) == f"https://t.me/c/1234567890/{fake_message_id}"
assert (
fake_message.get_url(force_private=True)
== f"https://t.me/c/1234567890/{fake_message_id}"
)