pytest 进阶：玩转组合测试和端到端测试，让你的代码更健壮
pytest 进阶：玩转组合测试和端到端测试，让你的代码更健壮
pytest 是 Python 中最流行的测试框架之一，以其简洁的语法、丰富的功能和强大的扩展性而闻名。除了基本的单元测试，pytest 还支持组合测试和端到端测试，帮助你构建更完善的测试体系，确保代码质量。
1. 组合测试：测试多个参数的组合
在实际开发中，你的代码经常需要处理多个参数的组合。例如，一个函数可能需要接收两个参数，每个参数都有多个取值。如何测试所有可能的参数组合，确保函数在各种情况下都能正常运行？
pytest 提供了 @pytest.mark.parametrize 装饰器来实现组合测试。例如，假设你有一个函数 calculate_area 用于计算矩形的面积，它接收两个参数 length 和 width。你可以使用 @pytest.mark.parametrize 测试不同长度和宽度组合下函数的返回值是否正确：
import pytest
def calculate_area(length, width):
return length * width
@pytest.mark.parametrize("length, width, expected", [
(1, 2, 2),
(3, 4, 12),
(5, 6, 30),
])
def test_calculate_area(length, width, expected):
assert calculate_area(length, width) == expected
在这个例子中，@pytest.mark.parametrize 装饰器接受三个参数：
"length, width, expected"：指定测试参数的名称。[ (1, 2, 2), (3, 4, 12), (5, 6, 30) ]：定义测试数据的列表，每个元组代表一组参数和预期结果。
运行测试后，pytest 会自动生成多个测试用例，分别对应不同的参数组合。
2. 端到端测试：模拟真实用户场景
端到端测试（End-to-End Testing）模拟用户在真实环境中的操作，验证整个系统从前端到后端的逻辑是否正确。例如，测试用户登录、创建订单、支付等流程，确保整个系统能够正常工作。
pytest 可以结合其他工具，例如 Selenium 和 requests，进行端到端测试。例如，你可以使用 Selenium 模拟浏览器打开网页，填写表单，提交操作，然后使用 requests 验证服务器端是否收到请求，并返回预期结果。
import pytest
from selenium import webdriver
from requests import get
@pytest.mark.e2e
def test_login(browser):
browser.get("https://www.example.com")
# 模拟用户输入用户名和密码
# ...
# 模拟点击登录按钮
# ...
# 验证登录成功
assert browser.title == "Welcome"
@pytest.mark.e2e
def test_create_order(browser):
# 模拟用户登录
# ...
# 模拟用户选择商品
# ...
# 模拟用户提交订单
# ...
# 验证订单创建成功
# ...
@pytest.fixture
def browser():
driver = webdriver.Chrome()
yield driver
driver.quit()
在这个例子中，@pytest.mark.e2e 装饰器表示这是一个端到端测试用例。browser 是一个 fixture，用于创建浏览器实例。每个测试用例都模拟了用户在网站上的操作，并验证最终的结果是否符合预期。
3. 小结
pytest 提供了组合测试和端到端测试功能，帮助你构建更强大的测试体系，提高代码质量。在实际开发中，你可以根据项目的具体情况选择合适的测试策略。
使用 pytest 进行组合测试和端到端测试的优点：
- 简化测试代码编写，提高测试效率
- 确保代码在各种情况下都能正常运行
- 提高代码质量，减少bug
- 增强代码可维护性
建议:
- 编写清晰、简洁的测试用例，方便阅读和维护
- 使用 fixture 管理测试资源，例如数据库连接、浏览器实例等
- 使用 pytest-xdist 并行执行测试，加快测试速度
- 将测试代码与业务代码分离，避免代码耦合