diff --git a/tests/test_api/test_methods/test_send_paid_media.py b/tests/test_api/test_methods/test_send_paid_media.py new file mode 100755 index 00000000..e9f9588e --- /dev/null +++ b/tests/test_api/test_methods/test_send_paid_media.py @@ -0,0 +1,43 @@ +import datetime + +from aiogram.methods import SendPaidMedia +from aiogram.types import ( + Message, + Chat, + PaidMediaPhoto, + PaidMediaInfo, + PhotoSize, + InputPaidMediaPhoto, +) +from tests.mocked_bot import MockedBot + + +class TestSendPaidMedia: + async def test_bot_method(self, bot: MockedBot): + prepare_result = bot.add_result_for( + SendPaidMedia, + ok=True, + result=Message( + message_id=42, + date=datetime.datetime.now(), + chat=Chat(id=42, type="private"), + paid_media=PaidMediaInfo( + paid_media=[ + PaidMediaPhoto( + photo=[ + PhotoSize( + file_id="test", width=42, height=42, file_unique_id="test" + ) + ] + ) + ], + star_count=1, + ), + ), + ) + + response: Message = await bot.send_paid_media( + chat_id=-42, star_count=1, media=[InputPaidMediaPhoto(media="file_id")] + ) + request = bot.get_request() + assert response == prepare_result.result