Python3+pytest+allure接口自动化测试框架搭建
myzbx 2025-03-20 16:59 43 浏览
项目结构
安装依赖
pip install pytest requests pytest-html pytest-xdist jsonpath
核心代码实现
配置文件 (config/config.py)
class Config:
# 环境配置
BASE_URL = "https://api.example.com"
TIMEOUT = 10
# 测试账号
USERNAME = "test_user"
PASSWORD = "test_password"
# 日志配置
LOG_LEVEL = "INFO"
API客户端 (api/client.py)
import requests
import json
import logging
from config.config import Config
class APIClient:
def __init__(self):
self.base_url = Config.BASE_URL
self.session = requests.Session()
self.timeout = Config.TIMEOUT
def get(self, url, params=None, headers=None):
return self._request("GET", url, params=params, headers=headers)
def post(self, url, data=None, json=None, headers=None):
return self._request("POST", url, data=data, json=json, headers=headers)
def put(self, url, data=None, json=None, headers=None):
return self._request("PUT", url, data=data, json=json, headers=headers)
def delete(self, url, headers=None):
return self._request("DELETE", url, headers=headers)
def _request(self, method, url, **kwargs):
full_url = self.base_url + url
logging.info(f"Request: {method} {full_url}")
if "headers" not in kwargs or kwargs["headers"] is None:
kwargs["headers"] = {"Content-Type": "application/json"}
if "timeout" not in kwargs:
kwargs["timeout"] = self.timeout
try:
response = self.session.request(method, full_url, **kwargs)
logging.info(f"Response status: {response.status_code}")
return response
except Exception as e:
logging.error(f"Request error: {e}")
raise
测试用例 (tests/test_api.py)
import pytest
import json
from api.client import APIClient
class TestAPI:
@pytest.fixture(scope="class")
def api_client(self):
return APIClient()
def test_get_users(self, api_client):
"""测试获取用户列表接口"""
response = api_client.get("/users")
assert response.status_code == 200
data = response.json()
assert "data" in data
assert isinstance(data["data"], list)
def test_create_user(self, api_client):
"""测试创建用户接口"""
user_data = {
"name": "测试用户",
"email": "test@example.com",
"password": "password123"
}
response = api_client.post("/users", json=user_data)
assert response.status_code == 201
data = response.json()
assert data["name"] == user_data["name"]
assert data["email"] == user_data["email"]
def test_update_user(self, api_client):
"""测试更新用户接口"""
# 假设我们已经知道用户ID为1
user_id = 1
update_data = {"name": "更新名称"}
response = api_client.put(f"/users/{user_id}", json=update_data)
assert response.status_code == 200
data = response.json()
assert data["name"] == update_data["name"]
pytest配置 (conftest.py)
import pytest
import logging
import os
from datetime import datetime
# 设置日志
def setup_logging():
log_dir = "logs"
if not os.path.exists(log_dir):
os.makedirs(log_dir)
log_file = f"{log_dir}/test_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(message)s",
handlers=[
logging.FileHandler(log_file),
logging.StreamHandler()
]
)
def pytest_configure(config):
setup_logging()
@pytest.fixture(scope="session", autouse=True)
def setup_teardown():
# 测试前置操作
logging.info("测试开始")
yield
# 测试后置操作
logging.info("测试结束")
测试运行脚本 (run.py)
import pytest
import argparse
import os
from datetime import datetime
def run_tests():
parser = argparse.ArgumentParser(description="运行API自动化测试")
parser.add_argument("--html", action="store_true", help="生成HTML报告")
parser.add_argument("--parallel", action="store_true", help="并行运行测试")
args = parser.parse_args()
# 创建报告目录
report_dir = "reports"
if not os.path.exists(report_dir):
os.makedirs(report_dir)
# 构建pytest参数
pytest_args = ["-v", "tests/"]
if args.html:
report_file = f"{report_dir}/report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.html"
pytest_args.extend(["--html", report_file, "--self-contained-html"])
if args.parallel:
pytest_args.extend(["-n", "auto"])
# 运行测试
pytest.main(pytest_args)
if __name__ == "__main__":
run_tests()
执行测试
# 普通执行
python run.py
# 生成HTML报告
python run.py --html
# 并行执行测试
python run.py --parallel
# 同时生成报告和并行执行
python run.py --html --parallel
添加Allure报告
更新依赖
pip install pytest requests pytest-html pytest-xdist jsonpath allure-pytest
安装Allure命令行工具Windows
scoop install allure
更新测试用例 (tests/test_api.py)
import pytest
import json
import allure
from api.client import APIClient
@allure.epic("API测试")
@allure.feature("用户管理接口")
class TestAPI:
@pytest.fixture(scope="class")
def api_client(self):
return APIClient()
@allure.story("获取用户列表")
@allure.title("测试获取所有用户")
@allure.severity(allure.severity_level.NORMAL)
def test_get_users(self, api_client):
"""测试获取用户列表接口"""
with allure.step("发送GET请求到/users"):
response = api_client.get("/users")
with allure.step("验证响应状态码"):
assert response.status_code == 200
with allure.step("验证响应数据结构"):
data = response.json()
assert "data" in data
assert isinstance(data["data"], list)
allure.attach(json.dumps(data, indent=2),
"响应数据",
allure.attachment_type.JSON)
@allure.story("创建新用户")
@allure.title("测试创建用户功能")
@allure.severity(allure.severity_level.CRITICAL)
def test_create_user(self, api_client):
"""测试创建用户接口"""
user_data = {
"name": "测试用户",
"email": "test@example.com",
"password": "password123"
}
allure.attach(json.dumps(user_data, indent=2),
"请求数据",
allure.attachment_type.JSON)
with allure.step("发送POST请求到/users"):
response = api_client.post("/users", json=user_data)
with allure.step("验证响应状态码"):
assert response.status_code == 201
with allure.step("验证创建的用户数据"):
data = response.json()
assert data["name"] == user_data["name"]
assert data["email"] == user_data["email"]
allure.attach(json.dumps(data, indent=2),
"响应数据",
allure.attachment_type.JSON)
@allure.story("更新用户")
@allure.title("测试更新用户信息")
@allure.severity(allure.severity_level.HIGH)
def test_update_user(self, api_client):
"""测试更新用户接口"""
# 假设我们已经知道用户ID为1
user_id = 1
update_data = {"name": "更新名称"}
allure.attach(json.dumps(update_data, indent=2),
"请求数据",
allure.attachment_type.JSON)
with allure.step(f"发送PUT请求到/users/{user_id}"):
response = api_client.put(f"/users/{user_id}", json=update_data)
with allure.step("验证响应状态码"):
assert response.status_code == 200
with allure.step("验证更新后的用户数据"):
data = response.json()
assert data["name"] == update_data["name"]
allure.attach(json.dumps(data, indent=2),
"响应数据",
allure.attachment_type.JSON)
更新API客户端 (api/client.py)
import requests
import json
import logging
import allure
from config.config import Config
class APIClient:
def __init__(self):
self.base_url = Config.BASE_URL
self.session = requests.Session()
self.timeout = Config.TIMEOUT
def get(self, url, params=None, headers=None):
return self._request("GET", url, params=params, headers=headers)
def post(self, url, data=None, json=None, headers=None):
return self._request("POST", url, data=data, json=json, headers=headers)
def put(self, url, data=None, json=None, headers=None):
return self._request("PUT", url, data=data, json=json, headers=headers)
def delete(self, url, headers=None):
return self._request("DELETE", url, headers=headers)
def _request(self, method, url, **kwargs):
full_url = self.base_url + url
logging.info(f"Request: {method} {full_url}")
# 记录请求信息到Allure
if "json" in kwargs and kwargs["json"] is not None:
allure.attach(json.dumps(kwargs["json"], indent=2, ensure_ascii=False),
f"{method} {url} - Request Body",
allure.attachment_type.JSON)
elif "data" in kwargs and kwargs["data"] is not None:
allure.attach(str(kwargs["data"]),
f"{method} {url} - Request Body",
allure.attachment_type.TEXT)
if "headers" not in kwargs or kwargs["headers"] is None:
kwargs["headers"] = {"Content-Type": "application/json"}
if "timeout" not in kwargs:
kwargs["timeout"] = self.timeout
try:
response = self.session.request(method, full_url, **kwargs)
logging.info(f"Response status: {response.status_code}")
# 记录响应信息到Allure
try:
response_body = response.json()
allure.attach(json.dumps(response_body, indent=2, ensure_ascii=False),
f"{method} {url} - Response Body",
allure.attachment_type.JSON)
except:
allure.attach(response.text,
f"{method} {url} - Response Body",
allure.attachment_type.TEXT)
return response
except Exception as e:
logging.error(f"Request error: {e}")
allure.attach(str(e),
f"{method} {url} - Exception",
allure.attachment_type.TEXT)
raise
更新conftest.py
import pytest
import logging
import os
import allure
from datetime import datetime
# 设置日志
def setup_logging():
log_dir = "logs"
if not os.path.exists(log_dir):
os.makedirs(log_dir)
log_file = f"{log_dir}/test_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(message)s",
handlers=[
logging.FileHandler(log_file),
logging.StreamHandler()
]
)
return log_file
def pytest_configure(config):
log_file = setup_logging()
# 添加环境信息到Allure报告
allure.environment(
python_version=platform.python_version(),
pytest_version=pytest.__version__,
platform=platform.system(),
host=Config.BASE_URL
)
@pytest.fixture(scope="session", autouse=True)
def setup_teardown():
# 测试前置操作
logging.info("测试开始")
yield
# 测试后置操作
logging.info("测试结束")
# 添加钩子函数记录测试结果
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call):
outcome = yield
report = outcome.get_result()
if report.when == "call":
if report.failed:
# 失败时捕获页面截图或其他信息
allure.attach(
f"测试用例: {item.function.__name__} 失败",
"失败信息",
allure.attachment_type.TEXT
)
更新运行脚本 (run.py)
import pytest
import argparse
import os
import shutil
import subprocess
from datetime import datetime
def run_tests():
parser = argparse.ArgumentParser(description="运行API自动化测试")
parser.add_argument("--html", action="store_true", help="生成HTML报告")
parser.add_argument("--allure", action="store_true", help="生成Allure报告")
parser.add_argument("--parallel", action="store_true", help="并行运行测试")
parser.add_argument("--clean", action="store_true", help="清理报告目录")
args = parser.parse_args()
# 创建报告目录
report_dir = "reports"
allure_result_dir = "allure-results"
allure_report_dir = f"{report_dir}/allure-report"
if args.clean:
if os.path.exists(report_dir):
shutil.rmtree(report_dir)
if os.path.exists(allure_result_dir):
shutil.rmtree(allure_result_dir)
if not os.path.exists(report_dir):
os.makedirs(report_dir)
# 构建pytest参数
pytest_args = ["-v", "tests/"]
if args.html:
report_file = f"{report_dir}/report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.html"
pytest_args.extend(["--html", report_file, "--self-contained-html"])
if args.allure:
pytest_args.extend(["--alluredir", allure_result_dir])
if args.parallel:
pytest_args.extend(["-n", "auto"])
# 运行测试
pytest.main(pytest_args)
# 生成Allure报告
if args.allure:
# 创建Allure报告目录
if not os.path.exists(allure_report_dir):
os.makedirs(allure_report_dir)
print("正在生成Allure报告...")
try:
subprocess.run(["allure", "generate", allure_result_dir, "-o", allure_report_dir, "--clean"], check=True)
print(f"Allure报告已生成: {os.path.abspath(allure_report_dir)}")
# 打开Allure报告
print("正在打开Allure报告...")
subprocess.run(["allure", "open", allure_report_dir], check=True)
except subprocess.CalledProcessError as e:
print(f"生成Allure报告失败: {e}")
except FileNotFoundError:
print("未找到allure命令,请确保Allure已正确安装")
if __name__ == "__main__":
run_tests()
添加缺少的导入到conftest.py中
import platform
from config.config import Config
执行测试并生成Allure报告
# 清理历史报告并生成Allure报告
python run.py --allure --clean
# 同时生成HTML和Allure报告
python run.py --html --allure
# 并行执行测试并生成Allure报告
python run.py --allure --parallel
相关推荐
- 26岁中科大教授攻克世界数学难题(中科大26岁数学家)
-
硬核研究世界顶尖数学期刊发表其研究论文“陈杲特任教授年仅26岁。他2008年入读中科大少年班,2012年赴纽约州立大学石溪分校,师从陈秀雄教授攻读博士。2017年博士毕业后历任普林斯顿高等研究院博士...
- CSP-J/S倒计时 如何做好充足准备?
-
CSP-J/S计算机软件能力认证作为信奥系列赛的首场测试,重要性也就不言而喻了。初试的题目往往是被忽略的环节,但初试是进入复试必须经过之路,成绩低的话无法晋级复试,就更别提复试奖项或者晋级NOIP...
- 信息学竞赛CSP- J/S满分小学生是怎么养成的
-
2024年四川省CSP-J的复赛中,一共有11名选手拿到400分满分,其中有一名小学生,让所有人惊讶又惊喜——嘉祥成华小学六年级学生邱明夷。大家都亲切的叫他小邱邱,因为他刚来新易元学信奥的时候是小学四...
- CSP-J/S什么情况容易得零分(csp-j/s是什么意思)
-
信息學奥赛CSP复赛啊,有这几个点不注意就很容易得0分,第一点呢就是复赛采用的是oi赛制,就是说如果结束后提交代码就无法在比赛的过程中,测试你的代码是否正确,只要有一点点错误就可能会导致0分,一定要严...
- 素数分布的解析理论:π(x)、Li(x)和J(x)的严格数学框架
-
本文从解析数论视角严格阐述素数计数函数π(x)、对数积分Li(x)及黎曼素数函数J(x)之间的数学关系。通过引入黎曼ζ函数的零点分布理论,构建了素数定理的精确表达式,并给出误差项的解析结构。一、基本定...
- 陈杲一本书三年读了三遍(陈红看一本书,三天看完)
-
【陈杲一本书三年读了三遍】14岁读大学、23岁获博士学位、26岁解出J方程和超临界厄米特—杨振宁—米尔斯方程的变形这一世界难题,用数学在相对论与量子力学间架起“新桥”。“天才”陈杲说自己“天赋是有一点...
- 数学不好能学信奥吗?要提前学数学吗?一文读懂数学与信奥的关系
-
在信奥的征途上,数学不仅是基石,更是推动孩子们不断前行的动力。然而,对于小学和初中阶段的孩子来说,如何提前布局数学学习,如何平衡数学与信奥的关系,成为了家长们普遍关心的问题。本文将从实际出发,探讨信奥...
- 手术机器人如何既准确又正确(手术机器人如何既准确又正确使用)
-
日前,北京积水潭医院矫形骨科周一新教授团队提出“臼杯矢状面角度的数学转换公式”,填补了该领域的国际空白,据此揭示了统治髋关节重建领域近50年的传统安全区(Lewinek安全区)不能成立之数学原理,并进...
- 天塌了,原来无痕内衣都是用胶粘出来的?
-
无痕内衣和轻薄贴身的夏装十分相配,它平整柔软的边缘也减少了摩擦皮肤的不适感。然而,这种内衣也有一些令人担忧的传闻。有人说无痕内衣都是用“胶水”粘出来的,有人说它稍有不慎就会开胶,甚至走在路上还会突然解...
- COD 消解器回流仪操作指南:3 步完成精准水质检测
-
化学需氧量(COD)检测是水质监测的核心指标,COD消解器回流仪作为实验室标配设备,其操作规范直接决定数据准确性。本文基于2025年最新版《水质检测设备操作规范》(HJ/T399-2025),...
- 产品图多导致页面变慢?WordPress外贸站“懒加载+CDN”一做
-
“哥,我这首页产品图很多,有灯具细节、展示效果、客户实拍,一开页面就卡得一批…”一个做户外照明出口的客户说。我看了下他站,确实首页就塞了20多张大图,图又是PNG原图,有的1MB一张我用C...
- JWT:一文搞懂现代身份验证的 "数字护照"
-
什么是JWT?先看一个生活比喻假设你去图书馆借书,管理员给你一张带盖章的借阅证,上面写着你的姓名、借阅权限和有效期。每次借书时,你只需出示这张证,管理员核对盖章无误就会放行——这就是JWT(...
- JS setTimeout 的另类用法(js中settime)
-
众所周知,JS中不消耗时间的函数一般是同步的,消耗时间的函数(比如:文件操作,网络操作)一般是异步的。JS默认是单线程的,异步是通过EventLoop来完成任务列表里的任务。当你的主线程中同步过程太...
- JVM GC诡异问题排查,k8s差点害死我……
-
前言本文将通过一个真实的生产环境案例,详细展示如何系统性地排查和解决JVM垃圾收集问题。这个案例涵盖了从问题发现、分析诊断到最终解决的完整过程,对于理解JVM调优实战具有重要的参考价值。系统背景我们的...
- 任天堂Switch 2首发评测:DLSS 3.5画质提升与Joy-Con摇杆耐久测试
-
任天堂Switch2终于来了!作为老玩家,我第一时间拿到机器,重点测试了两个大家最关心的点:DLSS3.5技术对画面的增强效果,以及Joy-Con摇杆的耐用性改进。下面用大白话聊聊实测体验,帮你避...
- 一周热门
- 最近发表
- 标签列表
-
- HTML 简介 (30)
- HTML 响应式设计 (31)
- HTML URL 编码 (32)
- HTML Web 服务器 (31)
- HTML 表单属性 (32)
- HTML 音频 (31)
- HTML5 支持 (33)
- HTML API (36)
- HTML 总结 (32)
- HTML 全局属性 (32)
- HTML 事件 (31)
- HTML 画布 (32)
- HTTP 方法 (30)
- 键盘快捷键 (30)
- CSS 语法 (35)
- CSS 选择器 (30)
- CSS 轮廓宽度 (31)
- CSS 谷歌字体 (33)
- CSS 链接 (31)
- CSS 定位 (31)
- CSS 图片库 (32)
- CSS 图像精灵 (31)
- SVG 文本 (32)
- 时钟启动 (33)
- HTML 游戏 (34)