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

upgrade tests for message content_types and sent_copy

Browse source
This commit is contained in:
Suren Khorenyan 2023-12-31 01:03:11 +03:00
parent ff28f19953
commit bfaf201fc8
1 changed files with 181 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

237
tests/test_api/test_types/test_message.py
View file

@ -51,6 +51,12 @@ from aiogram.types import (
ForumTopicEdited, ForumTopicEdited,
ForumTopicReopened, ForumTopicReopened,
Game, Game,
GeneralForumTopicHidden,
GeneralForumTopicUnhidden,
Giveaway,
GiveawayCompleted,
GiveawayCreated,
GiveawayWinners,
InlineKeyboardButton, InlineKeyboardButton,
InlineKeyboardMarkup, InlineKeyboardMarkup,
InputMediaPhoto, InputMediaPhoto,
@ -69,6 +75,7 @@ from aiogram.types import (
SuccessfulPayment, SuccessfulPayment,
User, User,
UserShared, UserShared,
UsersShared,
Venue, Venue,
Video, Video,
VideoChatEnded, VideoChatEnded,
@ -78,6 +85,7 @@ from aiogram.types import (
VideoNote, VideoNote,
Voice, Voice,
WebAppData, WebAppData,
WriteAccessAllowed,
) )
from aiogram.types.message import ContentType, Message from aiogram.types.message import ContentType, Message
@ -446,13 +454,23 @@ TEST_FORUM_TOPIC_REOPENED = Message(
from_user=User(id=42, is_bot=False, first_name="Test"), from_user=User(id=42, is_bot=False, first_name="Test"),
forum_topic_reopened=ForumTopicReopened(), forum_topic_reopened=ForumTopicReopened(),
) )
TEST_USER_SHARED = Message( TEST_MESSAGE_USER_SHARED = Message(
message_id=42, message_id=42,
date=datetime.datetime.now(), date=datetime.datetime.now(),
chat=Chat(id=42, type="private"), chat=Chat(id=42, type="private"),
from_user=User(id=42, is_bot=False, first_name="Test"), from_user=User(id=42, is_bot=False, first_name="Test"),
user_shared=UserShared(request_id=42, user_id=42), user_shared=UserShared(request_id=42, user_id=42),
) )
TEST_MESSAGE_USERS_SHARED = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
users_shared=UsersShared(
request_id=0,
user_ids=[1, 2],
),
)
TEST_CHAT_SHARED = Message( TEST_CHAT_SHARED = Message(
message_id=42, message_id=42,
date=datetime.datetime.now(), date=datetime.datetime.now(),
@ -469,6 +487,73 @@ TEST_MESSAGE_STORY = Message(
forward_signature="Test", forward_signature="Test",
forward_date=datetime.datetime.now(), forward_date=datetime.datetime.now(),
) )
TEST_MESSAGE_GIVEAWAY = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
giveaway=Giveaway(
chats=[Chat(id=42, type="private")],
winners_selection_date=datetime.datetime.now() + datetime.timedelta(days=7),
winner_count=10,
),
)
TEST_MESSAGE_GIVEAWAY_CREATED = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
giveaway_created=GiveawayCreated(),
)
TEST_MESSAGE_GIVEAWAY_WINNERS = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
giveaway_winners=GiveawayWinners(
chat=Chat(id=77, type="private"),
giveaway_message_id=123,
winners_selection_date=datetime.datetime.now(),
winner_count=1,
winners=[User(id=42, is_bot=False, first_name="Test")],
),
)
TEST_MESSAGE_GIVEAWAY_COMPLETED = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
giveaway_completed=GiveawayCompleted(winner_count=10),
)
TEST_MESSAGE_HAS_MEDIA_SPOILER = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
has_media_spoiler=True,
)
TEST_MESSAGE_GENERAL_FORUM_TOPIC_HIDDEN = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
general_forum_topic_hidden=GeneralForumTopicHidden(),
)
TEST_MESSAGE_GENERAL_FORUM_TOPIC_UNHIDDEN = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
general_forum_topic_unhidden=GeneralForumTopicUnhidden(),
)
TEST_MESSAGE_WRITE_ACCESS_ALLOWED = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=None,
write_access_allowed=WriteAccessAllowed(),
)
TEST_MESSAGE_UNKNOWN = Message( TEST_MESSAGE_UNKNOWN = Message(
message_id=42, message_id=42,
date=datetime.datetime.now(), date=datetime.datetime.now(),
@ -524,27 +609,107 @@ MESSAGES_AND_CONTENT_TYPES = [
[TEST_FORUM_TOPIC_EDITED, ContentType.FORUM_TOPIC_EDITED], [TEST_FORUM_TOPIC_EDITED, ContentType.FORUM_TOPIC_EDITED],
[TEST_FORUM_TOPIC_CLOSED, ContentType.FORUM_TOPIC_CLOSED], [TEST_FORUM_TOPIC_CLOSED, ContentType.FORUM_TOPIC_CLOSED],
[TEST_FORUM_TOPIC_REOPENED, ContentType.FORUM_TOPIC_REOPENED], [TEST_FORUM_TOPIC_REOPENED, ContentType.FORUM_TOPIC_REOPENED],
[TEST_USER_SHARED, ContentType.USER_SHARED], [TEST_MESSAGE_USER_SHARED, ContentType.USER_SHARED],
[TEST_MESSAGE_USERS_SHARED, ContentType.USERS_SHARED],
[TEST_CHAT_SHARED, ContentType.CHAT_SHARED], [TEST_CHAT_SHARED, ContentType.CHAT_SHARED],
[TEST_MESSAGE_STORY, ContentType.STORY], [TEST_MESSAGE_STORY, ContentType.STORY],
[TEST_MESSAGE_GIVEAWAY, ContentType.GIVEAWAY],
[TEST_MESSAGE_GIVEAWAY_CREATED, ContentType.GIVEAWAY_CREATED],
[TEST_MESSAGE_GIVEAWAY_WINNERS, ContentType.GIVEAWAY_WINNERS],
[TEST_MESSAGE_GIVEAWAY_COMPLETED, ContentType.GIVEAWAY_COMPLETED],
[TEST_MESSAGE_HAS_MEDIA_SPOILER, ContentType.HAS_MEDIA_SPOILER],
[TEST_MESSAGE_GENERAL_FORUM_TOPIC_HIDDEN, ContentType.GENERAL_FORUM_TOPIC_HIDDEN],
[TEST_MESSAGE_GENERAL_FORUM_TOPIC_UNHIDDEN, ContentType.GENERAL_FORUM_TOPIC_UNHIDDEN],
[TEST_MESSAGE_WRITE_ACCESS_ALLOWED, ContentType.WRITE_ACCESS_ALLOWED],
[TEST_MESSAGE_UNKNOWN, ContentType.UNKNOWN], [TEST_MESSAGE_UNKNOWN, ContentType.UNKNOWN],
] ]
def test_all_known_content_type_have_example_messages(): MESSAGES_AND_COPY_METHODS = [
""" [TEST_MESSAGE_TEXT, SendMessage],
Test if all ContentType options have example messages. [TEST_MESSAGE_AUDIO, SendAudio],
[TEST_MESSAGE_ANIMATION, SendAnimation],
[TEST_MESSAGE_DOCUMENT, SendDocument],
[TEST_MESSAGE_GAME, None],
[TEST_MESSAGE_PHOTO, SendPhoto],
[TEST_MESSAGE_STICKER, SendSticker],
[TEST_MESSAGE_VIDEO, SendVideo],
[TEST_MESSAGE_VIDEO_NOTE, SendVideoNote],
[TEST_MESSAGE_VOICE, SendVoice],
[TEST_MESSAGE_CONTACT, SendContact],
[TEST_MESSAGE_VENUE, SendVenue],
[TEST_MESSAGE_LOCATION, SendLocation],
[TEST_MESSAGE_STORY, ForwardMessage],
[TEST_MESSAGE_NEW_CHAT_MEMBERS, None],
[TEST_MESSAGE_LEFT_CHAT_MEMBER, None],
[TEST_MESSAGE_INVOICE, None],
[TEST_MESSAGE_SUCCESSFUL_PAYMENT, None],
[TEST_MESSAGE_CONNECTED_WEBSITE, None],
[TEST_MESSAGE_MIGRATE_FROM_CHAT_ID, None],
[TEST_MESSAGE_MIGRATE_TO_CHAT_ID, None],
[TEST_MESSAGE_PINNED_MESSAGE, None],
[TEST_MESSAGE_NEW_CHAT_TITLE, None],
[TEST_MESSAGE_NEW_CHAT_PHOTO, None],
[TEST_MESSAGE_DELETE_CHAT_PHOTO, None],
[TEST_MESSAGE_GROUP_CHAT_CREATED, None],
[TEST_MESSAGE_SUPERGROUP_CHAT_CREATED, None],
[TEST_MESSAGE_CHANNEL_CHAT_CREATED, None],
[TEST_MESSAGE_PASSPORT_DATA, None],
[TEST_MESSAGE_PROXIMITY_ALERT_TRIGGERED, None],
[TEST_MESSAGE_POLL, SendPoll],
[TEST_MESSAGE_MESSAGE_AUTO_DELETE_TIMER_CHANGED, None],
[TEST_MESSAGE_VIDEO_CHAT_STARTED, None],
[TEST_MESSAGE_VIDEO_CHAT_ENDED, None],
[TEST_MESSAGE_VIDEO_CHAT_PARTICIPANTS_INVITED, None],
[TEST_MESSAGE_DICE, SendDice],
[TEST_MESSAGE_USER_SHARED, None],
[TEST_CHAT_SHARED, None],
[TEST_MESSAGE_GIVEAWAY_COMPLETED, None],
[TEST_MESSAGE_HAS_MEDIA_SPOILER, None],
[TEST_MESSAGE_WEB_APP_DATA, None],
[TEST_FORUM_TOPIC_CREATED, None],
[TEST_FORUM_TOPIC_EDITED, None],
[TEST_FORUM_TOPIC_CLOSED, None],
[TEST_FORUM_TOPIC_REOPENED, None],
[TEST_MESSAGE_GENERAL_FORUM_TOPIC_HIDDEN, None],
[TEST_MESSAGE_GENERAL_FORUM_TOPIC_UNHIDDEN, None],
[TEST_MESSAGE_GIVEAWAY_CREATED, None],
[TEST_MESSAGE_USERS_SHARED, None],
[TEST_MESSAGE_VIDEO_CHAT_SCHEDULED, None],
[TEST_MESSAGE_WRITE_ACCESS_ALLOWED, None],
[TEST_MESSAGE_GIVEAWAY, None],
[TEST_MESSAGE_GIVEAWAY_WINNERS, None],
[TEST_MESSAGE_UNKNOWN, None],
]
On new Bot API updates new ContentType entries are created.
TestMessage.test_content_type checks what content type is returned. class TestAllMessageTypesTested:
Make sure MESSAGES_AND_CONTENT_TYPES has examples @pytest.fixture(scope="function")
for all the ContentType entries, fail otherwise. def known_content_types(self):
""" content_types = {t for t in ContentType}
content_types_w_example_messages = {t[1] for t in MESSAGES_AND_CONTENT_TYPES} content_types.remove(ContentType.ANY)
# content_types_w_example_messages.remove(ContentType.UNKNOWN) return content_types
known_content_types = {t for t in ContentType}
known_content_types.remove(ContentType.ANY) def test_for_content_type_tests(self, known_content_types):
assert content_types_w_example_messages == known_content_types """
Test if all ContentType options have example messages.
On new Bot API updates new ContentType entries are created.
TestMessage.test_content_type checks what content type is returned.
Make sure MESSAGES_AND_CONTENT_TYPES has examples
for all the ContentType entries, fail otherwise.
"""
content_types_w_example_messages = {t[1] for t in MESSAGES_AND_CONTENT_TYPES}
assert content_types_w_example_messages == known_content_types
def test_for_copy_methods(self, known_content_types):
"""
Test if all known message types are checked for copy_message.
Also relies on the previous test (both should be green)
"""
checked_content_types = {m[0].content_type for m in MESSAGES_AND_COPY_METHODS}
assert checked_content_types == known_content_types
class TestMessage: class TestMessage:
@ -657,47 +822,7 @@ class TestMessage:
@pytest.mark.parametrize( @pytest.mark.parametrize(
"message,expected_method", "message,expected_method",
[ MESSAGES_AND_COPY_METHODS,
[TEST_MESSAGE_TEXT, SendMessage],
[TEST_MESSAGE_AUDIO, SendAudio],
[TEST_MESSAGE_ANIMATION, SendAnimation],
[TEST_MESSAGE_DOCUMENT, SendDocument],
[TEST_MESSAGE_GAME, None],
[TEST_MESSAGE_PHOTO, SendPhoto],
[TEST_MESSAGE_STICKER, SendSticker],
[TEST_MESSAGE_VIDEO, SendVideo],
[TEST_MESSAGE_VIDEO_NOTE, SendVideoNote],
[TEST_MESSAGE_VOICE, SendVoice],
[TEST_MESSAGE_CONTACT, SendContact],
[TEST_MESSAGE_VENUE, SendVenue],
[TEST_MESSAGE_LOCATION, SendLocation],
[TEST_MESSAGE_STORY, ForwardMessage],
[TEST_MESSAGE_NEW_CHAT_MEMBERS, None],
[TEST_MESSAGE_LEFT_CHAT_MEMBER, None],
[TEST_MESSAGE_INVOICE, None],
[TEST_MESSAGE_SUCCESSFUL_PAYMENT, None],
[TEST_MESSAGE_CONNECTED_WEBSITE, None],
[TEST_MESSAGE_MIGRATE_FROM_CHAT_ID, None],
[TEST_MESSAGE_MIGRATE_TO_CHAT_ID, None],
[TEST_MESSAGE_PINNED_MESSAGE, None],
[TEST_MESSAGE_NEW_CHAT_TITLE, None],
[TEST_MESSAGE_NEW_CHAT_PHOTO, None],
[TEST_MESSAGE_DELETE_CHAT_PHOTO, None],
[TEST_MESSAGE_GROUP_CHAT_CREATED, None],
[TEST_MESSAGE_SUPERGROUP_CHAT_CREATED, None],
[TEST_MESSAGE_CHANNEL_CHAT_CREATED, None],
[TEST_MESSAGE_PASSPORT_DATA, None],
[TEST_MESSAGE_PROXIMITY_ALERT_TRIGGERED, None],
[TEST_MESSAGE_POLL, SendPoll],
[TEST_MESSAGE_MESSAGE_AUTO_DELETE_TIMER_CHANGED, None],
[TEST_MESSAGE_VIDEO_CHAT_STARTED, None],
[TEST_MESSAGE_VIDEO_CHAT_ENDED, None],
[TEST_MESSAGE_VIDEO_CHAT_PARTICIPANTS_INVITED, None],
[TEST_MESSAGE_DICE, SendDice],
[TEST_USER_SHARED, None],
[TEST_CHAT_SHARED, None],
[TEST_MESSAGE_UNKNOWN, None],
],
) )
def test_send_copy( def test_send_copy(
self, self,