Fixed tests that fails

This commit is contained in:
JRoot Junior 2026-01-02 01:28:53 +02:00
parent a179881c62
commit f7574296b0
No known key found for this signature in database
GPG key ID: 738964250D5FF6E2
4 changed files with 34 additions and 1 deletions

View file

@ -742,6 +742,8 @@ class Message(MaybeInaccessibleMessage):
return ContentType.GIFT return ContentType.GIFT
if self.unique_gift: if self.unique_gift:
return ContentType.UNIQUE_GIFT return ContentType.UNIQUE_GIFT
if self.gift_upgrade_sent:
return ContentType.GIFT_UPGRADE_SENT
if self.paid_message_price_changed: if self.paid_message_price_changed:
return ContentType.PAID_MESSAGE_PRICE_CHANGED return ContentType.PAID_MESSAGE_PRICE_CHANGED
if self.suggested_post_approved: if self.suggested_post_approved:

View file

@ -19,6 +19,7 @@ class TestGetChat:
limited_gifts=True, limited_gifts=True,
unique_gifts=True, unique_gifts=True,
premium_subscription=True, premium_subscription=True,
gifts_from_channels=True,
), ),
), ),
) )

View file

@ -11,6 +11,7 @@ class TestSetBusinessAccountGiftSettings:
business_connection_id="test_connection_id", business_connection_id="test_connection_id",
show_gift_button=True, show_gift_button=True,
accepted_gift_types=AcceptedGiftTypes( accepted_gift_types=AcceptedGiftTypes(
gifts_from_channels=True,
unlimited_gifts=True, unlimited_gifts=True,
limited_gifts=True, limited_gifts=True,
unique_gifts=True, unique_gifts=True,

View file

@ -16,7 +16,6 @@ from aiogram.methods import (
PinChatMessage, PinChatMessage,
SendAnimation, SendAnimation,
SendAudio, SendAudio,
SendChecklist,
SendContact, SendContact,
SendDice, SendDice,
SendDocument, SendDocument,
@ -661,6 +660,7 @@ TEST_MESSAGE_UNIQUE_GIFT = Message(
from_user=User(id=42, is_bot=False, first_name="Test"), from_user=User(id=42, is_bot=False, first_name="Test"),
unique_gift=UniqueGiftInfo( unique_gift=UniqueGiftInfo(
gift=UniqueGift( gift=UniqueGift(
gift_id="test_gift_id",
base_name="test_gift", base_name="test_gift",
name="test_unique_gift", name="test_unique_gift",
number=1, number=1,
@ -704,6 +704,33 @@ TEST_MESSAGE_UNIQUE_GIFT = Message(
origin="upgrade", origin="upgrade",
), ),
) )
TEST_MESSAGE_GIFT_UPGRADE_SENT = Message(
message_id=42,
chat=Chat(id=42, type="private"),
from_user=User(id=42, is_bot=False, first_name="Test"),
date=datetime.datetime.now(),
gift_upgrade_sent=GiftInfo(
gift=Gift(
id="test_gift_id",
sticker=Sticker(
file_id="test_file_id",
file_unique_id="test_file_unique_id",
type="regular",
width=512,
height=512,
is_animated=False,
is_video=False,
),
star_count=100,
),
owned_gift_id="test_owned_gift_id",
convert_star_count=50,
prepaid_upgrade_star_count=25,
can_be_upgraded=True,
text="Test gift message",
is_private=False,
),
)
TEST_MESSAGE_CHECKLIST = Message( TEST_MESSAGE_CHECKLIST = Message(
message_id=42, message_id=42,
date=datetime.datetime.now(), date=datetime.datetime.now(),
@ -874,6 +901,7 @@ MESSAGES_AND_CONTENT_TYPES = [
[TEST_REFUND_PAYMENT, ContentType.REFUNDED_PAYMENT], [TEST_REFUND_PAYMENT, ContentType.REFUNDED_PAYMENT],
[TEST_MESSAGE_GIFT, ContentType.GIFT], [TEST_MESSAGE_GIFT, ContentType.GIFT],
[TEST_MESSAGE_UNIQUE_GIFT, ContentType.UNIQUE_GIFT], [TEST_MESSAGE_UNIQUE_GIFT, ContentType.UNIQUE_GIFT],
[TEST_MESSAGE_GIFT_UPGRADE_SENT, ContentType.GIFT_UPGRADE_SENT],
[TEST_MESSAGE_PAID_MESSAGE_PRICE_CHANGED, ContentType.PAID_MESSAGE_PRICE_CHANGED], [TEST_MESSAGE_PAID_MESSAGE_PRICE_CHANGED, ContentType.PAID_MESSAGE_PRICE_CHANGED],
[TEST_MESSAGE_SUGGESTED_POST_APPROVED, ContentType.SUGGESTED_POST_APPROVED], [TEST_MESSAGE_SUGGESTED_POST_APPROVED, ContentType.SUGGESTED_POST_APPROVED],
[TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, ContentType.SUGGESTED_POST_APPROVAL_FAILED], [TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, ContentType.SUGGESTED_POST_APPROVAL_FAILED],
@ -947,6 +975,7 @@ MESSAGES_AND_COPY_METHODS = [
[TEST_REFUND_PAYMENT, None], [TEST_REFUND_PAYMENT, None],
[TEST_MESSAGE_GIFT, None], [TEST_MESSAGE_GIFT, None],
[TEST_MESSAGE_UNIQUE_GIFT, None], [TEST_MESSAGE_UNIQUE_GIFT, None],
[TEST_MESSAGE_GIFT_UPGRADE_SENT, None],
[TEST_MESSAGE_PAID_MESSAGE_PRICE_CHANGED, None], [TEST_MESSAGE_PAID_MESSAGE_PRICE_CHANGED, None],
[TEST_MESSAGE_SUGGESTED_POST_APPROVED, None], [TEST_MESSAGE_SUGGESTED_POST_APPROVED, None],
[TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, None], [TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, None],