Add prototype of class-based handlers

tests/test_dispatcher/test_handler/test_base.py

@@ -0,0 +1,47 @@
+import asyncio
+from typing import Any
+
+import pytest
+
+from aiogram import Bot
+from aiogram.api.types import Update
+from aiogram.dispatcher.handler.base import BaseHandler
+
+
+class MyHandler(BaseHandler):
+    async def handle(self) -> Any:
+        await asyncio.sleep(0.1)
+        return 42
+
+
+class TestBaseClassBasedHandler:
+    @pytest.mark.asyncio
+    async def test_base_handler(self):
+        event = Update(update_id=42)
+        handler = MyHandler(event=event, key=42)
+
+        assert handler.event == event
+        assert handler.data["key"] == 42
+        assert hasattr(handler, "bot")
+        assert not hasattr(handler, "filters")
+        assert await handler == 42
+
+    @pytest.mark.asyncio
+    async def test_bot_mixin_from_context(self):
+        event = Update(update_id=42)
+        handler = MyHandler(event=event, key=42)
+        bot = Bot("42:TEST")
+
+        assert handler.bot is None
+
+        Bot.set_current(bot)
+        assert handler.bot == bot
+
+    @pytest.mark.asyncio
+    async def test_bot_mixin_from_data(self):
+        event = Update(update_id=42)
+        bot = Bot("42:TEST")
+        handler = MyHandler(event=event, key=42, bot=bot)
+
+        assert "bot" in handler.data
+        assert handler.bot == bot

tests/test_dispatcher/test_handler/test_message.py

@@ -0,0 +1,28 @@
+import datetime
+from typing import Any
+
+import pytest
+
+from aiogram.api.types import Chat, Message, User
+from aiogram.dispatcher.handler.message import MessageHandler
+
+
+class MyHandler(MessageHandler):
+    async def handle(self) -> Any:
+        return self.event.text
+
+
+class TestClassBasedMessageHandler:
+    @pytest.mark.asyncio
+    async def test_message_handler(self):
+        event = Message(
+            message_id=42,
+            date=datetime.datetime.now(),
+            text="test",
+            chat=Chat(id=42, type="private"),
+            from_user=User(id=42, is_bot=False, first_name="Test"),
+        )
+        handler = MyHandler(event=event)
+
+        assert handler.from_user == event.from_user
+        assert handler.chat == event.chat