本文详解如何使用 pytest 和 pytest-mock 对 Flask 路由中的数据库连接（db_connection）和查询函数（db_query）进行精准打桩，实现隔离依赖、可控返回值的单元测试。
本文详解如何使用 pytest 和 pytest-mock 对 flask 路由中的数据库连接（`db_connection`）和查询函数（`db_query`）进行精准打桩，实现隔离依赖、可控返回值的单元测试。
在 Flask 应用中对涉及外部依赖（如数据库）的路由进行单元测试时，关键原则是解耦真实 I/O，聚焦逻辑验证。直接调用真实数据库不仅慢、不稳定，还会使测试失去可重复性和可移植性。正确的做法是使用 mocker.patch() 对被测函数内部调用的数据库工具函数进行打桩（mock），并精确控制其返回值。
✅ 正确打桩路径：必须匹配导入位置
一个常见误区是按“定义位置”打桩（如 app.utils.db.db_connection），而实际应按被测代码中导入并使用的路径打桩。观察你的路由代码：
from app.utils.db import db_connection, db_query # ← 这里是关键！
因此，在测试中必须 patch "app.utils.db.db_connection" 和 "app.utils.db.db_query"（注意模块名是 db，而非 database —— 请严格对照你项目中真实的模块路径）。若路径错误，mock 将失效，测试仍会触发真实数据库调用。
Skill Weave Chains — 技能链路由引擎
下载
开箱即用的技能链路由引擎。13 条预定义链覆盖搜索、开发、审查、MLOps、法律、创意等场景，三层路由架构（触发词→SAD反馈→DAG编排），recall@10=96.97%。配置驱动（chains.yaml），零代码扩展。pip install skill-weave-chains 一键安装。
✅ 完整可运行测试示例
以下是一个结构清晰、符合最佳实践的 pytest 测试用例：
import json
import pytest
from app import app # 确保导入主 Flask 应用实例
# 测试数据（模拟 DB 查询返回的原始数据）
SOME_ROUTE_JSON_DATA = '{"key": "value", "id": 123}'
@pytest.fixture
def client():
"""提供 Flask 测试客户端，自动管理应用上下文"""
with app.test_client() as client:
yield client
def test_get_data_for_some_route(mocker, client):
# ✅ 正确路径打桩：匹配 routes.py 中的 import 路径
mock_conn = mocker.patch("app.utils.db.db_connection")
mock_query = mocker.patch("app.utils.db.db_query")
# ✅ 设定 db_query 的返回值（即模拟数据库查询结果）
# 注意：此处返回的是解析后的 Python 对象（list/dict），非 JSON 字符串
mock_query.return_value = [{"key": "value", "id": 123}]
# ✅ 发起 HTTP 请求（含认证头）
response = client.get(
"/some/route",
headers={"Authorization": f"Bearer 12345"} # 注意 Bearer 前缀
)
# ✅ 断言响应状态与数据
assert response.status_code == 200
assert response.get_json() == {"key": "value", "id": 123} # 即 data[0] 的结果
# ✅ 验证函数调用行为（可选但强烈推荐）
mock_conn.assert_called_once()
mock_query.assert_called_once_with(mock_conn.return_value, "GET_STUFF")⚠️ 关键注意事项
- return_value 类型需匹配实际逻辑：你的路由中执行了 data[0]，因此 db_query 的 mock 返回值应为 list（如 [{"key": "value"}]），而非单个 dict；否则会触发 IndexError。
- 避免 loads() 误用：原问题中 loads(SOME_ROUTE_JSON_DATA) 是冗余操作 —— db_query 应返回已解析的 Python 对象，而非 JSON 字符串；mock 直接返回 dict 或 list 更自然、更贴近真实行为。
-
认证头格式：Flask 路由使用 @auth 装饰器时，通常期望 Authorization: Bearer
格式，请确保测试中 header 一致。 - 模块路径一致性：务必检查 app/utils/db.py 文件名与 from app.utils.db import ... 中的 db 是否完全一致（例如不是 database.py 或 db_utils.py）。
通过以上方式，你就能完全隔离数据库依赖，在毫秒级内完成对业务逻辑（如数据解析、HTTP 响应构造）的可靠验证，大幅提升测试稳定性与开发效率。
