fix tests

This commit is contained in:
Oleg A 2024-07-03 01:19:10 +03:00
parent 8fab9d1234
commit da2d62afb7
No known key found for this signature in database
GPG key ID: 5FE046817A9657C5
5 changed files with 30 additions and 6 deletions

View file

@ -68,6 +68,8 @@ from aiogram.types import (
Location,
MessageAutoDeleteTimerChanged,
MessageEntity,
PaidMediaInfo,
PaidMediaPhoto,
PassportData,
PhotoSize,
Poll,
@ -331,6 +333,22 @@ TEST_MESSAGE_CHANNEL_CHAT_CREATED = Message(
chat=Chat(id=-10042, type="channel"),
from_user=User(id=42, is_bot=False, first_name="Test"),
)
TEST_MESSAGE_PAID_MEDIA = Message(
message_id=42,
date=datetime.datetime.now(),
paid_media=PaidMediaInfo(
star_count=100500,
paid_media=[
PaidMediaPhoto(
photo=[
PhotoSize(file_id="file id", file_unique_id="file id", width=42, height=42)
],
)
],
),
chat=Chat(id=42, type="private"),
from_user=User(id=42, is_bot=False, first_name="Test"),
)
TEST_MESSAGE_PASSPORT_DATA = Message(
message_id=42,
date=datetime.datetime.now(),
@ -603,6 +621,7 @@ MESSAGES_AND_CONTENT_TYPES = [
[TEST_MESSAGE_GROUP_CHAT_CREATED, ContentType.GROUP_CHAT_CREATED],
[TEST_MESSAGE_SUPERGROUP_CHAT_CREATED, ContentType.SUPERGROUP_CHAT_CREATED],
[TEST_MESSAGE_CHANNEL_CHAT_CREATED, ContentType.CHANNEL_CHAT_CREATED],
[TEST_MESSAGE_PAID_MEDIA, ContentType.PAID_MEDIA],
[TEST_MESSAGE_PASSPORT_DATA, ContentType.PASSPORT_DATA],
[TEST_MESSAGE_PROXIMITY_ALERT_TRIGGERED, ContentType.PROXIMITY_ALERT_TRIGGERED],
[TEST_MESSAGE_POLL, ContentType.POLL],
@ -669,6 +688,7 @@ MESSAGES_AND_COPY_METHODS = [
[TEST_MESSAGE_GROUP_CHAT_CREATED, None],
[TEST_MESSAGE_SUPERGROUP_CHAT_CREATED, None],
[TEST_MESSAGE_CHANNEL_CHAT_CREATED, None],
[TEST_MESSAGE_PAID_MEDIA, None],
[TEST_MESSAGE_PASSPORT_DATA, None],
[TEST_MESSAGE_PROXIMITY_ALERT_TRIGGERED, None],
[TEST_MESSAGE_POLL, SendPoll],