Pytest中的fixture装饰器详解

pytest是Python生态中最流行的自动化测试框架,它通过简洁的语法、强大的功能(如fixture、参数化、插件扩展等)和丰富的插件生态,帮助开发者高效完成单元测试、集成测试和端到端测试。fixture是pytest框架中最核心、最强大的功能之一,它提供了一种优雅的方式来管理测试依赖项(如数据库连接、测试数据、API客户端等),实现了测试代码的模块化和复用。

fixture基础概念

  1. fixture的作用
  • 依赖注入:自动将资源(如数据库连接)注入到测试函数中
  • 资源管理:统一管理测试资源的创建和清理
  • 复用性:多个测试用例可以共享同一个fixture

2.基本语法

csharp 复制代码
import pytest
@pytest.fixture
def fixture_name():
    # 初始化代码
    resource = create_resource()
    yield resource  # 返回资源给测试用例使用
    # 清理代码(测试完成后执行)
    resource.cleanup()

2.fixture核心特性

  1. 作用域控制
  • 通过scope参数控制fixture的生命周期:

  • function(默认):每个测试函数执行一次

  • class:每个测试类执行一次

  • module:每个Python模块执行一次

  • package:每个包执行一次

  • session:整个测试会话执行一次

csharp 复制代码
@pytest.fixture(scope="module")
def shared_resource():
    return create_expensive_resource()
  1. 自动使用

通过autouse=True让fixture自动执行,无需显式调用

csharp 复制代码
@pytest.fixture
def db_connection():
    return connect_to_database()
@pytest.fixture
def user_repo(db_connection):
    return UserRepository(db_connection)
  1. 依赖其他fixture

fixture可以依赖其他fixture

csharp 复制代码
@pytest.fixture
def db_connection():
    return connect_to_database()
@pytest.fixture
def user_repo(db_connection):
    return UserRepository(db_connection)

3.fixture使用案例

1)API客户端初始化

csharp 复制代码
import pytest
import requests
@pytest.fixture(scope="module")
def api_client():
    client = requests.Session()
    client.base_url = "https://api.example.com"
    client.headers = {"Authorization": "Bearer test_token"}
    yield client  # 整个测试模块共享同一个客户端
    client.close()  # 模块测试完成后关闭
def test_get_user(api_client):
    response = api_client.get("/users/1")
    assert response.status_code == 200
    assert response.json()["name"] == "Alice"

2)Web浏览器实例管理(结合Selenium)

csharp 复制代码
import pytest
from selenium import webdriver
@pytest.fixture(scope="class")
def browser():
    driver = webdriver.Chrome()
    driver.implicitly_wait(10)
    yield driver  # 整个测试类共享同一个浏览器实例
    driver.quit()  # 类测试完成后关闭浏览器
class TestWebLogin:
    def test_login(self, browser):
        browser.get("https://example.com/login")
        browser.find_element("#username").send_keys("admin")
        browser.find_element("#password").send_keys("secret")
        browser.find_element("#submit").click()
        assert "Dashboard" in browser.title

4.fixture高级技巧

1)动态fixture参数化

csharp 复制代码
@pytest.fixture(params=["sqlite", "mysql", "postgres"])
def database(request):
    return create_db_connection(request.param)
def test_db_operations(database):
    assert database.execute("SELECT 1") == 1

2)多个fixture组合

csharp 复制代码
@pytest.fixture
def auth_token():
    return "test_token_123"
@pytest.fixture
def authenticated_client(api_client, auth_token):
    api_client.headers["Authorization"] = f"Bearer {auth_token}"
    return api_client
def test_protected_endpoint(authenticated_client):
    response = authenticated_client.get("/protected")
    assert response.status_code == 200
相关推荐
程序员小远9 小时前
Pytest+Selenium UI自动化测试实战实例
自动化测试·软件测试·python·selenium·测试工具·ui·pytest
惜.己2 天前
pytest简单使用和生成测试报告
开发语言·python·pytest
better-tomorrow2 天前
pytest-log
pytest
别在内卷了5 天前
测试学习之——Pytest Day4
python·学习·pytest
别在内卷了6 天前
测试学习之——Pytest Day3
python·学习·pytest
土小帽软件测试7 天前
mac系统安装、启动Jenkins,创建pytest接口自动化任务
macos·jenkins·pytest
Chasing__Dreams7 天前
pytest--1--pytest-mock常用的方法
pytest
chao_7898 天前
更灵活方便的初始化、清除方法——fixture【pytest】
服务器·自动化测试·python·pytest
别在内卷了9 天前
测试学习之——Pytest Day2
服务器·学习·pytest
chao_78910 天前
pytest 实战演练【pytest】
自动化测试·单元测试·自动化·pytest