From f7574296b080f37a42fd6af4c4616075a544c8a4 Mon Sep 17 00:00:00 2001 From: JRoot Junior Date: Fri, 2 Jan 2026 01:28:53 +0200 Subject: [PATCH] Fixed tests that fails --- aiogram/types/message.py | 2 ++ tests/test_api/test_methods/test_get_chat.py | 1 + ...test_set_business_account_gift_settings.py | 1 + tests/test_api/test_types/test_message.py | 31 ++++++++++++++++++- 4 files changed, 34 insertions(+), 1 deletion(-) diff --git a/aiogram/types/message.py b/aiogram/types/message.py index 23df9503..53032360 100644 --- a/aiogram/types/message.py +++ b/aiogram/types/message.py @@ -742,6 +742,8 @@ class Message(MaybeInaccessibleMessage): return ContentType.GIFT if self.unique_gift: return ContentType.UNIQUE_GIFT + if self.gift_upgrade_sent: + return ContentType.GIFT_UPGRADE_SENT if self.paid_message_price_changed: return ContentType.PAID_MESSAGE_PRICE_CHANGED if self.suggested_post_approved: diff --git a/tests/test_api/test_methods/test_get_chat.py b/tests/test_api/test_methods/test_get_chat.py index a5cfaf52..bc0eb8cd 100644 --- a/tests/test_api/test_methods/test_get_chat.py +++ b/tests/test_api/test_methods/test_get_chat.py @@ -19,6 +19,7 @@ class TestGetChat: limited_gifts=True, unique_gifts=True, premium_subscription=True, + gifts_from_channels=True, ), ), ) diff --git a/tests/test_api/test_methods/test_set_business_account_gift_settings.py b/tests/test_api/test_methods/test_set_business_account_gift_settings.py index 44b095ef..9aaa7201 100644 --- a/tests/test_api/test_methods/test_set_business_account_gift_settings.py +++ b/tests/test_api/test_methods/test_set_business_account_gift_settings.py @@ -11,6 +11,7 @@ class TestSetBusinessAccountGiftSettings: business_connection_id="test_connection_id", show_gift_button=True, accepted_gift_types=AcceptedGiftTypes( + gifts_from_channels=True, unlimited_gifts=True, limited_gifts=True, unique_gifts=True, diff --git a/tests/test_api/test_types/test_message.py b/tests/test_api/test_types/test_message.py index 4648117b..f211f300 100644 --- a/tests/test_api/test_types/test_message.py +++ b/tests/test_api/test_types/test_message.py @@ -16,7 +16,6 @@ from aiogram.methods import ( PinChatMessage, SendAnimation, SendAudio, - SendChecklist, SendContact, SendDice, SendDocument, @@ -661,6 +660,7 @@ TEST_MESSAGE_UNIQUE_GIFT = Message( from_user=User(id=42, is_bot=False, first_name="Test"), unique_gift=UniqueGiftInfo( gift=UniqueGift( + gift_id="test_gift_id", base_name="test_gift", name="test_unique_gift", number=1, @@ -704,6 +704,33 @@ TEST_MESSAGE_UNIQUE_GIFT = Message( 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( message_id=42, date=datetime.datetime.now(), @@ -874,6 +901,7 @@ MESSAGES_AND_CONTENT_TYPES = [ [TEST_REFUND_PAYMENT, ContentType.REFUNDED_PAYMENT], [TEST_MESSAGE_GIFT, ContentType.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_SUGGESTED_POST_APPROVED, ContentType.SUGGESTED_POST_APPROVED], [TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, ContentType.SUGGESTED_POST_APPROVAL_FAILED], @@ -947,6 +975,7 @@ MESSAGES_AND_COPY_METHODS = [ [TEST_REFUND_PAYMENT, None], [TEST_MESSAGE_GIFT, None], [TEST_MESSAGE_UNIQUE_GIFT, None], + [TEST_MESSAGE_GIFT_UPGRADE_SENT, None], [TEST_MESSAGE_PAID_MESSAGE_PRICE_CHANGED, None], [TEST_MESSAGE_SUGGESTED_POST_APPROVED, None], [TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, None],