python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

chore: mark expected warnings

Browse source
This commit is contained in:
Oleg A 2023-09-18 15:25:45 +03:00
parent 54bd29c805
commit c5e3afa0ca
No known key found for this signature in database
GPG key ID: 5FE046817A9657C5
2 changed files with 26 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

35
tests/test_dispatcher/test_dispatcher.py
View file

@ -187,8 +187,9 @@ class TestDispatcher:
async def test_process_update_empty(self, bot: MockedBot): async def test_process_update_empty(self, bot: MockedBot):
dispatcher = Dispatcher() dispatcher = Dispatcher()
result = await dispatcher._process_update(bot=bot, update=Update(update_id=42)) with pytest.warns(RuntimeWarning, match="Detected unknown update type") as record:
assert not result result = await dispatcher._process_update(bot=bot, update=Update(update_id=42))
assert not result
async def test_process_update_handled(self, bot: MockedBot): async def test_process_update_handled(self, bot: MockedBot):
dispatcher = Dispatcher() dispatcher = Dispatcher()
@ -197,7 +198,8 @@ class TestDispatcher:
async def update_handler(update: Update): async def update_handler(update: Update):
pass pass
assert await dispatcher._process_update(bot=bot, update=Update(update_id=42)) with pytest.warns(RuntimeWarning, match="Detected unknown update type"):
assert await dispatcher._process_update(bot=bot, update=Update(update_id=42))
@pytest.mark.parametrize( @pytest.mark.parametrize(
"event_type,update,has_chat,has_user", "event_type,update,has_chat,has_user",
@ -479,9 +481,13 @@ class TestDispatcher:
async def test_listen_unknown_update(self): async def test_listen_unknown_update(self):
dp = Dispatcher() dp = Dispatcher()
with (
with pytest.raises(SkipHandler): pytest.raises(SkipHandler),
pytest.warns(RuntimeWarning, match="Detected unknown update type") as record,
):
await dp._listen_update(Update(update_id=42)) await dp._listen_update(Update(update_id=42))
if not record:
pytest.fail("Expected 'Detected unknown update type' warning.")
async def test_listen_unhandled_update(self): async def test_listen_unhandled_update(self):
dp = Dispatcher() dp = Dispatcher()
@ -608,7 +614,9 @@ class TestDispatcher:
async def update_handler(update: Update): async def update_handler(update: Update):
raise Exception("Kaboom!") raise Exception("Kaboom!")
assert await dispatcher._process_update(bot=bot, update=Update(update_id=42)) with pytest.warns(RuntimeWarning, match="Detected unknown update type"):
assert await dispatcher._process_update(bot=bot, update=Update(update_id=42))
log_records = [rec.message for rec in caplog.records] log_records = [rec.message for rec in caplog.records]
assert len(log_records) == 1 assert len(log_records) == 1
assert "Cause exception while process update" in log_records[0] assert "Cause exception while process update" in log_records[0]
@ -834,12 +842,17 @@ class TestDispatcher:
dispatcher = Dispatcher() dispatcher = Dispatcher()
dispatcher.message.register(invalid_message_handler) dispatcher.message.register(invalid_message_handler)
response = await dispatcher.feed_webhook_update(bot, RAW_UPDATE, _timeout=0.1) pattern = r"Detected slow response into webhook"
assert response is None with pytest.warns(RuntimeWarning, match=pattern) as record:
await asyncio.sleep(0.5) response = await dispatcher.feed_webhook_update(bot, RAW_UPDATE, _timeout=0.1)
assert response is None
await asyncio.sleep(0.5)
log_records = [rec.message for rec in caplog.records] log_records = [rec.message for rec in caplog.records]
assert "Cause exception while process update" in log_records[0] assert "Cause exception while process update" in log_records[0]
if not record:
pytest.fail("Expected 'Detected slow response into webhook' warning.")
def test_specify_updates_calculation(self): def test_specify_updates_calculation(self):
def simple_msg_handler() -> None: def simple_msg_handler() -> None:

3
tests/test_filters/test_exception.py
View file

@ -72,4 +72,5 @@ class TestDispatchException:
async def handler0(event): async def handler0(event):
return "Handled" return "Handled"
assert await dp.feed_update(bot, Update(update_id=0)) == "Handled" with pytest.warns(RuntimeWarning, match="Detected unknown update type"):
assert await dp.feed_update(bot, Update(update_id=0)) == "Handled"