aiogram/tests/test_api/test_methods/test_set_chat_permissions.py

29 lines
1.1 KiB
Python
Raw Normal View History

2019-11-18 22:22:46 +02:00
import pytest
from aiogram.api.methods import Request, SetChatPermissions
2019-11-19 22:44:12 +02:00
from aiogram.api.types import ChatPermissions
2019-11-18 22:22:46 +02:00
from tests.mocked_bot import MockedBot
class TestSetChatPermissions:
@pytest.mark.asyncio
async def test_method(self, bot: MockedBot):
2019-11-19 22:44:12 +02:00
prepare_result = bot.add_result_for(SetChatPermissions, ok=True, result=True)
2019-11-18 22:22:46 +02:00
2019-11-19 22:44:12 +02:00
response: bool = await SetChatPermissions(
chat_id=-42, permissions=ChatPermissions(can_send_messages=False)
)
2019-11-18 22:22:46 +02:00
request: Request = bot.get_request()
assert request.method == "setChatPermissions"
assert response == prepare_result.result
@pytest.mark.asyncio
async def test_bot_method(self, bot: MockedBot):
2019-11-19 22:44:12 +02:00
prepare_result = bot.add_result_for(SetChatPermissions, ok=True, result=True)
2019-11-18 22:22:46 +02:00
2019-11-19 22:44:12 +02:00
response: bool = await bot.set_chat_permissions(
chat_id=-42, permissions=ChatPermissions(can_send_messages=False)
)
2019-11-18 22:22:46 +02:00
request: Request = bot.get_request()
assert request.method == "setChatPermissions"
assert response == prepare_result.result