mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-11 18:01:04 +00:00
25 lines
1,004 B
Python
25 lines
1,004 B
Python
from aiogram.methods import Request, RestrictChatMember
|
|
from aiogram.types import ChatPermissions
|
|
from tests.mocked_bot import MockedBot
|
|
|
|
|
|
class TestRestrictChatMember:
|
|
async def test_method(self, bot: MockedBot):
|
|
prepare_result = bot.add_result_for(RestrictChatMember, ok=True, result=True)
|
|
|
|
response: bool = await RestrictChatMember(
|
|
chat_id=-42, user_id=42, permissions=ChatPermissions()
|
|
)
|
|
request: Request = bot.get_request()
|
|
assert request.method == "restrictChatMember"
|
|
assert response == prepare_result.result
|
|
|
|
async def test_bot_method(self, bot: MockedBot):
|
|
prepare_result = bot.add_result_for(RestrictChatMember, ok=True, result=True)
|
|
|
|
response: bool = await bot.restrict_chat_member(
|
|
chat_id=-42, user_id=42, permissions=ChatPermissions()
|
|
)
|
|
request: Request = bot.get_request()
|
|
assert request.method == "restrictChatMember"
|
|
assert response == prepare_result.result
|