style: isort + black
This commit is contained in:
@@ -3,231 +3,248 @@ from datetime import datetime, timedelta, timezone
|
||||
from unittest.mock import AsyncMock, Mock, patch
|
||||
|
||||
import pytest
|
||||
from helper_bot.utils.auto_unban_scheduler import (AutoUnbanScheduler,
|
||||
get_auto_unban_scheduler)
|
||||
|
||||
from helper_bot.utils.auto_unban_scheduler import (
|
||||
AutoUnbanScheduler,
|
||||
get_auto_unban_scheduler,
|
||||
)
|
||||
|
||||
|
||||
class TestAutoUnbanScheduler:
|
||||
"""Тесты для класса AutoUnbanScheduler"""
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def scheduler(self):
|
||||
"""Создает экземпляр планировщика для тестов"""
|
||||
return AutoUnbanScheduler()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_bot_db(self):
|
||||
"""Создает мок базы данных"""
|
||||
mock_db = Mock()
|
||||
mock_db.get_users_for_unblock_today = AsyncMock(return_value={
|
||||
123: "test_user1",
|
||||
456: "test_user2"
|
||||
})
|
||||
mock_db.get_users_for_unblock_today = AsyncMock(
|
||||
return_value={123: "test_user1", 456: "test_user2"}
|
||||
)
|
||||
mock_db.delete_user_blacklist = AsyncMock(return_value=True)
|
||||
return mock_db
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_bdf(self):
|
||||
"""Создает мок фабрики зависимостей"""
|
||||
mock_factory = Mock()
|
||||
mock_factory.settings = {
|
||||
'Telegram': {
|
||||
'group_for_logs': '-1001234567890',
|
||||
'important_logs': '-1001234567891'
|
||||
"Telegram": {
|
||||
"group_for_logs": "-1001234567890",
|
||||
"important_logs": "-1001234567891",
|
||||
}
|
||||
}
|
||||
return mock_factory
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_bot(self):
|
||||
"""Создает мок бота"""
|
||||
mock_bot = Mock()
|
||||
mock_bot.send_message = AsyncMock()
|
||||
return mock_bot
|
||||
|
||||
|
||||
def test_scheduler_initialization(self, scheduler):
|
||||
"""Тест инициализации планировщика"""
|
||||
assert scheduler.bot_db is not None
|
||||
assert scheduler.scheduler is not None
|
||||
assert scheduler.bot is None
|
||||
|
||||
|
||||
def test_set_bot(self, scheduler, mock_bot):
|
||||
"""Тест установки бота"""
|
||||
scheduler.set_bot(mock_bot)
|
||||
assert scheduler.bot == mock_bot
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
async def test_auto_unban_users_success(self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot):
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_auto_unban_users_success(
|
||||
self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot
|
||||
):
|
||||
"""Тест успешного выполнения автоматического разбана"""
|
||||
# Настройка моков
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
scheduler.bot_db = mock_bot_db
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
# Выполнение теста
|
||||
await scheduler.auto_unban_users()
|
||||
|
||||
|
||||
# Проверки
|
||||
mock_bot_db.get_users_for_unblock_today.assert_called_once()
|
||||
assert mock_bot_db.delete_user_blacklist.call_count == 2
|
||||
mock_bot.send_message.assert_called_once()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
async def test_auto_unban_users_no_users(self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot):
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_auto_unban_users_no_users(
|
||||
self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot
|
||||
):
|
||||
"""Тест разбана когда нет пользователей для разблокировки"""
|
||||
# Настройка моков
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
mock_bot_db.get_users_for_unblock_today = AsyncMock(return_value={})
|
||||
scheduler.bot_db = mock_bot_db
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
# Выполнение теста
|
||||
await scheduler.auto_unban_users()
|
||||
|
||||
|
||||
# Проверки
|
||||
mock_bot_db.get_users_for_unblock_today.assert_called_once()
|
||||
mock_bot_db.delete_user_blacklist.assert_not_called()
|
||||
mock_bot.send_message.assert_not_called()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
async def test_auto_unban_users_partial_failure(self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot):
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_auto_unban_users_partial_failure(
|
||||
self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot
|
||||
):
|
||||
"""Тест разбана с частичными ошибками"""
|
||||
# Настройка моков
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
mock_bot_db.get_users_for_unblock_today = AsyncMock(return_value={
|
||||
123: "test_user1",
|
||||
456: "test_user2"
|
||||
})
|
||||
mock_bot_db.get_users_for_unblock_today = AsyncMock(
|
||||
return_value={123: "test_user1", 456: "test_user2"}
|
||||
)
|
||||
# Первый вызов успешен, второй - ошибка
|
||||
mock_bot_db.delete_user_blacklist = AsyncMock(side_effect=[True, False])
|
||||
scheduler.bot_db = mock_bot_db
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
# Выполнение теста
|
||||
await scheduler.auto_unban_users()
|
||||
|
||||
|
||||
# Проверки
|
||||
assert mock_bot_db.delete_user_blacklist.call_count == 2
|
||||
mock_bot.send_message.assert_called_once()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
async def test_auto_unban_users_exception(self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot):
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_auto_unban_users_exception(
|
||||
self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot
|
||||
):
|
||||
"""Тест разбана с исключением"""
|
||||
# Настройка моков
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
mock_bot_db.get_users_for_unblock_today = AsyncMock(side_effect=Exception("Database error"))
|
||||
mock_bot_db.get_users_for_unblock_today = AsyncMock(
|
||||
side_effect=Exception("Database error")
|
||||
)
|
||||
scheduler.bot_db = mock_bot_db
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
# Выполнение теста
|
||||
await scheduler.auto_unban_users()
|
||||
|
||||
|
||||
# Проверки
|
||||
mock_bot.send_message.assert_called_once()
|
||||
# Проверяем, что сообщение об ошибке было отправлено
|
||||
call_args = mock_bot.send_message.call_args
|
||||
assert "Ошибка автоматического разбана" in call_args[1]['text']
|
||||
|
||||
assert "Ошибка автоматического разбана" in call_args[1]["text"]
|
||||
|
||||
def test_generate_report(self, scheduler):
|
||||
"""Тест генерации отчета"""
|
||||
users = {123: "test_user1", 456: "test_user2"}
|
||||
failed_users = ["456 (test_user2)"]
|
||||
|
||||
|
||||
report = scheduler._generate_report(1, 1, failed_users, users)
|
||||
|
||||
|
||||
assert "Отчет об автоматическом разбане" in report
|
||||
assert "Успешно разблокировано: 1" in report
|
||||
assert "Ошибок: 1" in report
|
||||
assert "ID: 123" in report
|
||||
assert "456 (test_user2)" in report
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_send_report(self, mock_get_instance, scheduler, mock_bdf, mock_bot):
|
||||
"""Тест отправки отчета"""
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
report = "Test report"
|
||||
await scheduler._send_report(report)
|
||||
|
||||
|
||||
# Проверяем, что send_message был вызван
|
||||
mock_bot.send_message.assert_called_once()
|
||||
|
||||
|
||||
# Проверяем аргументы вызова
|
||||
call_args = mock_bot.send_message.call_args
|
||||
assert call_args[1]['text'] == report
|
||||
assert call_args[1]['parse_mode'] == 'HTML'
|
||||
|
||||
assert call_args[1]["text"] == report
|
||||
assert call_args[1]["parse_mode"] == "HTML"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
async def test_send_error_report(self, mock_get_instance, scheduler, mock_bdf, mock_bot):
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_send_error_report(
|
||||
self, mock_get_instance, scheduler, mock_bdf, mock_bot
|
||||
):
|
||||
"""Тест отправки отчета об ошибке"""
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
error_msg = "Test error"
|
||||
await scheduler._send_error_report(error_msg)
|
||||
|
||||
|
||||
# Проверяем, что send_message был вызван
|
||||
mock_bot.send_message.assert_called_once()
|
||||
|
||||
|
||||
# Проверяем аргументы вызова
|
||||
call_args = mock_bot.send_message.call_args
|
||||
assert "Ошибка автоматического разбана" in call_args[1]['text']
|
||||
assert error_msg in call_args[1]['text']
|
||||
assert call_args[1]['parse_mode'] == 'HTML'
|
||||
|
||||
assert "Ошибка автоматического разбана" in call_args[1]["text"]
|
||||
assert error_msg in call_args[1]["text"]
|
||||
assert call_args[1]["parse_mode"] == "HTML"
|
||||
|
||||
def test_start_scheduler(self, scheduler):
|
||||
"""Тест запуска планировщика"""
|
||||
with patch.object(scheduler.scheduler, 'add_job') as mock_add_job, \
|
||||
patch.object(scheduler.scheduler, 'start') as mock_start:
|
||||
|
||||
with (
|
||||
patch.object(scheduler.scheduler, "add_job") as mock_add_job,
|
||||
patch.object(scheduler.scheduler, "start") as mock_start,
|
||||
):
|
||||
|
||||
scheduler.start_scheduler()
|
||||
|
||||
|
||||
mock_add_job.assert_called_once()
|
||||
mock_start.assert_called_once()
|
||||
|
||||
|
||||
def test_stop_scheduler(self, scheduler):
|
||||
"""Тест остановки планировщика"""
|
||||
# Сначала запускаем планировщик
|
||||
scheduler.start_scheduler()
|
||||
|
||||
|
||||
# Проверяем, что планировщик запущен
|
||||
assert scheduler.scheduler.running
|
||||
|
||||
|
||||
# Теперь останавливаем (должно пройти без ошибок)
|
||||
scheduler.stop_scheduler()
|
||||
|
||||
|
||||
# Проверяем, что метод выполнился без исключений
|
||||
# APScheduler может не сразу остановиться, но это нормально
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
async def test_run_manual_unban(self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot):
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_run_manual_unban(
|
||||
self, mock_get_instance, scheduler, mock_bot_db, mock_bdf, mock_bot
|
||||
):
|
||||
"""Тест ручного запуска разбана"""
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
mock_bot_db.get_users_for_unblock_today.return_value = {}
|
||||
scheduler.bot_db = mock_bot_db
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
await scheduler.run_manual_unban()
|
||||
|
||||
|
||||
mock_bot_db.get_users_for_unblock_today.assert_called_once()
|
||||
|
||||
|
||||
class TestGetAutoUnbanScheduler:
|
||||
"""Тесты для функции get_auto_unban_scheduler"""
|
||||
|
||||
|
||||
def test_get_auto_unban_scheduler(self):
|
||||
"""Тест получения глобального экземпляра планировщика"""
|
||||
scheduler = get_auto_unban_scheduler()
|
||||
assert isinstance(scheduler, AutoUnbanScheduler)
|
||||
|
||||
|
||||
# Проверяем, что возвращается один и тот же экземпляр
|
||||
scheduler2 = get_auto_unban_scheduler()
|
||||
assert scheduler is scheduler2
|
||||
@@ -235,17 +252,17 @@ class TestGetAutoUnbanScheduler:
|
||||
|
||||
class TestDateHandling:
|
||||
"""Тесты для обработки дат"""
|
||||
|
||||
|
||||
def test_moscow_timezone(self):
|
||||
"""Тест работы с московским временем"""
|
||||
scheduler = AutoUnbanScheduler()
|
||||
|
||||
|
||||
# Проверяем, что дата формируется в правильном формате
|
||||
moscow_tz = timezone(timedelta(hours=3))
|
||||
today = datetime.now(moscow_tz).strftime("%Y-%m-%d")
|
||||
|
||||
|
||||
assert len(today) == 10 # YYYY-MM-DD
|
||||
assert today.count('-') == 2
|
||||
assert today.count("-") == 2
|
||||
assert today[:4].isdigit() # Год
|
||||
assert today[5:7].isdigit() # Месяц
|
||||
assert today[8:10].isdigit() # День
|
||||
@@ -254,35 +271,37 @@ class TestDateHandling:
|
||||
@pytest.mark.asyncio
|
||||
class TestAsyncOperations:
|
||||
"""Тесты асинхронных операций"""
|
||||
|
||||
@patch('helper_bot.utils.auto_unban_scheduler.get_global_instance')
|
||||
|
||||
@patch("helper_bot.utils.auto_unban_scheduler.get_global_instance")
|
||||
async def test_async_auto_unban_flow(self, mock_get_instance):
|
||||
"""Тест полного асинхронного потока разбана"""
|
||||
# Создаем моки
|
||||
mock_bdf = Mock()
|
||||
mock_bdf.settings = {
|
||||
'Telegram': {
|
||||
'group_for_logs': '-1001234567890',
|
||||
'important_logs': '-1001234567891'
|
||||
"Telegram": {
|
||||
"group_for_logs": "-1001234567890",
|
||||
"important_logs": "-1001234567891",
|
||||
}
|
||||
}
|
||||
mock_get_instance.return_value = mock_bdf
|
||||
|
||||
|
||||
mock_bot_db = Mock()
|
||||
mock_bot_db.get_users_for_unblock_today = AsyncMock(return_value={123: "test_user"})
|
||||
mock_bot_db.get_users_for_unblock_today = AsyncMock(
|
||||
return_value={123: "test_user"}
|
||||
)
|
||||
mock_bot_db.delete_user_blacklist = AsyncMock(return_value=True)
|
||||
|
||||
|
||||
mock_bot = Mock()
|
||||
mock_bot.send_message = AsyncMock()
|
||||
|
||||
|
||||
# Создаем планировщик
|
||||
scheduler = AutoUnbanScheduler()
|
||||
scheduler.bot_db = mock_bot_db
|
||||
scheduler.set_bot(mock_bot)
|
||||
|
||||
|
||||
# Выполняем разбан
|
||||
await scheduler.auto_unban_users()
|
||||
|
||||
|
||||
# Проверяем результаты
|
||||
mock_bot_db.get_users_for_unblock_today.assert_called_once()
|
||||
mock_bot_db.delete_user_blacklist.assert_called_once_with(123)
|
||||
|
||||
Reference in New Issue
Block a user