Add auto unban functionality and update related tests and dependencies

2025-08-27 20:56:22 +03:00
parent 748670816f
commit 86b6903920
9 changed files with 737 additions and 48 deletions
--- a/tests/test_auto_unban_scheduler.py
+++ b/tests/test_auto_unban_scheduler.py
@@ -0,0 +1,288 @@
+import pytest
+import asyncio
+from datetime import datetime, timezone, timedelta
+from unittest.mock import Mock, patch, AsyncMock
+
+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.return_value = {
+            123: "test_user1",
+            456: "test_user2"
+        }
+        mock_db.delete_user_blacklist.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'
+            }
+        }
+        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):
+        """Тест успешного выполнения автоматического разбана"""
+        # Настройка моков
+        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):
+        """Тест разбана когда нет пользователей для разблокировки"""
+        # Настройка моков
+        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.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):
+        """Тест разбана с частичными ошибками"""
+        # Настройка моков
+        mock_get_instance.return_value = mock_bdf
+        mock_bot_db.get_users_for_unblock_today.return_value = {
+            123: "test_user1",
+            456: "test_user2"
+        }
+        # Первый вызов успешен, второй - ошибка
+        mock_bot_db.delete_user_blacklist.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):
+        """Тест разбана с исключением"""
+        # Настройка моков
+        mock_get_instance.return_value = mock_bdf
+        mock_bot_db.get_users_for_unblock_today.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']
+    
+    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 "test_user1" in report
+        assert "456 (test_user2)" in report
+    
+    @pytest.mark.asyncio
+    @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'
+    
+    @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):
+        """Тест отправки отчета об ошибке"""
+        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'
+    
+    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:
+            
+            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):
+        """Тест ручного запуска разбана"""
+        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
+
+
+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[:4].isdigit()  # Год
+        assert today[5:7].isdigit()  # Месяц
+        assert today[8:10].isdigit()  # День
+
+
+@pytest.mark.asyncio
+class TestAsyncOperations:
+    """Тесты асинхронных операций"""
+    
+    @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'
+            }
+        }
+        mock_get_instance.return_value = mock_bdf
+        
+        mock_bot_db = Mock()
+        mock_bot_db.get_users_for_unblock_today.return_value = {123: "test_user"}
+        mock_bot_db.delete_user_blacklist.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)
+        mock_bot.send_message.assert_called_once()