aiogram/tests/test_api/test_methods/test_get_updates.py

24 lines
855 B
Python
Raw Normal View History

2019-11-19 00:04:02 +02:00
from typing import List
from aiogram.methods import GetUpdates, Request
from aiogram.types import Update
2019-11-18 22:22:46 +02:00
from tests.mocked_bot import MockedBot
class TestGetUpdates:
async def test_method(self, bot: MockedBot):
2019-11-19 00:04:02 +02:00
prepare_result = bot.add_result_for(GetUpdates, ok=True, result=[Update(update_id=42)])
2019-11-18 22:22:46 +02:00
response: List[Update] = await GetUpdates()
request: Request = bot.get_request()
assert request.method == "getUpdates"
assert response == prepare_result.result
async def test_bot_method(self, bot: MockedBot):
2019-11-19 00:04:02 +02:00
prepare_result = bot.add_result_for(GetUpdates, ok=True, result=[Update(update_id=42)])
2019-11-18 22:22:46 +02:00
response: List[Update] = await bot.get_updates()
request: Request = bot.get_request()
assert request.method == "getUpdates"
assert response == prepare_result.result