init

tests/test_basic.py

@@ -6,6 +6,9 @@ LangChain + LangGraph 脚手架基础测试
 import unittest
 import sys
 import os
+from unittest.mock import patch
+from typing import cast
+from langchain_core.messages import AIMessage
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.ini')
@@ -27,6 +30,23 @@ OPENAI_API_KEY = your_openai_api_key_here
 """)
 
 from workflows.workflow_manager import WorkflowManager, WorkflowType
+from services.common.app_errors import AppError, ErrorCode
+
+
+class FakeModel:
+    def invoke(self, messages):
+        if len(messages) == 2 and getattr(messages[0], 'content', '').startswith('You are a translation and normalization assistant'):
+            return AIMessage(content=messages[1].content)
+        return AIMessage(content='fallback')
+
+
+class EmptyTemplateMatcher:
+    def match(self, normalized_text: str):
+        return {
+            'table_name': None,
+            'candidates': [],
+            'raw': {'query': normalized_text},
+        }
 
 
 class TestWorkflowManager(unittest.TestCase):
@@ -34,8 +54,16 @@ class TestWorkflowManager(unittest.TestCase):
     
     def setUp(self):
         """设置测试夹具"""
-        self.manager = WorkflowManager()
-    
+        self.model_patcher = patch("agent.core.base_agent.create_chat_model", lambda model_section=None: FakeModel())
+        self.matcher_patcher = patch("agent.core.nodes.get_template_matcher", lambda: EmptyTemplateMatcher())
+        self.model_patcher.start()
+        self.matcher_patcher.start()
+        self.manager = WorkflowManager(enable_multi_turn=False)
+
+    def tearDown(self):
+        self.matcher_patcher.stop()
+        self.model_patcher.stop()
+
     def test_get_available_workflows(self):
         """测试可用工作流返回"""
         workflows = self.manager.get_available_workflows()
@@ -68,6 +96,94 @@ class TestWorkflowManager(unittest.TestCase):
         self.assertIsNotNone(session_info)
         self.assertEqual(session_info["workflow_type"], WorkflowType.CONVERSATION)
 
+    def test_conversation_session_is_temporarily_stateless(self):
+        """测试关闭多轮后，同一 session_id 也不会自动累积对话历史"""
+        first = self.manager.execute_workflow(
+            WorkflowType.CONVERSATION,
+            "Hello, first turn",
+            session_id="session-1",
+        )
+        second = self.manager.execute_workflow(
+            WorkflowType.CONVERSATION,
+            "Hello, second turn",
+            session_id="session-1",
+        )
+        other = self.manager.execute_workflow(
+            WorkflowType.CONVERSATION,
+            "Hello, other session",
+            session_id="session-2",
+        )
+
+        session_one = self.manager.get_session_info("session-1")
+        session_two = self.manager.get_session_info("session-2")
+
+        self.assertEqual(first["session_id"], "session-1")
+        self.assertEqual(second["session_id"], "session-1")
+        self.assertEqual(other["session_id"], "session-2")
+        self.assertNotIn("conversation_history", session_one)
+        self.assertNotIn("last_context", session_one)
+        self.assertNotIn("conversation_history", session_two)
+        self.assertNotIn("last_context", session_two)
+        self.assertEqual(len(first["result"].get("conversation_history") or []), 2)
+        self.assertEqual(len(second["result"].get("conversation_history") or []), 2)
+        self.assertEqual(len(other["result"].get("conversation_history") or []), 2)
+
+    def test_conversation_session_memory_can_be_enabled(self):
+        """测试开启多轮后，同一 session_id 会保存并复用会话级历史"""
+        manager = WorkflowManager(enable_multi_turn=True)
+
+        first = manager.execute_workflow(
+            WorkflowType.CONVERSATION,
+            "Hello, first turn",
+            session_id="session-enabled",
+        )
+        second = manager.execute_workflow(
+            WorkflowType.CONVERSATION,
+            "Hello, second turn",
+            session_id="session-enabled",
+        )
+
+        session_info = manager.get_session_info("session-enabled")
+
+        self.assertIn("conversation_history", session_info)
+        self.assertIn("last_context", session_info)
+        self.assertGreaterEqual(len(session_info["conversation_history"]), 4)
+        self.assertEqual(len(first["result"].get("conversation_history") or []), 2)
+        self.assertGreaterEqual(len(second["result"].get("conversation_history") or []), 4)
+
+    def test_reusing_session_id_with_different_workflow_raises(self):
+        """测试同一个 session_id 不能绑定到不同工作流"""
+        self.manager.execute_workflow(
+            WorkflowType.CONVERSATION,
+            "Hello, test session",
+            session_id="shared-session",
+        )
+
+        with self.assertRaises(ValueError):
+            self.manager.execute_workflow(
+                WorkflowType.TOOL_USING,
+                "2 + 2",
+                session_id="shared-session",
+            )
+
+    def test_execute_workflow_rejects_none_user_input(self):
+        """测试空 query 会在进入 Agent 之前被拒绝"""
+        with self.assertRaises(AppError) as ctx:
+            self.manager.execute_workflow(WorkflowType.CONVERSATION, cast(str, None))
+
+        self.assertEqual(ctx.exception.code, ErrorCode.INVALID_REQUEST)
+        self.assertEqual(ctx.exception.status_code, 400)
+        self.assertEqual(ctx.exception.detail, {"field": "user_input", "reason": "missing_or_blank"})
+
+    def test_execute_workflow_rejects_blank_user_input(self):
+        """测试全空白 query 会在进入 Agent 之前被拒绝"""
+        with self.assertRaises(AppError) as ctx:
+            self.manager.execute_workflow(WorkflowType.CONVERSATION, "   ")
+
+        self.assertEqual(ctx.exception.code, ErrorCode.INVALID_REQUEST)
+        self.assertEqual(ctx.exception.status_code, 400)
+        self.assertEqual(ctx.exception.detail, {"field": "user_input", "reason": "missing_or_blank"})
+
 
 class TestConfiguration(unittest.TestCase):
     """测试配置校验"""