test_item_gen/modules/test_standard_manager.py

#  @line_count 250
"""测试规范管理器，整合规范加载、选择和Prompt构建"""
from typing import List, Dict, Any, Optional, TYPE_CHECKING
from .test_standard_loader import TestStandardLoader
from .standard_selector import StandardSelector

if TYPE_CHECKING:
    from .api_client import APIClient


class TestStandardManager:
    """测试规范管理器，负责管理测试规范、选择适用规范、构建Prompt"""
    
    def __init__(self, loader: Optional[TestStandardLoader] = None,
                 selector: Optional[StandardSelector] = None,
                 api_client: Optional['APIClient'] = None):
        """
        初始化测试规范管理器
        
        Args:
            loader: 测试规范加载器，如果为None则创建新实例
            selector: 规范选择器，如果为None则创建新实例
            api_client: API客户端，用于AI辅助选择
        """
        self.loader = loader or TestStandardLoader()
        self.api_client = api_client
        self.selector = selector or StandardSelector(self.loader, api_client)
    
    def get_applicable_standards(self, requirement: Dict[str, Any], 
                                use_ai: bool = True) -> List[str]:
        """
        为需求获取适用的测试规范
        
        Args:
            requirement: 需求字典
            use_ai: 是否使用AI辅助选择
            
        Returns:
            测试规范ID列表
        """
        return self.selector.select_standards(requirement, use_ai=use_ai)
    
    def build_prompt(self, requirement: Dict[str, Any], 
                    standard_ids: Optional[List[str]] = None,
                    include_full_content: bool = False) -> str:
        """
        构建包含测试规范的分层Prompt
        
        Args:
            requirement: 需求字典
            standard_ids: 测试规范ID列表，如果为None则自动选择
            include_full_content: 是否包含完整规范内容（默认只包含摘要）
            
        Returns:
            格式化后的Prompt文本
        """
        # 如果没有指定规范，自动选择
        if standard_ids is None:
            standard_ids = self.get_applicable_standards(requirement)
        
        # Level 1: 测试规范摘要
        standards_summary = self._build_standards_summary(standard_ids, include_full_content)
        
        # Level 2: 需求信息
        requirement_info = self._format_requirement(requirement)
        
        # 构建完整Prompt
        prompt = f"""你是一位资深的软件测试专家，请严格按照以下测试规范生成测试用例。

【适用的测试规范】
根据需求类型"{requirement.get('requirement_type', '未知')}"，以下测试规范适用：

{standards_summary}

【需求信息】
{requirement_info}

【生成要求】
1. 必须覆盖上述测试规范中的所有关键测试点
2. 每个适用的测试规范至少生成1个测试项
3. 测试用例要详细、可执行，包含：
   - 前置条件
   - 详细测试步骤（编号清晰）
   - 预期结果
   - 优先级（高/中/低）
4. 确保覆盖正常功能、异常情况、边界条件
5. 每个测试项应标注引用的测试规范（standard_reference字段）

【输出格式】
以JSON格式输出，格式如下：
{{
  "test_items": [
    {{
      "name": "测试项名称",
      "test_type": "功能测试",
      "priority": "高",
      "test_objective": "测试目标",
      "standard_reference": "功能测试",
      "test_cases": [
        {{
          "name": "测试用例名称",
          "preconditions": "前置条件",
          "test_steps": ["步骤1", "步骤2", "步骤3"],
          "expected_result": "预期结果",
          "priority": "高",
          "test_type": "功能测试"
        }}
      ]
    }}
  ]
}}"""
        
        return prompt
    
    def _build_standards_summary(self, standard_ids: List[str], 
                                include_full: bool = False) -> str:
        """
        构建测试规范摘要
        
        Args:
            standard_ids: 测试规范ID列表
            include_full: 是否包含完整内容
            
        Returns:
            格式化后的规范摘要
        """
        if not standard_ids:
            return "无适用的测试规范"
        
        return self.loader.format_standards_summary(standard_ids)
    
    def _format_requirement(self, requirement: Dict[str, Any]) -> str:
        """
        格式化需求信息
        
        Args:
            requirement: 需求字典
            
        Returns:
            格式化后的需求信息文本
        """
        lines = []
        lines.append(f"需求编号：{requirement.get('requirement_id', 'N/A')}")
        lines.append(f"需求类型：{requirement.get('requirement_type', 'N/A')}")
        lines.append(f"需求描述：{requirement.get('description', 'N/A')}")
        
        module_name = requirement.get('module_name', '')
        if module_name:
            lines.append(f"所属模块：{module_name}")
        
        # 接口信息（如果有）
        interface_info = requirement.get('interface_info')
        if interface_info:
            lines.append("\n【接口信息】")
            if interface_info.get('接口名称'):
                lines.append(f"接口名称：{interface_info['接口名称']}")
            if interface_info.get('接口类型'):
                lines.append(f"接口类型：{interface_info['接口类型']}")
            if interface_info.get('来源'):
                lines.append(f"来源：{interface_info['来源']}")
            if interface_info.get('目的地'):
                lines.append(f"目的地：{interface_info['目的地']}")
        
        return "\n".join(lines)
    
    def get_standards_summary(self, standard_ids: List[str]) -> str:
        """
        获取测试规范摘要（便捷方法）
        
        Args:
            standard_ids: 测试规范ID列表
            
        Returns:
            规范摘要文本
        """
        return self.loader.format_standards_summary(standard_ids)
    
    def confirm_test_types(self, requirement: Dict[str, Any], 
                          candidate_standards: List[str]) -> List[str]:
        """
        确认最终使用的测试类型（两阶段生成策略的阶段1）
        
        Args:
            requirement: 需求字典
            candidate_standards: 候选测试规范ID列表
            
        Returns:
            确认后的测试规范ID列表
        """
        # 如果候选类型不多（<=3），直接使用
        if len(candidate_standards) <= 3:
            return candidate_standards
        
        # 如果候选类型较多，可以进一步筛选
        # 这里简化处理，直接返回前5个
        return candidate_standards[:5]