如何优雅地在多项目间复用pytest fixture库组件？

我太懂这种重复复制代码的痛苦了——把跨项目的数据库测试配置做成可安装库是完全可行的，而且有几种比你临时方案更优雅的实现方式：

方案1：利用pytest的pytest_plugins机制（最推荐） #

这是pytest官方推荐的复用fixture的方式，能让你的库fixture像内置fixture一样直接被项目使用，完全不用复制代码。

步骤： #

1. 在你的my_db_fixture库中，创建根目录的conftest.py，把所有核心fixture定义在这里：

# my_db_fixture/conftest.py
import pytest
from sqlalchemy import create_engine, Session

@pytest.fixture(scope="session")
def db_engine():
    # 这里写你的标准数据库配置逻辑
    engine = create_engine("postgresql://default_user:default_pass@localhost/test_db")
    yield engine
    # 清理逻辑
    engine.dispose()

@pytest.fixture(scope="function")
def db_session(db_engine):
    # 基于db_engine构建的会话fixture
    connection = db_engine.connect()
    transaction = connection.begin()
    session = Session(bind=connection)
    yield session
    # 测试后回滚清理
    session.rollback()
    transaction.rollback()
    connection.close()

2. 在你的项目根目录的conftest.py里，只需要添加一行：

# 项目根目录/conftest.py
pytest_plugins = ["my_db_fixture.conftest"]

之后项目里的所有测试用例，就可以直接使用db_engine、db_session这些fixture了，完全看不到库的内部实现，也不用复制任何代码。

方案2：导出内部fixture函数，按需封装 #

如果你的项目需要对库fixture做少量适配（比如调整作用域、添加项目特有的前置操作），可以在库中用下划线前缀隐藏内部函数，然后在项目中轻量封装：

步骤： #

1. 在库中创建my_db_fixture/fixtures.py，导出内部实现：

# my_db_fixture/fixtures.py
import pytest
from sqlalchemy import create_engine, Session

def _db_engine():
    # 核心数据库引擎逻辑
    engine = create_engine("postgresql://default_user:default_pass@localhost/test_db")
    yield engine
    engine.dispose()

def _db_session(engine):
    # 核心会话逻辑
    connection = engine.connect()
    transaction = connection.begin()
    session = Session(bind=connection)
    yield session
    session.rollback()
    transaction.rollback()
    connection.close()

2. 在项目的conftest.py中调用这些内部函数，封装成项目可用的fixture：

# 项目根目录/conftest.py
import pytest
from my_db_fixture.fixtures import _db_engine, _db_session

@pytest.fixture(scope="module")  # 可以根据项目需求修改作用域
def db_engine():
    yield from _db_engine()

@pytest.fixture(scope="function")
def db_session(db_engine):
    # 这里可以添加项目特有的适配逻辑
    yield from _db_session(db_engine)

这种方式既复用了核心逻辑，又保留了项目定制的灵活性，同时下划线前缀的内部函数不会被意外暴露。

方案3：通过环境变量/配置做动态适配 #

如果不同项目需要不同的数据库配置（比如不同的连接地址），可以在库fixture中通过环境变量读取配置，避免硬编码：

# my_db_fixture/conftest.py
import os
import pytest
from sqlalchemy import create_engine

@pytest.fixture(scope="session")
def db_url():
    # 优先读取项目的环境变量，没有则用默认值
    return os.getenv("TEST_DB_URL", "postgresql://default_user:default_pass@localhost/test_db")

@pytest.fixture(scope="session")
def db_engine(db_url):
    engine = create_engine(db_url)
    yield engine
    engine.dispose()

项目只需要在运行测试前设置TEST_DB_URL环境变量，就能自定义数据库配置，完全不用修改fixture代码。

内容的提问来源于stack exchange，提问作者Michael Grazebrook