From 1e89a4b5804589ff517a9f2287df74f1a64d30e8 Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Sun, 8 Jan 2023 16:40:04 +0200 Subject: [PATCH] Make context manager in tests compatible with Python 3.8 --- tests/test_utils/test_callback_answer.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/tests/test_utils/test_callback_answer.py b/tests/test_utils/test_callback_answer.py index 8c0d76c2..cf641f46 100644 --- a/tests/test_utils/test_callback_answer.py +++ b/tests/test_utils/test_callback_answer.py @@ -198,16 +198,13 @@ class TestCallbackAnswerMiddleware: stack.append("answer") middleware = CallbackAnswerMiddleware() - with ( - patch( - "aiogram.utils.callback_answer.CallbackAnswerMiddleware.construct_callback_answer", - new_callable=MagicMock, - side_effect=lambda **kwargs: CallbackAnswer(**{"answered": False, **properties}), - ), - patch( - "aiogram.utils.callback_answer.CallbackAnswerMiddleware.answer", - new=answer, - ), + with patch( + "aiogram.utils.callback_answer.CallbackAnswerMiddleware.construct_callback_answer", + new_callable=MagicMock, + side_effect=lambda **kwargs: CallbackAnswer(**{"answered": False, **properties}), + ), patch( + "aiogram.utils.callback_answer.CallbackAnswerMiddleware.answer", + new=answer, ): await middleware(handler, event, {})