fix linter, fix ci, fix tests
This commit is contained in:
@@ -5,15 +5,12 @@
|
||||
from unittest.mock import AsyncMock, MagicMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from helper_bot.handlers.admin.exceptions import AdminError
|
||||
from helper_bot.handlers.admin.utils import (
|
||||
escape_html,
|
||||
format_ban_confirmation,
|
||||
format_user_info,
|
||||
handle_admin_error,
|
||||
return_to_admin_menu,
|
||||
)
|
||||
from helper_bot.handlers.admin.utils import (escape_html,
|
||||
format_ban_confirmation,
|
||||
format_user_info,
|
||||
handle_admin_error,
|
||||
return_to_admin_menu)
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@@ -124,7 +121,9 @@ class TestReturnToAdminMenu:
|
||||
state.set_state.assert_called_once_with("ADMIN")
|
||||
mock_kb.assert_called_once()
|
||||
assert message.answer.call_count == 1
|
||||
message.answer.assert_called_with("Вернулись в меню", reply_markup="keyboard_markup")
|
||||
message.answer.assert_called_with(
|
||||
"Вернулись в меню", reply_markup="keyboard_markup"
|
||||
)
|
||||
|
||||
async def test_additional_message_sent_first(self):
|
||||
"""При additional_message сначала отправляется оно, затем меню."""
|
||||
@@ -145,7 +144,9 @@ class TestReturnToAdminMenu:
|
||||
|
||||
assert message.answer.call_count == 2
|
||||
message.answer.assert_any_call("Дополнительный текст")
|
||||
message.answer.assert_any_call("Вернулись в меню", reply_markup="keyboard_markup")
|
||||
message.answer.assert_any_call(
|
||||
"Вернулись в меню", reply_markup="keyboard_markup"
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@@ -190,7 +191,5 @@ class TestHandleAdminError:
|
||||
message, ValueError("Что-то пошло не так"), state, "test"
|
||||
)
|
||||
|
||||
message.answer.assert_any_call(
|
||||
"Произошла внутренняя ошибка. Попробуйте позже."
|
||||
)
|
||||
message.answer.assert_any_call("Произошла внутренняя ошибка. Попробуйте позже.")
|
||||
state.set_state.assert_called_once_with("ADMIN")
|
||||
|
||||
Reference in New Issue
Block a user