diff --git a/tests/test_api/test_methods/test_copy_messages.py b/tests/test_api/test_methods/test_copy_messages.py new file mode 100644 index 00000000..85759a3e --- /dev/null +++ b/tests/test_api/test_methods/test_copy_messages.py @@ -0,0 +1,28 @@ +from random import randint +from aiogram.methods import CopyMessages +from aiogram.types import MessageId +from tests.mocked_bot import MockedBot + + +class TestCopyMessages: + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for( + CopyMessages, + ok=True, + result=[ + MessageId(message_id=randint(100, 200)), + MessageId(message_id=randint(300, 400)), + ], + ) + + response: list[MessageId] = await bot.copy_messages( + chat_id=randint(1000, 9999), + from_chat_id=randint(1000, 9999), + message_ids=[ + randint(1000, 4999), + randint(5000, 9999), + ], + ) + request = bot.get_request() + assert request + assert response == prepare_result.result diff --git a/tests/test_api/test_methods/test_delete_messages.py b/tests/test_api/test_methods/test_delete_messages.py new file mode 100644 index 00000000..fba51ec6 --- /dev/null +++ b/tests/test_api/test_methods/test_delete_messages.py @@ -0,0 +1,19 @@ +from aiogram.methods import DeleteMessages +from tests.mocked_bot import MockedBot + + +class TestDeleteMessages: + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for( + DeleteMessages, + ok=True, + result=True, + ) + + response: bool = await bot.delete_messages( + chat_id=42, + message_ids=[13, 77], + ) + request = bot.get_request() + assert request + assert response == prepare_result.result diff --git a/tests/test_api/test_methods/test_forward_message.py b/tests/test_api/test_methods/test_forward_message.py index f04ec0f0..daa434db 100644 --- a/tests/test_api/test_methods/test_forward_message.py +++ b/tests/test_api/test_methods/test_forward_message.py @@ -20,4 +20,5 @@ class TestForwardMessage: response: Message = await bot.forward_message(chat_id=42, from_chat_id=42, message_id=42) request = bot.get_request() + assert request assert response == prepare_result.result diff --git a/tests/test_api/test_methods/test_forward_messages.py b/tests/test_api/test_methods/test_forward_messages.py new file mode 100644 index 00000000..2bb212e2 --- /dev/null +++ b/tests/test_api/test_methods/test_forward_messages.py @@ -0,0 +1,28 @@ +from random import randint +from aiogram.methods import ForwardMessages +from aiogram.types import MessageId +from tests.mocked_bot import MockedBot + + +class TestForwardMessages: + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for( + ForwardMessages, + ok=True, + result=[ + MessageId(message_id=randint(100, 200)), + MessageId(message_id=randint(200, 300)), + ], + ) + + response: list[MessageId] = await bot.forward_messages( + chat_id=randint(10, 50), + from_chat_id=randint(50, 99), + message_ids=[ + randint(400, 500), + randint(600, 700), + ], + ) + request = bot.get_request() + assert request + assert response == prepare_result.result diff --git a/tests/test_api/test_methods/test_get_user_chat_boosts.py b/tests/test_api/test_methods/test_get_user_chat_boosts.py new file mode 100644 index 00000000..be261159 --- /dev/null +++ b/tests/test_api/test_methods/test_get_user_chat_boosts.py @@ -0,0 +1,52 @@ +from datetime import datetime, timedelta +from random import randint + +from aiogram.methods import GetUserChatBoosts +from aiogram.types import ( + ChatBoost, + ChatBoostSourceGiveaway, + ChatBoostSourcePremium, + User, + UserChatBoosts, +) +from tests.mocked_bot import MockedBot + + +class TestGetUserChatBoosts: + async def test_bot_method(self, bot: MockedBot): + now = datetime.now() + user = User( + id=randint(200, 500), + is_bot=False, + first_name="name", + ) + prepare_result = bot.add_result_for( + GetUserChatBoosts, + ok=True, + result=UserChatBoosts( + boosts=[ + ChatBoost( + boost_id="eggs", + add_date=now - timedelta(days=7), + expiration_date=now + timedelta(days=14), + source=ChatBoostSourceGiveaway( + giveaway_message_id=randint(100, 300), + ), + ), + ChatBoost( + boost_id="spam", + add_date=now - timedelta(days=3), + expiration_date=now + timedelta(days=21), + source=ChatBoostSourcePremium(user=user), + ), + ] + ), + ) + + response: UserChatBoosts = await bot.get_user_chat_boosts( + chat_id=randint(100, 200), + user_id=user.id, + ) + request = bot.get_request() + assert request + assert response == prepare_result.result diff --git a/tests/test_api/test_methods/test_set_message_reaction.py b/tests/test_api/test_methods/test_set_message_reaction.py new file mode 100644 index 00000000..ee99e99f --- /dev/null +++ b/tests/test_api/test_methods/test_set_message_reaction.py @@ -0,0 +1,25 @@ +from random import randint + +from aiogram.methods import SetMessageReaction +from aiogram.types import ReactionTypeCustomEmoji +from tests.mocked_bot import MockedBot + + +class TestSetMessageReaction: + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for( + SetMessageReaction, + ok=True, + result=True, + ) + + response: bool = await bot.set_message_reaction( + chat_id=randint(200, 300), + message_id=randint(100, 200), + reaction=[ + ReactionTypeCustomEmoji(custom_emoji="qwerty"), + ], + ) + request = bot.get_request() + assert request + assert response == prepare_result.result