第一次提交

tests/__init__.py

@@ -0,0 +1 @@
+"""测试模块初始化文件"""

tests/test_agents.py

@@ -0,0 +1,212 @@
+"""
+Agent 单元测试模块
+测试各个 Agent 的基本功能
+"""
+import pytest
+import os
+import sys
+from pathlib import Path
+
+# 添加项目根目录到路径
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+
+class TestLLMConfig:
+    """测试 LLM 配置模块"""
+    
+    def test_get_llm_config(self):
+        """测试获取基本配置"""
+        from config.llm_config import get_llm_config
+        
+        config = get_llm_config(
+            model="qwen3.5-flash",
+            api_key="test_key",
+            temperature=0.7
+        )
+        
+        assert "config_list" in config
+        assert len(config["config_list"]) == 1
+        assert config["config_list"][0]["model"] == "qwen3.5-flash"
+        assert config["config_list"][0]["api_key"] == "test_key"
+        assert config["config_list"][0]["temperature"] == 0.7
+    
+    def test_get_agent_llm_config(self):
+        """测试获取特定 Agent 配置"""
+        from config.llm_config import get_agent_llm_config
+        
+        config = get_agent_llm_config("PM_Agent")
+        
+        assert "config_list" in config
+        assert config["config_list"][0]["model"] == "qwen3.5-flash"
+    
+    def test_system_prompts_exist(self):
+        """测试系统提示词存在"""
+        from config.llm_config import PM_PROMPT, QA_PROMPT, DEV_PROMPT, ORCH_PROMPT
+        
+        assert PM_PROMPT is not None
+        assert len(PM_PROMPT) > 100
+        
+        assert QA_PROMPT is not None
+        assert len(QA_PROMPT) > 100
+        
+        assert DEV_PROMPT is not None
+        assert len(DEV_PROMPT) > 100
+        
+        assert ORCH_PROMPT is not None
+        assert len(ORCH_PROMPT) > 100
+
+
+class TestLogger:
+    """测试日志模块"""
+    
+    def test_logger_creation(self):
+        """测试日志记录器创建"""
+        from utils.logger import ConversationLogger
+        
+        logger = ConversationLogger(log_dir="test_logs")
+        
+        assert logger is not None
+        assert logger.session_id is not None
+    
+    def test_log_message(self):
+        """测试消息记录"""
+        from utils.logger import ConversationLogger
+        import shutil
+        
+        # 清理测试目录
+        test_dir = Path("test_logs")
+        if test_dir.exists():
+            shutil.rmtree(test_dir)
+        
+        logger = ConversationLogger(log_dir="test_logs")
+        logger.log_message(
+            agent_name="Test_Agent",
+            message="这是一条测试消息",
+            role="assistant"
+        )
+        
+        assert len(logger.conversation_history) == 1
+        assert logger.conversation_history[0]["agent_name"] == "Test_Agent"
+        assert logger.conversation_history[0]["message"] == "这是一条测试消息"
+        
+        # 清理
+        shutil.rmtree(test_dir)
+    
+    def test_callback_handler(self):
+        """测试回调处理器"""
+        from utils.callback_handler import MessageCallbackHandler
+        
+        handler = MessageCallbackHandler()
+        
+        received_messages = []
+        
+        def test_callback(msg):
+            received_messages.append(msg)
+        
+        handler.register_callback(test_callback)
+        handler.on_message(
+            agent_name="Test_Agent",
+            message="测试回调消息"
+        )
+        
+        assert len(received_messages) == 1
+        assert received_messages[0]["agent_name"] == "Test_Agent"
+
+
+class TestAgentsCreation:
+    """测试 Agent 创建"""
+    
+    @pytest.mark.skipif(not os.getenv("DASHSCOPE_API_KEY"), reason="需要 API Key")
+    def test_create_pm_agent(self):
+        """测试创建 PM Agent"""
+        from agents import create_pm_agent
+        from config.llm_config import get_llm_config
+        
+        api_key = os.getenv("DASHSCOPE_API_KEY")
+        llm_config = get_llm_config(api_key=api_key)
+        
+        agent = create_pm_agent(llm_config=llm_config)
+        
+        assert agent is not None
+        assert agent.name == "PM_Agent"
+    
+    @pytest.mark.skipif(not os.getenv("DASHSCOPE_API_KEY"), reason="需要 API Key")
+    def test_create_qa_agent(self):
+        """测试创建 QA Agent"""
+        from agents import create_qa_agent
+        from config.llm_config import get_llm_config
+        
+        api_key = os.getenv("DASHSCOPE_API_KEY")
+        llm_config = get_llm_config(api_key=api_key)
+        
+        agent = create_qa_agent(llm_config=llm_config)
+        
+        assert agent is not None
+        assert agent.name == "QA_Agent"
+    
+    @pytest.mark.skipif(not os.getenv("DASHSCOPE_API_KEY"), reason="需要 API Key")
+    def test_create_dev_agent(self):
+        """测试创建 Dev Agent"""
+        from agents import create_dev_agent
+        from config.llm_config import get_llm_config
+        
+        api_key = os.getenv("DASHSCOPE_API_KEY")
+        llm_config = get_llm_config(api_key=api_key)
+        
+        agent = create_dev_agent(llm_config=llm_config)
+        
+        assert agent is not None
+        assert agent.name == "Dev_Agent"
+    
+    @pytest.mark.skipif(not os.getenv("DASHSCOPE_API_KEY"), reason="需要 API Key")
+    def test_create_orchestrator(self):
+        """测试创建 Orchestrator Agent"""
+        from agents import create_orchestrator_agent
+        from config.llm_config import get_llm_config
+        
+        api_key = os.getenv("DASHSCOPE_API_KEY")
+        llm_config = get_llm_config(api_key=api_key)
+        
+        agent = create_orchestrator_agent(llm_config=llm_config)
+        
+        assert agent is not None
+        assert agent.name == "Orchestrator"
+
+
+class TestWorkspaceManagement:
+    """测试工作空间管理"""
+    
+    def test_workspace_directory_exists(self):
+        """测试工作目录存在"""
+        workspace_dir = Path(__file__).parent.parent / "workspace"
+        workspace_dir.mkdir(parents=True, exist_ok=True)
+        
+        assert workspace_dir.exists()
+        assert workspace_dir.is_dir()
+    
+    def test_create_sample_files(self):
+        """测试创建示例文件"""
+        workspace_dir = Path(__file__).parent.parent / "workspace"
+        workspace_dir.mkdir(parents=True, exist_ok=True)
+        
+        # 创建测试文件
+        test_file = workspace_dir / "test_sample.txt"
+        test_content = "这是一个测试文件"
+        
+        with open(test_file, 'w', encoding='utf-8') as f:
+            f.write(test_content)
+        
+        assert test_file.exists()
+        
+        with open(test_file, 'r', encoding='utf-8') as f:
+            content = f.read()
+        
+        assert content == test_content
+        
+        # 清理
+        test_file.unlink()
+
+
+if __name__ == "__main__":
+    # 运行测试
+    pytest.main([__file__, "-v", "--tb=short"])