AI生成测试用例:一个Prompt模板让AI从Excel模板生成自动化脚本

// test_data 示例{  "username": "test_user_001",  "password": "Test@123",  "email": "test@example.com",  "profile": {    "age": 25,    "city": "Shanghai"  }}// expected_response 示例{  "code": 0,  "message": "success",  "data.userId": "^[A-Z0-9]{8,16}{test_data.username}"}

USR_001    # 用户模块-第1个用例ORD_003    # 订单模块-第3个用例PAY_007    # 支付模块-第7个用例

# 良好示例"Given用户已登录_When提交有效订单_Then返回创建成功""Given商品库存为0_When用户尝试下单_Then提示库存不足"# 糟糕示例"测试下单功能""订单创建""库存不足场景"

工作簿结构：├── 接口定义        # 接口元数据├── 测试用例        # 主用例数据├── 测试数据池      # 可复用的测试数据├── 断言规则库      # 预定义断言模式└── 业务规则        # 关联业务逻辑说明

SYSTEM_PROMPT = """你是一个专业的测试用例生成专家，擅长将Excel中的测试数据转换为可执行的自动化测试代码。## 输出要求1. 只输出Python代码，不要包含任何解释性文字2. 使用Pytest框架3. 代码必须可以直接运行4. 每个测试函数对应Excel中的一个测试用例5. 测试函数命名规范：test_{case_id}_{简短描述}## 代码规范1. 使用requests库发送HTTP请求2. 使用pytest.mark装饰器标记用例优先级3. 使用conftest.py中定义的fixture获取base_url和headers4. 断言要具体，明确assert什么字段、什么值5. 对于有依赖的用例，在函数docstring中标注依赖关系## 变量引用规则- `{expected.field}` 引用Excel中expected_response字段- 支持正则表达式断言，格式为：assert re.match(pattern, value)## 错误处理1. 网络异常：使用pytest.skip跳过用例，标注原因2. 断言失败：使用assert的具体消息说明期望值和实际值3. 依赖失败：使用pytest.mark.dependency标记依赖关系"""USER_PROMPT_TEMPLATE = """请根据以下测试数据生成Pytest测试用例代码：### Excel数据

### 项目配置- 基础URL: {base_url}- 接口前缀: {api_prefix}- 认证方式: {auth_type}### 测试类命名{test_class_name}### 注意事项1. 登录接口无需生成测试用例（已在conftest.py中处理）2. 需要认证的接口使用 @pytest.mark.usefixtures("auth_token") 装饰器3. POST/PUT请求的body使用 json= 参数4. 响应断言优先验证 code/message 字段，再验证业务数据"""

def build_user_prompt(excel_data: dict, config: dict) -> str:    """构建用户Prompt"""    return USER_PROMPT_TEMPLATE.format(        excel_data=_format_excel_data(excel_data),        base_url=config.get("base_url", "http://localhost:8080"),        api_prefix=config.get("api_prefix", "/api/v1"),        auth_type=config.get("auth_type", "Bearer Token"),        test_class_name=f"Test{excel_data['module_name'].title().replace('_', '')}"    )def _format_excel_data(data: list) -> str:    """格式化Excel数据为Markdown表格"""    if not data:        return ""    headers = list(data[0].keys())    lines = ["| " + " | ".join(headers) + " |"]    lines.append("| " + " | ".join(["---"] * len(headers)) + " |")    for row in data:        values = [str(row.get(h, ""))[:100] for h in headers]  # 截断过长内容        lines.append("| " + " | ".join(values) + " |")    return "\n".join(lines)

class ExcelParser:    def __init__(self, file_path: str):        self.file_path = file_path        self.warnings = []        self.errors = []    def parse_with_validation(self) -> list[dict]:        """带验证的解析"""        raw_data = self._read_excel()        # 第一级：基础格式校验        validated_data = []        for idx, row in enumerate(raw_data, start=2):  # Excel行号从2开始            try:                parsed_row = self._validate_row(row, idx)                if parsed_row:                    validated_data.append(parsed_row)            except MissingRequiredField as e:                self.warnings.append(f"行{idx}: 缺失必填字段 {e.field}，已跳过")            except InvalidFieldValue as e:                self.warnings.append(f"行{idx}: 字段 {e.field} 值无效，已使用默认值")        # 第二级：数据完整性检查        if len(validated_data) == 0:            raise EmptyDataError("没有有效的测试数据")        # 第三级：跨行关联检查        validated_data = self._resolve_references(validated_data)        return validated_data    def _validate_row(self, row: dict, row_num: int) -> dict:        """验证单行数据"""        # 检查必填字段        required_fields = ['case_id', 'case_title', 'http_method', 'api_path']        for field in required_fields:            if not row.get(field):                raise MissingRequiredField(field)        # 规范化http_method        method = row['http_method'].upper().strip()        if method not in ['GET', 'POST', 'PUT', 'DELETE', 'PATCH']:            method = 'GET'  # 默认值            self.warnings.append(f"行{row_num}: http_method无效，已设为GET")        row['http_method'] = method        # 解析JSON字段        for json_field in ['test_data', 'expected_response', 'assertion_rules']:            if row.get(json_field) and isinstance(row[json_field], str):                try:                    row[json_field] = json.loads(row[json_field])                except json.JSONDecodeError as e:                    row[json_field] = {}  # 空字典作为默认值                    self.warnings.append(f"行{row_num}: {json_field} JSON解析失败，使用空对象")        return row    def _resolve_references(self, data: list) -> list:        """解析数据引用"""        # 构建数据池索引        data_pool = {}        for row in data:            if row.get('data_id'):                data_pool[row['data_id']] = row        # 替换引用        for row in data:            self._replace_refs(row, data_pool)        return data    def _replace_refs(self, row: dict, pool: dict):        """递归替换引用"""        for field, value in row.items():            if isinstance(value, str) and value.startswith('”}USR_REG_002Given已注册用户_When再次注册_Then提示用户名已存在POST/api/v1/users/registerP1异常{“username”:“existing_user”,“password”:“Test@123456”,“email”:“existing@test.com”}200{“code”:1001,“message”:“用户名已存在”}USR_REG_003Given无效邮箱_When注册_Then提示邮箱格式错误POST/api/v1/users/registerP1异常{“username”:“new_user”,“password”:“Test@123”,“email”:“invalid-email”}200{“code”:1002,“message”:“邮箱格式不正确”}PTS_EAN_001Given用户已登录_When积分获取_Then返回积分变动POST/api/v1/points/earnP0功能{“userId”:“{USR_REG_001.userId}”}200{“code”:0,“data.balance”:“^\d+", user_id), \            f"userId格式不符，期望^[A-Z0-9]{{8,16}}", str(balance)), \            f"balance应为数字，实际: {balance}"

import astimport reclass CodeQualityValidator:    """代码质量验证器"""    def __init__(self):        self.issues = []    def validate_file(self, file_path: str) -> dict:        """验证测试文件"""        with open(file_path, 'r', encoding='utf-8') as f:            code = f.read()        # 语法检查        try:            ast.parse(code)        except SyntaxError as e:            self.issues.append(f"语法错误: {e}")        # 命名规范检查        naming_pattern = r'def (test_\w+)'        matches = re.findall(naming_pattern, code)        for func_name in matches:            if not re.match(r'test_[A-Z]{2,}_\d+', func_name):                self.issues.append(f"命名不规范: {func_name}")        return {            "file": file_path,            "issues": self.issues,            "passed": len(self.issues) == 0        }