python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-11 01:54:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Propagate update to context in router

Browse source
This commit is contained in:
Alex Root Junior 2020-01-26 02:34:32 +02:00
parent f0f1523142
commit 3dd5530241
3 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/test_dispatcher/test_handler/test_base.py
View file

@ -1,10 +1,11 @@
import asyncio
import datetime
from typing import Any
import pytest
from aiogram import Bot
from aiogram.api.types import Update
from aiogram.api.types import Chat, Message, Update
from aiogram.dispatcher.handler.base import BaseHandler
@ -27,7 +28,7 @@ class TestBaseClassBasedHandler:
assert await handler == 42
@pytest.mark.asyncio
async def test_bot_mixin_from_context(self):
async def test_bot_from_context(self):
event = Update(update_id=42)
handler = MyHandler(event=event, key=42)
bot = Bot("42:TEST")
@ -38,10 +39,20 @@ class TestBaseClassBasedHandler:
assert handler.bot == bot
@pytest.mark.asyncio
async def test_bot_mixin_from_data(self):
async def test_bot_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
def test_update_from_data(self):
event = Message(
message_id=42, chat=Chat(id=42, type="private"), date=datetime.datetime.now()
)
update = Update(update_id=42, message=event)
handler = MyHandler(event=event, update=update)
assert handler.event == event
assert handler.update == update