more_dots/tests/test_basic.py

#!/usr/bin/env python3
"""
LangChain + LangGraph 脚手架基础测试
"""

import unittest
import sys
import os
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')
if not os.path.exists(config_path):
    with open(config_path, 'w') as f:
        f.write("""
[General]
DEFAULT_MODEL_SECTION = gpt-4o
MAX_RETRIES = 1
TIMEOUT = 10

[gpt-4o]
MODEL_NAME = gpt-4o
OPENAI_API_KEY = your_openai_api_key_here

[gpt-3.5-turbo]
MODEL_NAME = gpt-3.5-turbo
OPENAI_API_KEY = your_openai_api_key_here
""")

from workflows.workflow_manager import WorkflowManager, WorkflowType


class TestWorkflowManager(unittest.TestCase):
    """测试 WorkflowManager 功能"""
    
    def setUp(self):
        """设置测试夹具"""
        self.manager = WorkflowManager()
    
    def test_get_available_workflows(self):
        """测试可用工作流返回"""
        workflows = self.manager.get_available_workflows()
        self.assertIsInstance(workflows, list)
        self.assertGreater(len(workflows), 0)
        self.assertIn("conversation", workflows)
        self.assertIn("tool_using", workflows)
    
    def test_get_workflow(self):
        """测试获取工作流实例"""
        conversation_workflow = self.manager.get_workflow(WorkflowType.CONVERSATION)
        self.assertIsNotNone(conversation_workflow)
        
        tool_workflow = self.manager.get_workflow(WorkflowType.TOOL_USING)
        self.assertIsNotNone(tool_workflow)
    
    def test_session_management(self):
        """测试会话创建与获取"""
        # 执行工作流以创建会话
        result = self.manager.execute_workflow(
            WorkflowType.CONVERSATION,
            "Hello, test session"
        )
        
        session_id = result["session_id"]
        self.assertIsNotNone(session_id)
        
        # 测试会话信息获取
        session_info = self.manager.get_session_info(session_id)
        self.assertIsNotNone(session_info)
        self.assertEqual(session_info["workflow_type"], WorkflowType.CONVERSATION)


class TestConfiguration(unittest.TestCase):
    """测试配置校验"""
    
    def test_config_loading(self):
        """测试能从 config.ini 加载配置"""
        from config import Config
        model_config = Config.get_model_config()
        self.assertIn('model', model_config)
        self.assertIn('api_key', model_config)


if __name__ == "__main__":
    unittest.main()