undo fix, update signatures in tests

This commit is contained in:
darksidecat 2021-10-04 21:14:51 +03:00
parent bdd5bb0d16
commit bc32ff727d
5 changed files with 14 additions and 16 deletions

View file

@ -245,14 +245,14 @@ class TestBaseSession:
flag_after = False
@bot.session.middleware
async def my_middleware(b, method, make_request):
async def my_middleware(make_request, b, method):
nonlocal flag_before, flag_after
flag_before = True
try:
assert isinstance(b, Bot)
assert isinstance(method, TelegramMethod)
return await make_request(bot, method)
return await make_request(b, method)
finally:
flag_after = True