aiogram/tests/test_api/test_methods/test_set_chat_photo.py

30 lines
1 KiB
Python
Raw Normal View History

2019-11-18 22:22:46 +02:00
import pytest
2019-11-19 22:44:40 +02:00
from aiogram.methods import Request, SetChatPhoto
from aiogram.types import BufferedInputFile, InputFile
2019-11-18 22:22:46 +02:00
from tests.mocked_bot import MockedBot
class TestSetChatPhoto:
@pytest.mark.asyncio
async def test_method(self, bot: MockedBot):
2019-11-19 22:44:12 +02:00
prepare_result = bot.add_result_for(SetChatPhoto, ok=True, result=True)
2019-11-18 22:22:46 +02:00
2019-11-19 22:44:12 +02:00
response: bool = await SetChatPhoto(
chat_id=-42, photo=BufferedInputFile(b"", filename="file.png")
)
2019-11-18 22:22:46 +02:00
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):
2019-11-19 22:44:12 +02:00
prepare_result = bot.add_result_for(SetChatPhoto, ok=True, result=True)
2019-11-18 22:22:46 +02:00
2019-11-19 22:44:12 +02:00
response: bool = await bot.set_chat_photo(
chat_id=-42, photo=BufferedInputFile(b"", filename="file.png")
)
2019-11-18 22:22:46 +02:00
request: Request = bot.get_request()
assert request.method == "setChatPhoto"
assert response == prepare_result.result