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

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

backend/tests/agents/test_chat_agent_end_chat.py

@@ -0,0 +1,244 @@
+"""
+Copyright 2024-2026 ChatterMate
+
+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 Mock, patch, MagicMock
+from datetime import datetime
+from app.agents.chat_agent import ChatAgent, ChatResponse
+from app.models.session_to_agent import SessionStatus
+from sqlalchemy.orm import Session
+
+
+@pytest.fixture
+def mock_db():
+    mock_session = MagicMock(spec=Session)
+    return mock_session
+
+
+@pytest.fixture
+def mock_agent_data():
+    agent = Mock()
+    agent.ask_for_rating = True  # Default agent setting
+    return agent
+
+
+@pytest.fixture
+def chat_agent(mock_agent_data, mock_db):
+    from uuid import uuid4
+    from unittest.mock import patch, Mock
+    
+    # Mock the agent data returned by the database
+    mock_jira_agent_data = Mock()
+    mock_jira_agent_data.jira_enabled = False
+    mock_jira_agent_data.transfer_to_human = False
+    mock_jira_agent_data.ask_for_rating = True  # Default agent setting
+    mock_jira_agent_data.instructions = ["You are a helpful assistant", "Be polite and professional"]
+    
+    # Mock AI config
+    mock_ai_config = Mock()
+    mock_ai_config.model_name = "gpt-4"
+    mock_ai_config.model_type = "openai"
+    mock_ai_config.encrypted_api_key = "test-key"
+    
+    # Mock AI config repository
+    
+    # Mock knowledge search tool
+    mock_knowledge_tool = Mock()
+    
+    # Mock environment variables
+    mock_env = {}
+    
+    # Mock the encrypted agent storage
+    mock_storage = Mock()
+    mock_storage.get_session_state.return_value = {"status": "active"}
+    
+    with patch('app.agents.chat_agent.get_db') as mock_get_db, \
+         patch('app.agents.chat_agent.JiraRepository') as mock_jira_repo, \
+         patch('app.tools.knowledge_search_byagent.KnowledgeSearchByAgent', return_value=mock_knowledge_tool), \
+         patch('app.agents.chat_agent.EncryptedPostgresAgentStorage', return_value=mock_storage), \
+         patch('app.agents.chat_agent.settings.DATABASE_URL', "mock://test"), \
+         patch.dict('os.environ', mock_env, clear=True):
+        
+        # Mock database session
+        mock_get_db.return_value = iter([mock_db])
+        
+        # Mock JiraRepository methods
+        mock_jira_repo.return_value.get_agent_with_jira_config.return_value = mock_jira_agent_data
+        
+        agent = ChatAgent(
+            api_key="test-key",
+            model_name="gpt-4",
+            model_type="openai",
+            org_id=str(uuid4()),
+            agent_id=str(uuid4()),
+            customer_id=str(uuid4()),
+            session_id=str(uuid4())
+        )
+        
+        # Mock the agent's database session
+        agent.db = mock_db
+        return agent
+
+
+@pytest.fixture
+def sample_response():
+    return ChatResponse(
+        message="Thank you for chatting with us. Goodbye!",
+        transfer_to_human=False,
+        transfer_reason=None,
+        transfer_description=None,
+        end_chat=True,
+        request_rating=False,
+        create_ticket=False
+    )
+
+
+class TestChatAgentEndChat:
+    
+    @pytest.mark.asyncio
+    async def test_handle_end_chat_force_rating_true(self, chat_agent, sample_response, mock_db):
+        """Test _handle_end_chat with force_rating=True"""
+        
+        # Mock session repository
+        mock_session_repo = Mock()
+        
+        # Create a fixed datetime for testing
+        fixed_datetime = datetime(2025, 7, 25, 23, 30, 11, 457575)
+        
+        with patch('app.agents.chat_agent.SessionToAgentRepository', return_value=mock_session_repo), \
+             patch.object(chat_agent, 'agent_data', None), \
+             patch('app.agents.chat_agent.datetime') as mock_datetime:
+            # Configure datetime.now() to return our fixed datetime
+            mock_datetime.now.return_value = fixed_datetime
+            
+            result = await chat_agent._handle_end_chat(
+                response_content=sample_response,
+                session_id=chat_agent.agent.session_id,
+                db=mock_db,
+                force_rating=True
+            )
+            
+            # Verify rating was forced to True
+            assert result.request_rating is True
+            assert "Would you please take a moment to rate your experience?" in result.message
+            
+            # Verify session was updated
+            mock_session_repo.update_session.assert_called_once_with(
+                chat_agent.agent.session_id,
+                {
+                    "status": SessionStatus.CLOSED,
+                    "end_chat_reason": None,
+                    "end_chat_description": None,
+                    "closed_at": fixed_datetime
+                }
+            )
+    
+    @pytest.mark.asyncio
+    async def test_handle_end_chat_force_rating_false(self, chat_agent, sample_response, mock_db):
+        """Test _handle_end_chat with force_rating=False"""
+        
+        # Mock session repository
+        mock_session_repo = Mock()
+        
+        with patch('app.agents.chat_agent.SessionToAgentRepository', return_value=mock_session_repo):
+            # Set agent to normally ask for rating, but force it to False
+            mock_agent = Mock()
+            mock_agent.ask_for_rating = True
+            with patch.object(chat_agent, 'agent_data', mock_agent):
+                result = await chat_agent._handle_end_chat(
+                    response_content=sample_response,
+                    session_id=chat_agent.agent.session_id,
+                    db=mock_db,
+                    force_rating=False
+                )
+                
+                # Verify rating was forced to False despite agent setting
+                assert result.request_rating is False
+                assert "Would you please take a moment to rate your experience?" not in result.message
+                
+                # Verify session was updated
+                mock_session_repo.update_session.assert_called_once()
+    
+    @pytest.mark.asyncio
+    async def test_handle_end_chat_force_rating_none_uses_agent_default(self, chat_agent, sample_response, mock_db):
+        """Test _handle_end_chat with force_rating=None uses agent's default"""
+        
+        # Mock session repository
+        mock_session_repo = Mock()
+        
+        with patch('app.agents.chat_agent.SessionToAgentRepository', return_value=mock_session_repo):
+            # Set agent to ask for rating
+            mock_agent = Mock()
+            mock_agent.ask_for_rating = True
+            with patch.object(chat_agent, 'agent_data', mock_agent):
+                result = await chat_agent._handle_end_chat(
+                    response_content=sample_response,
+                    session_id=chat_agent.agent.session_id,
+                    db=mock_db,
+                    force_rating=None  # Use agent's default
+                )
+                
+                # Verify agent's default setting was used
+                assert result.request_rating is True
+                assert "Would you please take a moment to rate your experience?" in result.message
+    
+    @pytest.mark.asyncio
+    async def test_handle_end_chat_force_rating_none_agent_disabled(self, chat_agent, sample_response, mock_db):
+        """Test _handle_end_chat with force_rating=None when agent has rating disabled"""
+        
+        # Mock session repository
+        mock_session_repo = Mock()
+        
+        with patch('app.agents.chat_agent.SessionToAgentRepository', return_value=mock_session_repo):
+            # Set agent to not ask for rating
+            mock_agent = Mock()
+            mock_agent.ask_for_rating = False
+            with patch.object(chat_agent, 'agent_data', mock_agent):
+                result = await chat_agent._handle_end_chat(
+                    response_content=sample_response,
+                    session_id=chat_agent.agent.session_id,
+                    db=mock_db,
+                    force_rating=None  # Use agent's default
+                )
+                
+                # Verify agent's default setting was used
+                assert result.request_rating is False
+                assert "Would you please take a moment to rate your experience?" not in result.message
+    
+   
+    @pytest.mark.asyncio
+    async def test_handle_end_chat_backward_compatibility(self, chat_agent, sample_response, mock_db):
+        """Test _handle_end_chat backward compatibility (no force_rating parameter)"""
+        
+        # Mock session repository
+        mock_session_repo = Mock()
+        
+        with patch('app.agents.chat_agent.SessionToAgentRepository', return_value=mock_session_repo):
+            # Set agent to ask for rating
+            mock_agent = Mock()
+            mock_agent.ask_for_rating = True
+            with patch.object(chat_agent, 'agent_data', mock_agent):
+                # Call without force_rating parameter (backward compatibility)
+                result = await chat_agent._handle_end_chat(
+                    response_content=sample_response,
+                    session_id=chat_agent.agent.session_id,
+                    db=mock_db
+                    # No force_rating parameter - should default to None
+                )
+                
+                # Verify agent's default setting was used
+                assert result.request_rating is True
+                assert "Would you please take a moment to rate your experience?" in result.message