本文详解如何在 Pytest 中避免真实数据库连接，通过 FastAPI 的 dependency_overrides 机制安全替换 get_db 依赖，并配合 unittest.mock.patch 隔离业务逻辑层，彻底解决 Connection refused 错误。
本文详解如何在 pytest 中避免真实数据库连接，通过 fastapi 的 `dependency_overrides` 机制安全替换 `get_db` 依赖，并配合 `unittest.mock.patch` 隔离业务逻辑层，彻底解决 `connection refused` 错误。
在使用 Pytest 对 FastAPI 路由进行单元测试时，若路由函数依赖 SQLAlchemy 的数据库会话（如通过 Depends(get_db) 注入），而测试未切断与真实数据库的连接，就会触发类似以下错误：
sqlalchemy.exc.OperationalError: (psycopg2.OperationalError) connection to server at "localhost", port 5439 failed: Connection refused
根本原因在于：测试运行时仍尝试连接 PostgreSQL（或其他后端），但测试环境并未启动该服务，也无需真实 I/O。
✅ 正确解法不是 mock Session 实例本身（如传入 MagicMock() 给依赖函数），而是 覆盖 FastAPI 的依赖注入链 —— 即用一个返回伪造会话（如 mock.Mock()）的函数替代 get_db，确保 create_user 路由在调用 db = Depends(get_db) 时拿到的是可控的 mock 对象，而非试图连接数据库的真实工厂。
以下是推荐的、生产就绪的测试写法：
import pytest
from unittest import mock
from fastapi.testclient import TestClient
from app.main import app # 确保导入你的 FastAPI 实例
from app.routes.users import get_db, create_user_method
client = TestClient(app)
@mock.patch("app.routes.users.create_user_method")
def test_create_user_success(create_user_mock):
"""
测试用户注册成功场景：mock 业务方法 + override get_db 依赖
"""
# Step 1: 定义 fake get_db，返回一个轻量级 Mock 会话
def fake_get_db():
return mock.Mock()
# Step 2: 覆盖全局依赖（关键！）
app.dependency_overrides[get_db] = fake_get_db
# Step 3: 配置被 mock 的业务方法返回值
create_user_mock.return_value = (True, {}, "User Registered successfully.", 200)
# Step 4: 构造有效请求体（注意：原始示例中 payload 结构需匹配 User 模型）
# ⚠️ 注意：你定义的路由接收的是 User Pydantic 模型，因此 JSON 应直接为 user 字段，无需外层 "data"/"user"
payload = {
"first_name": "Shaheer",
"last_name": "Muhammad",
"email": "shaheer@example.com",
"mobile": "1234567890",
"password": "Shaheer@88888"
}
response = client.post("/api/v1/user/register", json=payload)
# Step 5: 断言响应
assert response.status_code == 200
data = response.json()
assert data["status"] is True
assert data["message"] == "User Registered successfully."
assert "data" in data
# ✅ 清理：务必恢复原始依赖，防止污染其他测试
app.dependency_overrides.clear()
? 关键注意事项：
- app.dependency_overrides[get_db] = ... 是核心：它拦截了 FastAPI 在请求生命周期中对 get_db 的调用，使其永远不执行真实数据库连接逻辑。
- 不要仅 patch create_user_method 就认为足够：即使你 mock 了业务方法，若 get_db 仍被调用（例如在 create_user_method 内部或其子调用中），依然会触发连接；因此必须显式 override 依赖。
- 务必调用 app.dependency_overrides.clear() 或重置特定依赖：否则后续测试可能意外复用该 mock，导致断言失败或掩盖真实问题。更健壮的做法是使用 pytest fixture 自动管理：
@pytest.fixture(autouse=True)
def clear_dependency_overrides():
yield
app.dependency_overrides.clear()
- Payload 结构需与 Pydantic 模型严格一致：你的 User 模型期望扁平字段（如 first_name, email），而非嵌套在 "data": {"user": {...}} 中——否则 FastAPI 解析失败，返回 422 错误，干扰主逻辑验证。
? 进阶建议：对于复杂数据库交互（如需要验证 ORM 查询行为），可进一步使用 SQLModel + sqlite:///:memory: 或 pytest-asyncio + aiosqlite 搭配 pytest-lazy-fixture 构建轻量集成测试，但纯单元测试阶段，依赖覆盖 + 方法 mock 已完全满足隔离性与速度要求。
通过以上方式，你将获得快速、稳定、可重复的路由层单元测试，彻底告别 Connection refused，聚焦于业务逻辑验证。
