直接mock sqlalchemy模型易失败,因其非可调用对象,真正需mock的是session实例及其query链式行为,须让mock支持.filter()等中间调用并仅在.all()等终端方法返回数据。

为什么直接用 pytest-mock Mock SQLAlchemy 模型容易失败
因为 SQLAlchemy 模型本身不是“可调用对象”,而是类;而 session.query(Model) 返回的是 Query 对象,它支持链式调用(如 .filter()、.all()),但这些方法在真实 session 中才被动态绑定。直接 patch 类或实例方法,常会遇到 AttributeError: 'Mock' object has no attribute 'filter' 或返回空列表却无法验证调用链。
真正要 Mock 的不是模型类本身,而是 session 实例及其返回的 Query 行为 —— 尤其是 .all()、.first()、.scalar() 这些终端方法。
- 别 patch
User类,它只是声明,不参与运行时查询逻辑 - 别 patch
session.query的返回值为普通 list,否则.filter().order_by().limit(1)会报错 - 必须让 mock query 支持链式调用,并在终端方法上返回可控数据
如何构造一个可链式调用的 Query Mock
核心思路:用 mocker.MagicMock 构建一个“假 Query”,让它对任意方法调用都返回自身(实现链式),只在终端方法(如 .all())上返回预设数据。
def mock_query(mocker, return_value=None):
query_mock = mocker.MagicMock()
# 所有方法调用都返回自身,支持链式
query_mock.__getattr__.return_value = query_mock
# 终端方法返回指定值
query_mock.all.return_value = return_value or []
query_mock.first.return_value = (return_value[0] if return_value else None)
query_mock.one_or_none.return_value = (return_value[0] if return_value else None)
query_mock.count.return_value = len(return_value) if return_value else 0
return query_mock
使用时,在测试中 patch session.query 并让它返回这个 mock:
mocker.patch('myapp.db.session.query', return_value=mock_query(mocker, [user1, user2]))- 注意路径要和实际 import 路径一致(比如是
from myapp.models import User,但 patch 的是session.query所在模块) - 如果函数内用了
db.session,需 patchmyapp.views.db.session,而非sqlalchemy.orm.session.Session
测试带 filter 条件的查询时,如何验证参数是否正确
仅 Mock 返回值不够,还需确认业务代码是否传入了预期的 filter 表达式。SQLAlchemy 的 filter 条件是 BinaryExpression 对象,不能直接用 == 比较,但可以用 call_args 提取原始参数并检查其结构。
- 给
query_mock.filter设置side_effect,捕获调用参数:query_mock.filter.side_effect = lambda *args: setattr(query_mock, '_last_filter', args) or query_mock - 执行业务函数后,检查
query_mock._last_filter是否包含User.name == 'alice'这类表达式(可通过str(expr)或expr.left.name等属性判断) - 更稳妥的方式:在被测函数里把 filter 条件提取为变量,测试时直接 assert 那个变量,而非依赖 mock 调用痕迹
Mock 失败的典型信号与绕过方案
出现 sqlalchemy.exc.NoResultFound 或 AttributeError: 'NoneType' object has no attribute 'id',往往不是数据库没连上,而是 mock 没覆盖到某个分支(比如 .one() 未定义返回值,或 session.add() 后没模拟 session.flush() 导致 ID 未生成)。
-
.one()和.one_or_none()行为不同:前者查不到或查多会抛异常,后者只返回None;mock 时要按实际调用选对应 return_value / side_effect -
session.add(obj)不改变 obj.id,除非后续flush()或commit();若测试依赖自增 ID,得手动设置obj.id = 1,或 mocksession.flush并赋值 - 使用
scoped_session时,patch 目标应是myapp.db.session(即你调用的地方),而不是sqlalchemy.orm.scoping.scoped_session
最易被忽略的一点:SQLAlchemy 的 lazy loading(如 relationship)会在访问属性时触发新查询 —— 如果没 mock 对应的关联表查询,测试会意外连接真实数据库。要么显式关闭 lazy loading(lazy='joined'),要么为每个可能触发的关联字段单独 mock 查询链。
Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!











