diff --git a/tests/test_api/test_methods/test_get_url.py b/tests/test_api/test_methods/test_get_url.py index f9ac8f6f..fbdbfa22 100644 --- a/tests/test_api/test_methods/test_get_url.py +++ b/tests/test_api/test_methods/test_get_url.py @@ -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}" + )