aiogram/tests/test_api/test_methods/test_set_chat_photo.py
Alex Root Junior d7a35f50a9 Reformat code
2019-11-19 22:44:40 +02:00

29 lines
1 KiB
Python

import pytest
from aiogram.api.methods import Request, SetChatPhoto
from aiogram.api.types import BufferedInputFile, InputFile
from tests.mocked_bot import MockedBot
class TestSetChatPhoto:
@pytest.mark.asyncio
async def test_method(self, bot: MockedBot):
prepare_result = bot.add_result_for(SetChatPhoto, ok=True, result=True)
response: bool = await SetChatPhoto(
chat_id=-42, photo=BufferedInputFile(b"", filename="file.png")
)
request: Request = bot.get_request()
assert request.method == "setChatPhoto"
assert response == prepare_result.result
@pytest.mark.asyncio
async def test_bot_method(self, bot: MockedBot):
prepare_result = bot.add_result_for(SetChatPhoto, ok=True, result=True)
response: bool = await bot.set_chat_photo(
chat_id=-42, photo=BufferedInputFile(b"", filename="file.png")
)
request: Request = bot.get_request()
assert request.method == "setChatPhoto"
assert response == prepare_result.result