本文详解如何在 FastAPI 集成测试中，通过 dependency_overrides 将 get_db 依赖动态替换为支持事务回滚的测试数据库会话，避免数据污染，确保测试隔离性与可重复性。
本文详解如何在 fastapi 集成测试中，通过 `dependency_overrides` 将 `get_db` 依赖动态替换为支持事务回滚的测试数据库会话，避免数据污染，确保测试隔离性与可重复性。
在 FastAPI 的端到端测试中，常需让 TestClient 使用与单元测试一致的、具备事务回滚能力的数据库会话（如基于 SQLAlchemy 的 scoped_session），而非生产环境的真实连接。但直接将 db_session fixture 赋值给 app.dependency_overrides[get_db] 会导致 422 Unprocessable Entity 错误——根本原因在于：get_db 是一个依赖函数（通常含 yield 的生成器），而 db_session 是 pytest fixture，二者调用时机与生命周期不匹配。
正确的做法是：用一个包装函数作为替代依赖，该函数内部调用并 yield fixture 提供的会话对象。同时需注意会话清理与依赖重置，防止跨测试污染。
✅ 正确实现步骤
1. 优化 db_session fixture：显式调用 session.remove()
@pytest.fixture(scope="session")
def db_session(connection):
transaction = connection.begin()
session_factory = sessionmaker(autocommit=False, autoflush=False, bind=connection)
session = scoped_session(session_factory)
try:
yield session
finally:
session.remove() # 关键：释放 scoped_session 线程局部存储
transaction.rollback()⚠️ session.remove() 不可省略。否则 scoped_session 可能在后续测试中复用旧会话，导致状态残留或 DetachedInstanceError。
2. 在 client fixture 中安全覆盖 get_db
@pytest.fixture(scope="module")
def client(db_session) -> Generator[TestClient, None, None]:
def override_get_db():
yield db_session # 直接 yield 已初始化的会话实例
app.dependency_overrides[get_db] = override_get_db
with TestClient(app) as c:
yield c
# 清理：恢复原始依赖，避免影响其他测试模块
app.dependency_overrides.clear()✅ 此处 override_get_db 是标准依赖函数签名（无参数、返回 Generator），与 get_db 原逻辑完全兼容；db_session 作为 fixture 参数被 pytest 自动注入，确保其已正确初始化。
3. 验证效果（示例测试）
def test_create_user(client: TestClient):
response = client.post("/users/", json={"name": "Alice", "email": "alice@test.com"})
assert response.status_code == 200
data = response.json()
assert data["name"] == "Alice"
# 数据库中应无持久化记录（事务已回滚）
# 后续测试无需手动清理? 关键注意事项
- 作用域匹配：db_session 使用 scope="session"（适合 DB 连接复用），而 client 使用 scope="module"，需确保 db_session 在 client 生命周期内始终有效；
- 依赖清除必须执行：app.dependency_overrides.clear() 应放在 yield 之后的 finally 或 with 结束位置，否则可能污染其他测试模块；
- 避免循环依赖：不要在 override_get_db 中再次请求 db_session fixture（pytest 不允许在依赖函数内调用 fixture）；
- 异步依赖？ 若 get_db 是 async def，则覆盖函数也需为 async def override_get_db() 并使用 async with 和 await，但本例基于同步 SQLAlchemy，故保持同步即可。
通过上述结构，你既能复用已有的、带事务回滚能力的数据库 fixture，又能确保 TestClient 完全走测试路径——从 API 层到 DB 层全程隔离，真正实现可靠、可重复的集成测试。
