fix linter, fix ci, fix tests
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user