mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
Add set_member_tag shortcut tests and align decoration expectations
This commit is contained in:
parent
a126f8e0b1
commit
705403b2b9
6 changed files with 147 additions and 8 deletions
|
|
@ -6,6 +6,11 @@ class TestPromoteChatMember:
|
|||
async def test_bot_method(self, bot: MockedBot):
|
||||
prepare_result = bot.add_result_for(PromoteChatMember, ok=True, result=True)
|
||||
|
||||
response: bool = await bot.promote_chat_member(chat_id=-42, user_id=42)
|
||||
bot.get_request()
|
||||
response: bool = await bot.promote_chat_member(
|
||||
chat_id=-42,
|
||||
user_id=42,
|
||||
can_manage_tags=True,
|
||||
)
|
||||
request = bot.get_request()
|
||||
assert request.can_manage_tags is True
|
||||
assert response == prepare_result.result
|
||||
|
|
|
|||
|
|
@ -7,5 +7,8 @@ class TestSetChatMemberTag:
|
|||
prepare_result = bot.add_result_for(SetChatMemberTag, ok=True, result=True)
|
||||
|
||||
response: bool = await bot.set_chat_member_tag(chat_id=-42, user_id=42, tag="test")
|
||||
bot.get_request()
|
||||
request = bot.get_request()
|
||||
assert request.chat_id == -42
|
||||
assert request.user_id == 42
|
||||
assert request.tag == "test"
|
||||
assert response == prepare_result.result
|
||||
|
|
|
|||
|
|
@ -0,0 +1,84 @@
|
|||
from datetime import datetime
|
||||
|
||||
from aiogram.types import (
|
||||
ChatAdministratorRights,
|
||||
ChatMemberAdministrator,
|
||||
ChatMemberMember,
|
||||
ChatMemberRestricted,
|
||||
ChatPermissions,
|
||||
User,
|
||||
)
|
||||
|
||||
|
||||
class TestChatMemberTagPermissions:
|
||||
def test_chat_administrator_rights_can_manage_tags(self):
|
||||
rights = ChatAdministratorRights(
|
||||
is_anonymous=False,
|
||||
can_manage_chat=True,
|
||||
can_delete_messages=True,
|
||||
can_manage_video_chats=True,
|
||||
can_restrict_members=True,
|
||||
can_promote_members=True,
|
||||
can_change_info=True,
|
||||
can_invite_users=True,
|
||||
can_post_stories=True,
|
||||
can_edit_stories=True,
|
||||
can_delete_stories=True,
|
||||
can_manage_tags=True,
|
||||
)
|
||||
assert rights.can_manage_tags is True
|
||||
|
||||
def test_chat_member_administrator_can_manage_tags(self):
|
||||
admin = ChatMemberAdministrator(
|
||||
user=User(id=42, is_bot=False, first_name="User"),
|
||||
can_be_edited=True,
|
||||
is_anonymous=False,
|
||||
can_manage_chat=True,
|
||||
can_delete_messages=True,
|
||||
can_manage_video_chats=True,
|
||||
can_restrict_members=True,
|
||||
can_promote_members=True,
|
||||
can_change_info=True,
|
||||
can_invite_users=True,
|
||||
can_post_stories=True,
|
||||
can_edit_stories=True,
|
||||
can_delete_stories=True,
|
||||
can_manage_tags=True,
|
||||
)
|
||||
assert admin.can_manage_tags is True
|
||||
|
||||
def test_chat_permissions_can_edit_tag(self):
|
||||
permissions = ChatPermissions(can_edit_tag=True)
|
||||
assert permissions.can_edit_tag is True
|
||||
|
||||
def test_chat_member_member_tag(self):
|
||||
member = ChatMemberMember(
|
||||
user=User(id=42, is_bot=False, first_name="User"),
|
||||
tag="premium",
|
||||
)
|
||||
assert member.tag == "premium"
|
||||
|
||||
def test_chat_member_restricted_can_edit_tag_and_tag(self):
|
||||
restricted = ChatMemberRestricted(
|
||||
user=User(id=42, is_bot=False, first_name="User"),
|
||||
is_member=True,
|
||||
can_send_messages=True,
|
||||
can_send_audios=True,
|
||||
can_send_documents=True,
|
||||
can_send_photos=True,
|
||||
can_send_videos=True,
|
||||
can_send_video_notes=True,
|
||||
can_send_voice_notes=True,
|
||||
can_send_polls=True,
|
||||
can_send_other_messages=True,
|
||||
can_add_web_page_previews=True,
|
||||
can_edit_tag=True,
|
||||
can_change_info=True,
|
||||
can_invite_users=True,
|
||||
can_pin_messages=True,
|
||||
can_manage_topics=True,
|
||||
until_date=datetime.now(),
|
||||
tag="premium",
|
||||
)
|
||||
assert restricted.can_edit_tag is True
|
||||
assert restricted.tag == "premium"
|
||||
Loading…
Add table
Add a link
Reference in a new issue