More class based handlers

This commit is contained in:
Alex Root Junior 2020-01-21 23:33:44 +02:00
parent b144332287
commit 895b727ddf
24 changed files with 506 additions and 30 deletions

View file

@ -0,0 +1,34 @@
from typing import Any
import pytest
from aiogram.api.types import (
CallbackQuery,
InlineQuery,
Poll,
PollOption,
ShippingAddress,
ShippingQuery,
User,
)
from aiogram.dispatcher.handler import PollHandler
class TestShippingQueryHandler:
@pytest.mark.asyncio
async def test_attributes_aliases(self):
event = Poll(
id="query",
question="Q?",
options=[PollOption(text="A1", voter_count=1)],
is_closed=True,
)
class MyHandler(PollHandler):
async def handle(self) -> Any:
assert self.event == event
assert self.question == self.event.question
assert self.options == self.event.options
return True
assert await MyHandler(event)