mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
Add more tests
This commit is contained in:
parent
ee3a1e214e
commit
779fe6aa4a
1 changed files with 16 additions and 1 deletions
|
|
@ -4,7 +4,7 @@ from typing import Any, Awaitable, Callable, Dict, NoReturn, Union
|
|||
|
||||
import pytest
|
||||
|
||||
from aiogram.dispatcher.event.bases import SkipHandler
|
||||
from aiogram.dispatcher.event.bases import UNHANDLED, SkipHandler
|
||||
from aiogram.dispatcher.event.handler import HandlerObject
|
||||
from aiogram.dispatcher.event.telegram import TelegramEventObserver
|
||||
from aiogram.dispatcher.filters.base import BaseFilter
|
||||
|
|
@ -249,3 +249,18 @@ class TestTelegramEventObserver:
|
|||
router.message.filter(my_filter)
|
||||
assert len(router.message._handler.filters) == 1
|
||||
assert router.message._handler.filters[0].callback is my_filter
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_global_filter(self):
|
||||
r1 = Router()
|
||||
r2 = Router()
|
||||
|
||||
async def handler(evt):
|
||||
return evt
|
||||
|
||||
r1.message.filter(lambda evt: False)
|
||||
r1.message.register(handler)
|
||||
r2.message.register(handler)
|
||||
|
||||
assert await r1.message.trigger(None) is UNHANDLED
|
||||
assert await r2.message.trigger(None) is None
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue