diff --git a/tests/test_api/test_methods/test_ban_chat_member.py b/tests/test_api/test_methods/test_ban_chat_member.py new file mode 100644 index 00000000..bdbf1c41 --- /dev/null +++ b/tests/test_api/test_methods/test_ban_chat_member.py @@ -0,0 +1,24 @@ +import pytest + +from aiogram.methods import BanChatMember, Request +from tests.mocked_bot import MockedBot + + +class TestKickChatMember: + @pytest.mark.asyncio + async def test_method(self, bot: MockedBot): + prepare_result = bot.add_result_for(BanChatMember, ok=True, result=True) + + response: bool = await BanChatMember(chat_id=-42, user_id=42) + request: Request = bot.get_request() + assert request.method == "banChatMember" + assert response == prepare_result.result + + @pytest.mark.asyncio + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for(BanChatMember, ok=True, result=True) + + response: bool = await bot.ban_chat_member(chat_id=-42, user_id=42) + request: Request = bot.get_request() + assert request.method == "banChatMember" + assert response == prepare_result.result diff --git a/tests/test_api/test_methods/test_delete_my_commands.py b/tests/test_api/test_methods/test_delete_my_commands.py new file mode 100644 index 00000000..a52ff172 --- /dev/null +++ b/tests/test_api/test_methods/test_delete_my_commands.py @@ -0,0 +1,24 @@ +import pytest + +from aiogram.methods import BanChatMember, Request, DeleteMyCommands +from tests.mocked_bot import MockedBot + + +class TestKickChatMember: + @pytest.mark.asyncio + async def test_method(self, bot: MockedBot): + prepare_result = bot.add_result_for(DeleteMyCommands, ok=True, result=True) + + response: bool = await DeleteMyCommands() + request: Request = bot.get_request() + assert request.method == "deleteMyCommands" + assert response == prepare_result.result + + @pytest.mark.asyncio + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for(DeleteMyCommands, ok=True, result=True) + + response: bool = await bot.delete_my_commands() + request: Request = bot.get_request() + assert request.method == "deleteMyCommands" + assert response == prepare_result.result diff --git a/tests/test_api/test_methods/test_get_chat_member_count.py b/tests/test_api/test_methods/test_get_chat_member_count.py new file mode 100644 index 00000000..a3ecec72 --- /dev/null +++ b/tests/test_api/test_methods/test_get_chat_member_count.py @@ -0,0 +1,24 @@ +import pytest + +from aiogram.methods import GetChatMemberCount, Request +from tests.mocked_bot import MockedBot + + +class TestGetChatMembersCount: + @pytest.mark.asyncio + async def test_method(self, bot: MockedBot): + prepare_result = bot.add_result_for(GetChatMemberCount, ok=True, result=42) + + response: int = await GetChatMemberCount(chat_id=-42) + request: Request = bot.get_request() + assert request.method == "getChatMemberCount" + assert response == prepare_result.result + + @pytest.mark.asyncio + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for(GetChatMemberCount, ok=True, result=42) + + response: int = await bot.get_chat_member_count(chat_id=-42) + request: Request = bot.get_request() + assert request.method == "getChatMemberCount" + assert response == prepare_result.result