mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
66 lines
3 KiB
Python
66 lines
3 KiB
Python
|
import datetime
|
||
|
|
from aiogram.types import Chat, Message
|
||
|
|
from tests.mocked_bot import MockedBot
|
||
|
|
|
||
|
|
|
||
|
|
class TestGetMessageUrl:
|
||
|
|
def test_method(self, bot: MockedBot):
|
||
|
|
fake_message_id = 10
|
||
|
|
|
||
|
|
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,
|
||
|
|
)
|
||
|
|
|
||
|
|
# get_url() in PM must return None
|
||
|
|
assert fake_message.get_url() is None
|
||
|
|
|
||
|
|
# 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,
|
||
|
|
)
|
||
|
|
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,
|
||
|
|
)
|
||
|
|
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_message = Message(
|
||
|
|
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}"
|
||
|
|
|
||
|
|
# 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,
|
||
|
|
)
|
||
|
|
assert fake_message.chat.shifted_id == 1234567890
|
||
|
|
assert fake_message.get_url() == f"https://t.me/c/1234567890/{fake_message_id}"
|
||
|
|
|
||
|
|
# get_url() in channel with username must return link with username or private link
|
||
|
|
# 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,
|
||
|
|
)
|
||
|
|
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}"
|