aiogram/tests/test_api/test_methods/test_send_paid_media.py
Alex Root Junior 11efa8e186
Added full support of Bot API 7.6 (#1533)
* Added full support of Bot API 7.6

* Fixed imports

* fix tests (#1534)

* Fixed coverage

* Override InputPaidMedia media type to `str | InputFile`

* Added shortcut

* Fixed PaidMediaType enum

* Added changelog

---------

Co-authored-by: Oleg A <t0rr@mail.ru>
2024-07-06 20:31:49 +03:00

43 lines
1.2 KiB
Python
Executable file

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