mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-11 01:54:53 +00:00
Add middlewares (API + Docs + Tests)
This commit is contained in:
parent
e4cd4c1763
commit
5b6ec599b1
24 changed files with 1120 additions and 42 deletions
82
tests/test_dispatcher/test_middlewares/test_manager.py
Normal file
82
tests/test_dispatcher/test_middlewares/test_manager.py
Normal file
|
|
@ -0,0 +1,82 @@
|
|||
import pytest
|
||||
|
||||
from aiogram import Router
|
||||
from aiogram.api.types import Update
|
||||
from aiogram.dispatcher.middlewares.base import BaseMiddleware
|
||||
from aiogram.dispatcher.middlewares.manager import MiddlewareManager
|
||||
from aiogram.dispatcher.middlewares.types import MiddlewareStep
|
||||
|
||||
try:
|
||||
from asynctest import CoroutineMock, patch
|
||||
except ImportError:
|
||||
from unittest.mock import AsyncMock as CoroutineMock, patch # type: ignore
|
||||
|
||||
|
||||
@pytest.fixture("function")
|
||||
def router():
|
||||
return Router()
|
||||
|
||||
|
||||
@pytest.fixture("function")
|
||||
def manager(router: Router):
|
||||
return MiddlewareManager(router)
|
||||
|
||||
|
||||
class TestManager:
|
||||
def test_setup(self, manager: MiddlewareManager):
|
||||
middleware = BaseMiddleware()
|
||||
returned = manager.setup(middleware)
|
||||
assert returned is middleware
|
||||
assert middleware.configured
|
||||
assert middleware.manager is manager
|
||||
assert middleware in manager
|
||||
|
||||
@pytest.mark.parametrize("obj", [object, object(), None, BaseMiddleware])
|
||||
def test_setup_invalid_type(self, manager: MiddlewareManager, obj):
|
||||
with pytest.raises(TypeError):
|
||||
assert manager.setup(obj)
|
||||
|
||||
def test_configure_twice_different_managers(self, manager: MiddlewareManager, router: Router):
|
||||
middleware = BaseMiddleware()
|
||||
manager.setup(middleware)
|
||||
|
||||
assert middleware.configured
|
||||
|
||||
new_manager = MiddlewareManager(router)
|
||||
with pytest.raises(ValueError):
|
||||
new_manager.setup(middleware)
|
||||
with pytest.raises(ValueError):
|
||||
middleware.setup(new_manager)
|
||||
|
||||
def test_configure_twice(self, manager: MiddlewareManager):
|
||||
middleware = BaseMiddleware()
|
||||
manager.setup(middleware)
|
||||
|
||||
assert middleware.configured
|
||||
|
||||
with pytest.warns(RuntimeWarning, match="is already configured for this Router"):
|
||||
manager.setup(middleware)
|
||||
|
||||
with pytest.warns(RuntimeWarning, match="is already configured for this Router"):
|
||||
middleware.setup(manager)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@pytest.mark.parametrize("count", range(5))
|
||||
async def test_trigger(self, manager: MiddlewareManager, count: int):
|
||||
for _ in range(count):
|
||||
manager.setup(BaseMiddleware())
|
||||
|
||||
with patch(
|
||||
"aiogram.dispatcher.middlewares.base.BaseMiddleware.trigger",
|
||||
new_callable=CoroutineMock,
|
||||
) as mocked_call:
|
||||
await manager.trigger(
|
||||
step=MiddlewareStep.PROCESS,
|
||||
event_name="update",
|
||||
event=Update(update_id=42),
|
||||
data={},
|
||||
result=None,
|
||||
reverse=True,
|
||||
)
|
||||
|
||||
assert mocked_call.await_count == count
|
||||
Loading…
Add table
Add a link
Reference in a new issue