Pytest中的fixtures怎么用

在 Pytest 中，fixtures 是用于设置和清理测试环境的函数。它们可以帮助你管理测试数据、配置文件、数据库连接等资源。Fixtures 可以在测试函数中作为参数使用，Pytest 会自动调用它们并传递给测试函数。

以下是使用 fixtures 的基本步骤：

1. 导入 pytest 模块。

import pytest

2. 定义一个 fixture 函数。使用 @pytest.fixture 装饰器来标记这个函数是一个 fixture。在这个函数中，你可以执行一些设置操作，比如创建临时文件、初始化数据库连接等。返回值将被传递给测试函数。

@pytest.fixture
def sample_data():
    data = "This is some sample data"
    return data

3. 在测试函数中使用 fixture。将 fixture 函数名作为参数传递给测试函数。当测试函数被调用时，Pytest 会先调用 fixture 函数，然后将返回值传递给测试函数。

def test_sample_data(sample_data):
    assert "sample" in sample_data

4. 如果需要在多个测试函数中使用同一个 fixture，可以使用 scope 参数来控制 fixture 的作用范围。可选的作用范围包括：

   • function（默认）：每个测试函数都会调用一次 fixture 函数。
   • class：每个测试类中的所有测试函数共享一个 fixture 实例。
   • module：每个模块中的所有测试函数共享一个 fixture 实例。
   • session：整个测试会话期间只调用一次 fixture 函数。

@pytest.fixture(scope="module")
def db_connection():
    # 创建数据库连接
    connection = ...
    yield connection
    # 关闭数据库连接
    connection.close()

def test_query(db_connection):
    # 使用 db_connection 执行查询操作
    ...

5. 如果需要在测试函数中自定义 fixture 的返回值，可以使用 yield 语句。在 yield 之前执行的代码会在测试函数执行前运行，yield 之后的代码会在测试函数执行后运行。

@pytest.fixture
def sample_data():
    data = "This is some sample data"
    # 在测试函数执行前运行的代码
    ...
    yield data
    # 在测试函数执行后运行的代码
    ...

这就是 Pytest 中 fixtures 的基本用法。你可以根据实际需求灵活地使用它们来管理测试环境。