使用MagicMock模拟Session时SQLAlchemy的.filter方法失效问题

可能的原因及解决方案 #

1. 模拟会话未实现filter的过滤逻辑 #

如果你的mock_db_session是简单的MagicMock，默认不会处理链式调用的过滤逻辑。当调用db.query(User).filter(...)时，filter不会实际处理条件，直接返回原query对象，导致all()返回所有预存用户。

解决办法：手动给mock的query对象实现filter和all的逻辑，根据传入的条件返回对应数据：

@pytest.fixture
def mock_db_session(mocker):
    mock_session = mocker.Mock()
    mock_query = mocker.Mock()

    def mock_filter(condition):
        # 转字符串匹配SQLAlchemy条件表达式
        if str(condition) == str(User.deleted == False):
            mock_query.all.return_value = [user1]
        else:
            mock_query.all.return_value = []
        return mock_query

    mock_query.filter = mock_filter
    mock_session.query.return_value = mock_query
    return mock_session

2. SQLAlchemy条件表达式的匹配问题 #

User.deleted == False是SQLAlchemy生成的SQL表达式对象，不是Python原生布尔值，mock的filter可能无法正确识别这个条件。

解决办法：改用SQLAlchemy推荐的布尔比较方式User.deleted.is_(False)，再调整mock的匹配逻辑：

# 业务代码中修改过滤条件
db.query(User).filter(User.deleted.is_(False)).all()

# 测试fixture中同步调整匹配逻辑
def mock_filter(condition):
    if str(condition) == str(User.deleted.is_(False)):
        mock_query.all.return_value = [user1]
    # ... 其他逻辑

3. 依赖替换未正确生效 #

确认app.dependency_overrides[dependencies.get_db]是否在测试前正确绑定了mock_db_session，或者是否有其他代码覆盖了这个设置。

解决办法：在fixture中明确设置并在测试后清理：

@pytest.fixture
def client(app, mock_db_session):
    app.dependency_overrides[dependencies.get_db] = lambda: mock_db_session
    with TestClient(app) as client:
        yield client
    # 测试结束后清空依赖覆盖，避免影响其他用例
    app.dependency_overrides.clear()

内容的提问来源于stack exchange，提问作者Anthony Raterta