feat: Phase 3 - 工具层 + 测试 + 数据库迁移

tests/test_message_queue.py

@@ -0,0 +1,96 @@
+"""
+消息队列单元测试
+"""
+import pytest
+from app.services.message_queue import MessageQueue
+from unittest.mock import MagicMock, patch
+
+
+class TestMessageQueue:
+    """消息队列测试"""
+    
+    @pytest.fixture
+    def mock_redis(self):
+        """模拟 Redis 客户端"""
+        with patch('app.services.message_queue.redis_client') as mock:
+            mock.hset.return_value = 1
+            mock.rpush.return_value = 1
+            mock.lpop.return_value = None
+            mock.hgetall.return_value = {}
+            mock.delete.return_value = 1
+            mock.llen.return_value = 0
+            yield mock
+    
+    def test_enqueue(self, mock_redis):
+        """测试消息入队"""
+        message = MagicMock()
+        message.id = 'msg-123'
+        message.session_id = 'session-123'
+        message.sender_type = 'user'
+        message.sender_id = 'user-123'
+        message.content = 'Hello'
+        message.status = 'sent'
+        message.retry_count = 0
+        message.created_at = None
+        
+        result = MessageQueue.enqueue(message)
+        
+        assert result is True
+        mock_redis.hset.assert_called()
+        mock_redis.rpush.assert_called()
+    
+    def test_dequeue_empty(self, mock_redis):
+        """测试空队列出队"""
+        mock_redis.lpop.return_value = None
+        
+        result = MessageQueue.dequeue()
+        
+        assert result is None
+    
+    def test_dequeue_success(self, mock_redis):
+        """测试成功出队"""
+        mock_redis.lpop.return_value = 'msg-123'
+        mock_redis.hgetall.return_value = {
+            'id': 'msg-123',
+            'session_id': 'session-123',
+            'content': 'Hello',
+            'retry_count': '0',
+        }
+        
+        result = MessageQueue.dequeue()
+        
+        assert result is not None
+        assert result['id'] == 'msg-123'
+    
+    def test_ack(self, mock_redis):
+        """测试消息确认"""
+        result = MessageQueue.ack('msg-123')
+        
+        assert result is True
+        mock_redis.delete.assert_called()
+    
+    def test_retry_within_limit(self, mock_redis):
+        """测试重试（未超限）"""
+        mock_redis.hget.return_value = '1'
+        
+        result = MessageQueue.retry('msg-123')
+        
+        assert result is True
+        mock_redis.hset.assert_called()
+        mock_redis.rpush.assert_called()
+    
+    def test_retry_exceed_limit(self, mock_redis):
+        """测试重试（超限）"""
+        mock_redis.hget.return_value = '3'
+        
+        result = MessageQueue.retry('msg-123')
+        
+        assert result is False
+    
+    def test_get_pending_count(self, mock_redis):
+        """测试获取待处理数量"""
+        mock_redis.llen.return_value = 5
+        
+        count = MessageQueue.get_pending_count()
+        
+        assert count == 5