python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-08 17:13:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix test "test_dispatcher.test_dispatcher.TestDispatcher::test_feed_raw_update"

Browse source
This commit is contained in:
Alex Root Junior 2019-12-09 21:39:06 +02:00
parent 2d6fc85dc0
commit 38c725db46
1 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

39
tests/test_dispatcher/test_dispatcher.py
View file

@ -2,7 +2,6 @@ import datetime
import time
import pytest
from asynctest import MagicMock, patch
from aiogram import Bot
from aiogram.api.types import Chat, Message, Update, User
@ -55,24 +54,28 @@ class TestDispatcher:
dp = Dispatcher()
bot = Bot("42:TEST")
with patch(
"aiogram.dispatcher.dispatcher.Dispatcher.feed_update", new_callable=MagicMock
) as patched_feed_update:
patched_feed_update.__aiter__.return_value = ["test"]
async for result in dp.feed_raw_update(
bot=bot,
update={
"update_id": 42,
"message": {
"message_id": 42,
"date": int(time.time()),
"text": "test",
"chat": {"id": 42, "type": "private"},
"user": {"id": 42, "is_bot": False, "first_name": "Test"},
},
@dp.message_handler()
async def my_handler(message: Message):
assert message.text == "test"
return message.text
handled = False
async for result in dp.feed_raw_update(
bot=bot,
update={
"update_id": 42,
"message": {
"message_id": 42,
"date": int(time.time()),
"text": "test",
"chat": {"id": 42, "type": "private"},
"user": {"id": 42, "is_bot": False, "first_name": "Test"},
},
):
assert result == "test"
},
):
handled = True
assert result == "test"
assert handled
@pytest.mark.skip
@pytest.mark.asyncio