pytest + pytest-mock + pytest-parametrize为基础构建测试框架
一、框架核心组件整合
pytest作为基础测试框架，与pytest-mock（unittest.mock封装）和pytest-parametrize（参数化测试）形成黄金三角。三者协同可实现：
- 依赖隔离：通过mocker fixture模拟外部API/数据库交互
- 场景覆盖：利用参数化实现多输入组合测试
- 代码复用：通过fixture机制实现测试资源共享
二、项目结构设计（推荐方案）
my_project/ ├── src/ # 业务代码 │ ├── api/ # API模块 │ └── utils/ # 工具函数 ├── tests/ # 测试目录 │ ├── unit/ # 单元测试 │ │ ├── test_api.py │ │ └── test_utils.py │ ├── integration/ # 集成测试 │ ├── conftest.py # 全局fixture定义 │ └── pytest.ini # 配置文件 └── requirements.txt # 依赖管理三、核心功能实现示例
1. 参数化测试（pytest-parametrize）
import pytest # 基础参数化 @pytest.mark.parametrize("a,b,expected", [ (2, 3, 5), # 正常场景 (0, 0, 0), # 边界值 (-1, 1, 0), # 异常输入 ]) def test_add(a, b, expected): assert a + b == expected # 高级用法：结合fixture实现动态参数 @pytest.fixture(params=[ {"input": "valid", "expected": True}, {"input": "invalid", "expected": False} ]) def dynamic_data(request): return request.param def test_validation(dynamic_data): assert validate(dynamic_data["input"]) == dynamic_data["expected"]2. Mock服务集成（pytest-mock
def test_api_call(mocker): # 模拟requests.get方法 mock_get = mocker.patch("requests.get") mock_get.return_value.json.return_value = {"status": "ok"} # 执行测试 result = fetch_data("https://api.example.com") # 验证调用 mock_get.assert_called_once_with("https://api.example.com") assert result["status"] == "ok"3. 复杂场景测试（三者结合）
@pytest.mark.parametrize("scenario", [ {"url": "/v1/users", "expected_code": 200}, {"url": "/v2/users", "expected_code": 404}, ]) def test_api_versioning(scenario, mocker): # 模拟API响应 mock_response = mocker.Mock() mock_response.status_code = scenario["expected_code"] mocker.patch("requests.get", return_value=mock_response) # 执行测试 response = call_api(scenario["url"]) # 断言结果 assert response.status_code == scenario["expected_code"]四、最佳实践建议
fixture管理：
- 使用
conftest.py定义跨模块fixture - 合理设置作用域（function/module/session）
- 实现fixture依赖链（如db_conn → api_client）
- 使用
测试分类：
- 单元测试：使用mock隔离外部依赖
- 集成测试：通过
--integration标记单独执行 - 性能测试：添加
@pytest.mark.slow标记
报告与集成：
- 生成HTML报告：
pytest --html=report.html - CI集成：配置pytest.ini实现零配置执行
- 覆盖率分析：结合
pytest-cov生成覆盖率报告
- 生成HTML报告：
五、常用命令参考
# 运行所有测试 pytest # 运行指定测试 pytest tests/unit/test_api.py::test_create_user # 仅运行标记测试 pytest -m "smoke and not slow" # 生成详细报告 pytest --cov=src --cov-report=html这套框架设计遵循了测试金字塔原则，既保证了单元测试的执行效率，又通过集成测试验证了模块间协作。通过参数化和mock的结合使用，可在最少代码量下实现最大测试覆盖率。