From 779fe6aa4a00d6600dfb2d4a2b2e45284acaf8f0 Mon Sep 17 00:00:00 2001 From: evgfilim1 Date: Fri, 30 Jul 2021 19:23:33 +0500 Subject: [PATCH] Add more tests --- .../test_dispatcher/test_event/test_telegram.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/tests/test_dispatcher/test_event/test_telegram.py b/tests/test_dispatcher/test_event/test_telegram.py index de61029c..1a6a8cb8 100644 --- a/tests/test_dispatcher/test_event/test_telegram.py +++ b/tests/test_dispatcher/test_event/test_telegram.py @@ -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