当前位置： 首页 > news >正文

如何用AI工具开发一个轻量化CRM系统（七）：AI生成pytest测试脚本

news 2025/9/27 5:51:28

要实现根据Flask接口文档自动生成测试脚本、自动执行多次并汇总结果，可以通过解析Swagger文档+动态生成pytest测试用例+测试结果汇总的流程完成。以下是完整的Python实现方案：

在这里插入图片描述

一、环境准备

  1. 安装依赖库：
    pip install requests pytest pytest-html flasgger
    • requests：发送HTTP请求；
    • pytest：测试框架，支持自动发现和执行测试；
    • pytest-html：生成HTML测试报告；
    • flasgger：Flask的Swagger文档生成库（确保你的Flask应用已集成，用于提供结构化的Swagger JSON）。

二、核心逻辑实现

1. 获取并解析Swagger文档

Flask应用通常通过/swagger.json提供结构化的Swagger文档（需集成flasgger）。我们需要先获取该文档，解析出所有接口的信息（路径、方法、参数、预期响应）。

import requests
from typing import Dict, List, Anydef get_swagger_doc(swagger_url: str) -> Dict[str, Any]:"""获取Swagger JSON文档"""response = requests.get(swagger_url)response.raise_for_status()  # 确保请求成功return response.json()def parse_swagger_interfaces(swagger_doc: Dict[str, Any]) -> List[Dict[str, Any]]:"""解析Swagger文档，提取接口信息（路径、方法、参数、响应）"""interfaces = []paths = swagger_doc.get("paths", {})for path, methods in paths.items():for method, details in methods.items():# 仅处理GET/POST/PUT/DELETE方法if method.lower() in ["get", "post", "put", "delete"]:interface = {"path": path,"method": method.upper(),"parameters": details.get("parameters", []),  # 接口参数（查询/请求体）"responses": details.get("responses", {})    # 预期响应}interfaces.append(interface)return interfaces
2. 生成测试用例

根据解析出的接口信息，生成覆盖正常流、异常流、边界条件的测试用例。例如，登录接口的测试用例包括：正常登录、错误用户名、错误密码、缺少参数等。

def generate_test_cases(interfaces: List[Dict[str, Any]]) -> List[Dict[str, Any]]:"""根据接口信息生成测试用例"""test_cases = []for interface in interfaces:path = interface["path"]method = interface["method"]# 1. 登录接口（/login，POST）if path == "/login" and method == "POST":test_cases.extend([{"name": "login_success",  # 测试用例名称"request": {"json": {"username": "admin", "password": "password123"}},"expected": {"status_code": 200, "json": {"access_token": str, "message": "登录成功"}}},{"name": "login_wrong_username","request": {"json": {"username": "nonexistent", "password": "password123"}},"expected": {"status_code": 401, "json": {"message": "用户名或密码错误"}}},{"name": "login_missing_username","request": {"json": {"password": "password123"}},"expected": {"status_code": 400, "json": {"message": "缺少必填字段: username"}}}])# 2. 客户列表接口（/crm/customers，GET）elif path == "/crm/customers" and method == "GET":test_cases.extend([{"name": "get_customers_success","request": {},"expected": {"status_code": 200, "json": {"customers": list, "pagination": dict}}},{"name": "get_customers_unauthorized","request": {},"expected": {"status_code": 401, "json": {"message": "缺少访问令牌"}}}])# 3. 创建客户接口（/crm/customers，POST）elif path == "/crm/customers" and method == "POST":test_cases.extend([{"name": "create_customer_success","request": {"json": {"name": "李四", "email": "lisi@example.com", "phone": "13900139000"}},"expected": {"status_code": 201, "json": {"id": int, "name": "李四", "email": "lisi@example.com", "message": "客户创建成功"}}},{"name": "create_customer_missing_name","request": {"json": {"email": "lisi@example.com", "phone": "13900139000"}},"expected": {"status_code": 400, "json": {"message": "缺少必填字段: name"}}}])return test_cases
3. 动态生成pytest测试函数

利用pytest的动态测试生成能力，为每个测试用例生成独立的测试函数，自动发送请求并验证响应。

def generate_test_function(test_case: Dict[str, Any]) -> callable:"""为单个测试用例生成pytest测试函数"""def test_func():# 构造请求URL和方法method = test_case["request"].get("method", "POST").upper()path = test_case["path"]url = f"http://localhost:8000{path}"  # 替换为你的API地址request_data = test_case["request"]# 发送HTTP请求if method == "POST":response = requests.post(url, json=request_data.get("json", {}))elif method == "GET":response = requests.get(url, params=request_data.get("params", {}))elif method == "PUT":response = requests.put(url, json=request_data.get("json", {}))elif method == "DELETE":response = requests.delete(url, params=request_data.get("params", {}))else:raise ValueError(f"不支持的请求方法: {method}")# 验证1：状态码是否符合预期expected_status = test_case["expected"]["status_code"]assert response.status_code == expected_status, \f"测试用例[{test_case['name']}]失败：预期状态码{expected_status}，实际{response.status_code}"# 验证2：响应体结构和类型是否符合预期expected_json = test_case["expected"]["json"]response_json = response.json()for key, expected_type in expected_json.items():assert key in response_json, \f"测试用例[{test_case['name']}]失败：响应缺少字段[{key}]"assert isinstance(response_json[key], expected_type), \f"测试用例[{test_case['name']}]失败：字段[{key}]类型错误（预期{expected_type.__name__}，实际{type(response_json[key]).__name__}）"return test_func
4. 主程序：注册测试用例并执行

