python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Lint changes

Browse source
This commit is contained in:
Desiders 2024-06-13 23:25:28 +03:00
parent 94cff575f2
commit d980bd7724
1 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/test_dispatcher/test_middlewares/test_user_context.py
View file

@ -7,7 +7,15 @@ from aiogram.dispatcher.middlewares.user_context import (
EventContext,
UserContextMiddleware,
)
from aiogram.types import Chat, Update, User, ChatBoostUpdated, ChatBoost, ChatBoostSourcePremium, ChatBoostSourceGiftCode
from aiogram.types import (
Chat,
Update,
User,
ChatBoostUpdated,
ChatBoost,
ChatBoostSourcePremium,
ChatBoostSourceGiftCode,
)
async def next_handler(*args, **kwargs):
@ -53,7 +61,9 @@ class TestUserContextMiddleware:
expiration_date = datetime.now()
source = ChatBoostSourcePremium(user=user)
boost = ChatBoost(boost_id="Test", add_date=add_date, expiration_date=expiration_date, source=source)
boost = ChatBoost(
boost_id="Test", add_date=add_date, expiration_date=expiration_date, source=source
)
update = Update(update_id=42, chat_boost=ChatBoostUpdated(chat=chat, boost=boost))
await middleware(next_handler, update, data)
@ -70,7 +80,9 @@ class TestUserContextMiddleware:
data.clear()
source = ChatBoostSourceGiftCode(user=user)
boost = ChatBoost(boost_id="Test", add_date=add_date, expiration_date=expiration_date, source=source)
boost = ChatBoost(
boost_id="Test", add_date=add_date, expiration_date=expiration_date, source=source
)
update = Update(update_id=42, chat_boost=ChatBoostUpdated(chat=chat, boost=boost))
await middleware(next_handler, update, data)