aiogram/tests/test_api/test_methods/test_upload_sticker_file.py

35 lines
1.3 KiB
Python
Raw Normal View History

from aiogram.enums import StickerFormat
from aiogram.methods import Request, UploadStickerFile
from aiogram.types import BufferedInputFile, File
2019-11-18 22:22:46 +02:00
from tests.mocked_bot import MockedBot
class TestUploadStickerFile:
async def test_method(self, bot: MockedBot):
2019-11-19 22:44:12 +02:00
prepare_result = bot.add_result_for(
2020-01-11 19:56:38 +02:00
UploadStickerFile, ok=True, result=File(file_id="file id", file_unique_id="file id")
2019-11-19 22:44:12 +02:00
)
2019-11-18 22:22:46 +02:00
2019-11-19 22:44:12 +02:00
response: File = await UploadStickerFile(
user_id=42,
sticker=BufferedInputFile(b"", "file.png"),
sticker_format=StickerFormat.STATIC,
2019-11-19 22:44:12 +02:00
)
2019-11-18 22:22:46 +02:00
request: Request = bot.get_request()
assert request.method == "uploadStickerFile"
assert response == prepare_result.result
async def test_bot_method(self, bot: MockedBot):
2019-11-19 22:44:12 +02:00
prepare_result = bot.add_result_for(
2020-01-11 19:56:38 +02:00
UploadStickerFile, ok=True, result=File(file_id="file id", file_unique_id="file id")
2019-11-19 22:44:12 +02:00
)
2019-11-18 22:22:46 +02:00
2019-11-19 22:44:12 +02:00
response: File = await bot.upload_sticker_file(
user_id=42,
sticker=BufferedInputFile(b"", "file.png"),
sticker_format=StickerFormat.STATIC,
2019-11-19 22:44:12 +02:00
)
2019-11-18 22:22:46 +02:00
request: Request = bot.get_request()
assert request.method == "uploadStickerFile"
assert response == prepare_result.result