Fix follow-up test coverage for sender_tag and can_edit_tag

This commit is contained in:
Codex Agent 2026-03-02 08:56:11 +00:00
parent 705403b2b9
commit d2d7e60895
4 changed files with 13 additions and 0 deletions

View file

@ -764,6 +764,8 @@ class Message(MaybeInaccessibleMessage):
return ContentType.GIFT_UPGRADE_SENT
if self.paid_message_price_changed:
return ContentType.PAID_MESSAGE_PRICE_CHANGED
if self.sender_tag:
return ContentType.SENDER_TAG
if self.suggested_post_approved:
return ContentType.SUGGESTED_POST_APPROVED
if self.suggested_post_approval_failed:

View file

@ -819,6 +819,13 @@ TEST_MESSAGE_PAID_MESSAGE_PRICE_CHANGED = Message(
paid_message_star_count=100,
),
)
TEST_MESSAGE_SENDER_TAG = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=User(id=42, is_bot=False, first_name="Test"),
sender_tag="test-tag",
)
TEST_MESSAGE_SUGGESTED_POST_APPROVED = Message(
message_id=42,
date=datetime.datetime.now(),
@ -932,6 +939,7 @@ MESSAGES_AND_CONTENT_TYPES = [
[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_SENDER_TAG, ContentType.SENDER_TAG],
[TEST_MESSAGE_SUGGESTED_POST_APPROVED, ContentType.SUGGESTED_POST_APPROVED],
[TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, ContentType.SUGGESTED_POST_APPROVAL_FAILED],
[TEST_MESSAGE_SUGGESTED_POST_DECLINED, ContentType.SUGGESTED_POST_DECLINED],
@ -1008,6 +1016,7 @@ MESSAGES_AND_COPY_METHODS = [
[TEST_MESSAGE_UNIQUE_GIFT, None],
[TEST_MESSAGE_GIFT_UPGRADE_SENT, None],
[TEST_MESSAGE_PAID_MESSAGE_PRICE_CHANGED, None],
[TEST_MESSAGE_SENDER_TAG, None],
[TEST_MESSAGE_SUGGESTED_POST_APPROVED, None],
[TEST_MESSAGE_SUGGESTED_POST_APPROVAL_FAILED, None],
[TEST_MESSAGE_SUGGESTED_POST_DECLINED, None],

View file

@ -314,6 +314,7 @@ class TestChatMemberUpdatedStatusFilter:
"can_send_polls": True,
"can_send_other_messages": True,
"can_add_web_page_previews": True,
"can_edit_tag": True,
"can_post_stories": True,
"can_edit_stories": True,
"can_delete_stories": True,

View file

@ -70,6 +70,7 @@ CHAT_MEMBER_RESTRICTED = ChatMemberRestricted(
can_send_polls=False,
can_send_other_messages=False,
can_add_web_page_previews=False,
can_edit_tag=False,
can_change_info=False,
can_invite_users=False,
can_pin_messages=False,