pytest+request 接口自动化测试

<think>我们正在构建一个使用pytest和requests的接口自动化测试项目。根据用户的问题，我们需要详细解释如何实现这样的项目。参考引用内容，我们可以总结出以下几个关键点：1.使用requests库发送HTTP请求。2.使用pytest作为测试框架来管理测试用例。3.可能使用YAML文件进行数据驱动（引用[3]）。4.使用Allure生成测试报告（引用[2],[3]）。5.测试用例的组织和标记（引用[1]）。因此，我们将按照以下结构进行详解：一、项目结构二、环境准备三、请求的封装四、测试用例编写（使用pytest）五、数据驱动（结合YAML）六、测试报告（Allure）七、运行测试注意：引用内容中有些代码片段，我们可以参考并适当调整。开始详细说明：</think>### 使用 Pytest + Requests 构建接口自动化测试项目详解 #### 一、项目结构设计 ```bash project/ ├── conftest.py # pytest全局配置 ├── requirements.txt # 依赖库 ├── testcases/ # 测试用例目录 │ ├── __init__.py │ ├── test_login.py # 示例测试模块 │ └── test_user.py ├── utils/ # 工具类 │ ├── __init__.py │ ├── request_util.py # 请求封装 │ └── assert_util.py # 断言封装 └── data/ # 测试数据 ├── login_data.yaml └── user_data.yaml ``` #### 二、核心组件实现 1. **请求封装 (`utils/request_util.py`)** ```python import requests class RequestUtil: def __init__(self): self.session = requests.Session() def send_request(self, method, url, **kwargs): method = method.upper() try: response = self.session.request(method, url, **kwargs) response.raise_for_status() # 自动抛出HTTP错误 return response except requests.exceptions.RequestException as e: raise AssertionError(f"请求失败: {str(e)}") ``` 2. **断言工具 (`utils/assert_util.py`)** ```python def assert_response(response, expected_status=200, expected_data=None): # 状态码断言 assert response.status_code == expected_status, \ f"状态码错误: 期望{expected_status}, 实际{response.status_code}" # 响应体断言 if expected_data: actual_data = response.json() for key, value in expected_data.items(): assert key in actual_data, f"缺失字段: {key}" assert actual_data[key] == value, \ f"字段值不匹配: {key} 期望={value}, 实际={actual_data[key]}" ``` #### 三、测试用例编写 (`testcases/test_login.py`) ```python import pytest from utils.request_util import RequestUtil from utils.assert_util import assert_response @pytest.mark.login class TestLogin: @pytest.fixture(scope="class") def req(self): return RequestUtil() @pytest.mark.parametrize("case", [ {"username": "admin", "password": "123456", "expected": 200}, {"username": "invalid", "password": "xxx", "expected": 401} ]) def test_login(self, req, case): """登录接口测试""" url = "https://api.example.com/login" payload = { "username": case["username"], "password": case["password"] } response = req.send_request("POST", url, json=payload) assert_response(response, expected_status=case["expected"]) ``` #### 四、Pytest 配置 (`conftest.py`) ```python import pytest # 全局前置处理 @pytest.fixture(scope="session", autouse=True) def setup_session(): print("\n=== 测试开始 ===") yield print("\n=== 测试结束 ===") # 失败重试机制 def pytest_addoption(parser): parser.addoption("--reruns", action="store", default=1, help="失败重试次数") @pytest.hookimpl(tryfirst=True) def pytest_configure(config): config.addinivalue_line("markers", "login: 登录模块测试") ``` #### 五、运行与报告生成 1. **安装依赖 (`requirements.txt`)** ``` pytest==7.4.0 requests==2.31.0 pytest-rerunfailures==12.0 allure-pytest==2.13.2 ``` 2. **执行测试** ```bash # 运行全部测试 pytest -v # 运行标记为login的测试 pytest -m login # 失败重试3次 pytest --reruns 3 -v ``` 3. **生成Allure报告** ```bash pytest --alluredir=./report/allure_results allure serve ./report/allure_results # 本地查看报告 ``` #### 六、最佳实践 1. **数据驱动** ```python # 从YAML加载测试数据 import yaml @pytest.mark.parametrize("case", yaml.safe_load(open("data/login_data.yaml"))) def test_login_data_driven(req, case): ... ``` 2. **关联参数处理** ```python # 在conftest.py中存储全局token @pytest.fixture def auth_token(req): response = req.send_request("POST", login_url, json=credentials) return response.json()["token"] # 测试用例使用token def test_user_info(req, auth_token): headers = {"Authorization": f"Bearer {auth_token}"} ... ``` 3. **异步请求支持** ```python # 在request_util.py中添加 import aiohttp async def async_request(self, method, url, **kwargs): async with aiohttp.ClientSession() as session: async with session.request(method, url, **kwargs) as response: ... ```