将上述步骤整合，自动生成测试用例、注册测试函数，并通过pytest执行测试，生成HTML报告。

if __name__ == "__main__":# 1. 配置Swagger文档地址（根据你的Flask应用调整）SWAGGER_URL = "http://localhost:8000/swagger.json"try:# 2. 获取并解析Swagger文档swagger_doc = get_swagger_doc(SWAGGER_URL)interfaces = parse_swagger_interfaces(swagger_doc)print(f"✅ 成功解析到 {len(interfaces)} 个接口")# 3. 生成测试用例test_cases = generate_test_cases(interfaces)print(f"✅ 生成 {len(test_cases)} 个测试用例")# 4. 动态注册测试函数到pytestfor test_case in test_cases:test_name = f"test_{test_case['name']}"  # pytest测试函数命名规则test_func = generate_test_function(test_case)globals()[test_name] = test_func  # 将测试函数加入全局命名空间# 5. 执行测试（生成HTML报告）print("
🚀 开始运行测试...")pytest.main([__file__,          # 当前脚本"-v",              # 详细模式（显示每个测试用例的结果）"--html=report.html",  # 生成HTML报告"--self-contained-html"  # 报告包含所有依赖（无需额外文件）])except Exception as e:print(f"❌ 执行失败：{e}")exit(1)

三、使用说明

  1. 启动Flask应用：确保你的Flask应用运行在http://localhost:8000，并集成了flasgger（可通过/swagger.json访问Swagger文档）。
  2. 运行测试脚本：执行上述Python脚本：
    python test_crm_api.py
  3. 查看结果：
    • 命令行会输出详细的测试结果（通过/失败/错误）；
    • 生成report.html文件，打开后可查看测试用例详情、失败原因和汇总统计（通过率、失败数等）。

四、扩展功能

1. 自动执行多次测试

通过pytest的--count参数指定执行次数（例如执行5次）：

pytest.main([__file__,"-v","--html=report.html","--count=5"  # 执行5次测试
])
2. 更灵活的测试数据管理

若测试数据需要复用或参数化，可以使用pytest的@pytest.mark.parametrize装饰器。例如，登录接口的多组用户名/密码：

import pytest@pytest.mark.parametrize("username, password, expected_message", [("admin", "password123", "登录成功"),("nonexistent", "password123", "用户名或密码错误"),("admin", "wrongpassword", "用户名或密码错误")
])
def test_login_parametrized(username, password, expected_message):# 发送请求并验证response = requests.post("http://localhost:8000/login", json={"username": username, "password": password})assert response.status_code == 200 if "成功" in expected_message else 401assert expected_message in response.json()["message"]
3. 集成CI/CD

可以将测试脚本集成到CI/CD pipeline（如GitHub Actions、Jenkins），每次代码提交后自动执行测试并生成报告。

总结

该方案通过解析Swagger文档实现测试用例的自动生成，通过pytest动态测试实现自动执行，通过HTML报告实现结果汇总。完全符合“根据接口文档生成测试脚本、自动执行多次、汇总结果”的需求，且支持灵活扩展（如参数化测试、CI/CD集成）。

http://www.dtcms.com/a/409921.html

相关文章：

  • qData：一站式开源数据中台
  • 国外中文网站排行在线图片编辑网站源码
  • [数据结构]优先级队列
  • ARM内部寄存器
  • Laravel + UniApp AES加密/解密
  • 5G开户时切片配置参数详解
  • 面向新质生产力，职业院校“人工智能”课程教学解决方案
  • wap网站如何做福建外贸网站
  • ElasticSearch-提高篇
  • 第6篇、Flask 表单处理与用户认证完全指南：从零到实战
  • Visual Studio 2013 Update 4 中文版安装步骤（带TFS支持）附安装包​
  • 珠海 网站建设注册安全工程师题库
  • 上手 cpp-httplib：轻量级 C++ HTTP 库的安装与实战指南
  • 突破文档型数据库迁移困境：金仓多模方案破解电子证照系统国产化难题
  • 网站手机客户端开发wordpress制造商单页
  • Net 》》C# 》》try finally 执行顺序
  • 在 Unity 项目中使用 FFMpeg 进行音频转码（WAV 转 MP3）
  • 使用Java将Word文件转换为PNG图片
  • 如何用Fail2ban保护Linux服务器？防止SSH暴力破解教程
  • 开源 C# 快速开发（五）自定义控件--仪表盘
  • 华为FreeClip 2耳夹耳机：让「戴着不摘」成为新的使用习惯
  • 算法继续刷起-2025年09月26日
  • AI笔记在学习与工作中的高效运用
  • QML学习笔记（十四）QML的自定义模块
  • ubuntu一键安装vscode: 使用官方 APT 仓库
  • python做网站的 框架企业邮箱什么样子
  • 学习游戏制作记录（爆炸敌人的制作）
  • 第三方软件登记测试机构：【软件登记测试机构HTML5测试技术】
  • 元宇宙的宠物经济：虚拟宠物的饲养与交易
  • 如何在 CentOS 7 上安装 bzip2-libs-1.0.6-13.el7.x86_64.rpm 文件