test: 补充知识处理、检索与 Agent 回归用例

覆盖切块、向量检索、工具调用和历史预算；模型相关用例使用桩。

backend/tests/core/test_redis.py

@@ -0,0 +1,157 @@
+"""
+Copyright 2024-2026 ReplyPier
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import pytest
+from unittest.mock import patch, MagicMock
+
+@patch('app.core.redis.redis')
+@patch('app.core.redis.settings')
+def test_init_redis_disabled(mock_settings, mock_redis):
+    """Test Redis initialization when Redis is disabled"""
+    from app.core.redis import init_redis
+    
+    # Set up the mock
+    mock_settings.REDIS_ENABLED = False
+    
+    # Call the function
+    result = init_redis()
+    
+    # Verify that no Redis connection was created
+    assert result is None
+    mock_redis.from_url.assert_not_called()
+
+@patch('app.core.redis.redis')
+@patch('app.core.redis.settings')
+def test_init_redis_standard_url(mock_settings, mock_redis):
+    """Test Redis initialization with a standard Redis URL"""
+    from app.core.redis import init_redis
+    
+    # Set up the mocks
+    mock_settings.REDIS_ENABLED = True
+    mock_settings.REDIS_URL = "redis://localhost:6379"
+    mock_redis_client = MagicMock()
+    mock_redis.from_url.return_value = mock_redis_client
+    
+    # Call the function
+    result = init_redis()
+    
+    # Verify Redis was initialized with the correct parameters
+    mock_redis.from_url.assert_called_once_with(
+        "redis://localhost:6379",
+        socket_timeout=5.0,
+        socket_connect_timeout=5.0,
+        decode_responses=True
+    )
+    mock_redis_client.ping.assert_called_once()
+    assert result == mock_redis_client
+
+@patch('app.core.redis.redis')
+@patch('app.core.redis.settings')
+def test_init_redis_aws_elasticache(mock_settings, mock_redis):
+    """Test Redis initialization with AWS ElastiCache URL (should use TLS)"""
+    from app.core.redis import init_redis
+    
+    # Set up the mocks
+    mock_settings.REDIS_ENABLED = True
+    mock_settings.REDIS_URL = "redis://my-cluster.cache.amazonaws.com:6379"
+    mock_redis_client = MagicMock()
+    mock_redis.from_url.return_value = mock_redis_client
+    
+    # Call the function
+    result = init_redis()
+    
+    # Verify Redis was initialized with TLS URL
+    mock_redis.from_url.assert_called_once_with(
+        "rediss://my-cluster.cache.amazonaws.com:6379",
+        socket_timeout=5.0,
+        socket_connect_timeout=5.0,
+        decode_responses=True
+    )
+    mock_redis_client.ping.assert_called_once()
+    assert result == mock_redis_client
+
+@patch('app.core.redis.redis')
+@patch('app.core.redis.settings')
+def test_init_redis_connection_error(mock_settings, mock_redis):
+    """Test Redis initialization with connection error"""
+    from app.core.redis import init_redis
+    
+    # Set up the mocks
+    mock_settings.REDIS_ENABLED = True
+    mock_settings.REDIS_URL = "redis://localhost:6379"
+    mock_redis_client = MagicMock()
+    mock_redis.from_url.return_value = mock_redis_client
+    mock_redis_client.ping.side_effect = Exception("Connection error")
+    
+    # Call the function
+    result = init_redis()
+    
+    # Verify Redis client was created but ping failed
+    mock_redis.from_url.assert_called_once()
+    mock_redis_client.ping.assert_called_once()
+    assert result is None
+
+@patch('app.core.redis.init_redis')
+@patch('app.core.redis.redis_client', None)  # Reset the global redis_client
+@patch('app.core.redis.settings')
+def test_get_redis_first_call(mock_settings, mock_init_redis):
+    """Test get_redis when called for the first time"""
+    from app.core.redis import get_redis
+    
+    # Set up the mocks
+    mock_settings.REDIS_ENABLED = True
+    mock_redis_client = MagicMock()
+    mock_init_redis.return_value = mock_redis_client
+    
+    # Call the function
+    result = get_redis()
+    
+    # Verify init_redis was called and the result is correct
+    mock_init_redis.assert_called_once()
+    assert result == mock_redis_client
+
+@patch('app.core.redis.redis_client', MagicMock())  # Simulate existing redis_client
+@patch('app.core.redis.init_redis')
+@patch('app.core.redis.settings')
+def test_get_redis_cached_client(mock_settings, mock_init_redis, monkeypatch):
+    """Test get_redis when a Redis client is already cached"""
+    from app.core.redis import get_redis, redis_client
+    
+    # Set up the mocks
+    mock_settings.REDIS_ENABLED = True
+    existing_client = redis_client  # Use the mocked redis_client
+    
+    # Call the function
+    result = get_redis()
+    
+    # Verify init_redis was not called and the existing client is returned
+    mock_init_redis.assert_not_called()
+    assert result == existing_client
+
+@patch('app.core.redis.redis_client', None)  # Reset the global redis_client
+@patch('app.core.redis.settings')
+def test_get_redis_disabled(mock_settings):
+    """Test get_redis when Redis is disabled"""
+    from app.core.redis import get_redis
+    
+    # Set up the mocks
+    mock_settings.REDIS_ENABLED = False
+    
+    # Call the function
+    result = get_redis()
+    
+    # Verify no Redis client was initialized or returned
+    assert result is None