aiogram/tests/test_dispatcher/test_handler/test_poll.py

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,
            is_anonymous=False,
            type="quiz",
            allows_multiple_answers=False,
            total_voter_count=0,
            correct_option_id=0,
        )

        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)