mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
Add chat member tools (#1527)
* feat: add ChatMemberAdapter * chore: apply formatter * docs: added changelog * docs: rm redundant import * feat: add pre-defined groups
This commit is contained in:
parent
5f05dfc664
commit
46e033e6da
4 changed files with 200 additions and 6 deletions
95
tests/test_utils/test_chat_member.py
Normal file
95
tests/test_utils/test_chat_member.py
Normal file
|
|
@ -0,0 +1,95 @@
|
|||
from datetime import datetime
|
||||
from typing import Type
|
||||
|
||||
import pytest
|
||||
|
||||
from aiogram.types import (
|
||||
ChatMember,
|
||||
ChatMemberAdministrator,
|
||||
ChatMemberBanned,
|
||||
ChatMemberLeft,
|
||||
ChatMemberMember,
|
||||
ChatMemberOwner,
|
||||
ChatMemberRestricted,
|
||||
User,
|
||||
)
|
||||
from aiogram.utils.chat_member import ChatMemberAdapter
|
||||
|
||||
USER = User(
|
||||
id=42,
|
||||
first_name="John Doe",
|
||||
is_bot=False,
|
||||
).model_dump()
|
||||
|
||||
CHAT_MEMBER_ADMINISTRATOR = ChatMemberAdministrator(
|
||||
user=USER,
|
||||
can_be_edited=False,
|
||||
can_manage_chat=True,
|
||||
can_change_info=True,
|
||||
can_delete_messages=True,
|
||||
can_invite_users=True,
|
||||
can_restrict_members=True,
|
||||
can_pin_messages=True,
|
||||
can_manage_topics=False,
|
||||
can_promote_members=False,
|
||||
can_manage_video_chats=True,
|
||||
can_post_stories=True,
|
||||
can_edit_stories=True,
|
||||
can_delete_stories=True,
|
||||
is_anonymous=False,
|
||||
can_manage_voice_chats=False,
|
||||
).model_dump()
|
||||
|
||||
CHAT_MEMBER_BANNED = ChatMemberBanned(
|
||||
user=USER,
|
||||
until_date=datetime.now(),
|
||||
).model_dump()
|
||||
|
||||
CHAT_MEMBER_LEFT = ChatMemberLeft(
|
||||
user=USER,
|
||||
).model_dump()
|
||||
|
||||
CHAT_MEMBER_MEMBER = ChatMemberMember(
|
||||
user=USER,
|
||||
).model_dump()
|
||||
|
||||
CHAT_MEMBER_OWNER = ChatMemberOwner(
|
||||
user=USER,
|
||||
is_anonymous=True,
|
||||
).model_dump()
|
||||
|
||||
CHAT_MEMBER_RESTRICTED = ChatMemberRestricted(
|
||||
user=USER,
|
||||
is_member=True,
|
||||
can_send_messages=False,
|
||||
can_send_audios=False,
|
||||
can_send_documents=False,
|
||||
can_send_photos=False,
|
||||
can_send_videos=False,
|
||||
can_send_video_notes=False,
|
||||
can_send_voice_notes=False,
|
||||
can_send_polls=False,
|
||||
can_send_other_messages=False,
|
||||
can_add_web_page_previews=False,
|
||||
can_change_info=False,
|
||||
can_invite_users=False,
|
||||
can_pin_messages=False,
|
||||
can_manage_topics=False,
|
||||
until_date=datetime.now(),
|
||||
).model_dump()
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("data", "resolved_type"),
|
||||
[
|
||||
(CHAT_MEMBER_ADMINISTRATOR, ChatMemberAdministrator),
|
||||
(CHAT_MEMBER_BANNED, ChatMemberBanned),
|
||||
(CHAT_MEMBER_LEFT, ChatMemberLeft),
|
||||
(CHAT_MEMBER_MEMBER, ChatMemberMember),
|
||||
(CHAT_MEMBER_OWNER, ChatMemberOwner),
|
||||
(CHAT_MEMBER_RESTRICTED, ChatMemberRestricted),
|
||||
],
|
||||
)
|
||||
def test_chat_member_resolution(data: dict, resolved_type: Type[ChatMember]) -> None:
|
||||
chat_member = ChatMemberAdapter.validate_python(data)
|
||||
assert isinstance(chat_member, resolved_type)
|
||||
Loading…
Add table
Add a link
Reference in a new issue