fix linter, fix ci, fix tests

This commit is contained in:
2026-02-02 00:46:44 +03:00
parent 68041037bd
commit d87d4e492e
93 changed files with 1042 additions and 862 deletions

View File

@@ -6,7 +6,6 @@ from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from aiogram.types import CallbackQuery, Message, Update
from helper_bot.middlewares.rate_limit_middleware import RateLimitMiddleware
@@ -25,7 +24,9 @@ class TestRateLimitMiddleware:
"""Мок handler."""
return AsyncMock(return_value="handler_result")
async def test_event_with_message_calls_rate_limiter(self, middleware, mock_handler):
async def test_event_with_message_calls_rate_limiter(
self, middleware, mock_handler
):
"""При событии с message вызывается rate_limiter.send_with_rate_limit."""
event = MagicMock(spec=Message)
event.message = None
@@ -49,7 +50,9 @@ class TestRateLimitMiddleware:
mock_handler.assert_called_once_with(event, data)
assert result == "rate_limited_result"
async def test_update_with_message_calls_rate_limiter(self, middleware, mock_handler):
async def test_update_with_message_calls_rate_limiter(
self, middleware, mock_handler
):
"""При Update с message извлекается chat_id и вызывается rate_limiter."""
message = MagicMock(spec=Message)
message.chat = MagicMock()
@@ -69,7 +72,9 @@ class TestRateLimitMiddleware:
mock_send.assert_called_once()
assert mock_send.call_args[0][1] == 99999
async def test_event_without_message_calls_handler_directly(self, middleware, mock_handler):
async def test_event_without_message_calls_handler_directly(
self, middleware, mock_handler
):
"""При событии без message (например CallbackQuery) handler вызывается напрямую."""
event = MagicMock(spec=CallbackQuery)
event.message = None
@@ -93,6 +98,7 @@ class TestRateLimitMiddleware:
"send_with_rate_limit",
new_callable=AsyncMock,
) as mock_send:
async def call_passed_handler(inner_handler, chat_id):
return await inner_handler()