Add extensions for class-based error handler

This commit is contained in:
Alex Root Junior 2020-04-12 23:29:10 +03:00
parent 0fbd2819f9
commit 01c6303d67
3 changed files with 43 additions and 3 deletions

View file

@ -0,0 +1,29 @@
from typing import Any
import pytest
from aiogram.api.types import (
CallbackQuery,
InlineQuery,
Poll,
PollOption,
ShippingAddress,
ShippingQuery,
User,
)
from aiogram.dispatcher.handler import ErrorHandler, PollHandler
class TestErrorHandler:
@pytest.mark.asyncio
async def test_extensions(self):
event = KeyError("kaboom")
class MyHandler(ErrorHandler):
async def handle(self) -> Any:
assert self.event == event
assert self.exception_name == event.__class__.__name__
assert self.exception_message == str(event)
return True
assert await MyHandler(event)