测试阶段阻止SQLAlchemy提交至数据库的技术方案问询

方案1：自定义Session重写commit方法 #

直接继承SQLAlchemy的Session类，在测试环境中将commit方法替换为仅执行flush，这样业务代码里的session.commit()调用只会把变更刷到内存，不会真正提交到数据库，后续可以通过rollback()清空所有测试数据。

代码示例：

from sqlalchemy.orm import Session

class TestSession(Session):
    def commit(self):
        # 测试环境下只做flush，不提交
        self.flush()

初始化Session时根据环境切换：

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

engine = create_engine("你的Rails数据库连接字符串")
# 测试环境用自定义TestSession，生产环境用默认Session
SessionLocal = sessionmaker(
    class_=TestSession if is_test_environment else Session,
    bind=engine
)

方案2：利用事务嵌套实现自动回滚 #

SQLAlchemy支持事务嵌套，内层的commit实际上只是创建事务保存点，不会真正提交到数据库。你可以在测试开始时开启一个顶层事务，所有测试操作都在这个事务内执行，测试结束后统一回滚。

以pytest为例的 fixture 实现：

import pytest
from sqlalchemy.orm import Session

@pytest.fixture(scope="function")
def test_db_session():
    engine = create_engine("你的Rails数据库连接字符串")
    session = Session(bind=engine)
    # 开启顶层事务
    session.begin()
    try:
        yield session
    finally:
        # 测试结束后回滚所有变更
        session.rollback()
        session.close()

这种方式不需要修改业务代码中的commit调用，完全适配你不想改动生产逻辑的需求。

关于最佳实践的疑问 #

“仅在最后时刻调用commit”确实是SQLAlchemy的推荐实践——尽量将相关操作放在一个事务中，最后统一提交，减少数据库IO和事务冲突。但如果你的业务代码中已经存在多处commit（比如复用了生产环境的逻辑），上面的测试拦截方案会更实用，无需修改业务代码就能实现测试数据隔离。

另外，SQLAlchemy标准测试的“删表重建”确实不适合你的场景，因为Rails测试依赖现有表结构，而事务回滚的方式既能保证测试套件间的数据不泄漏，又能完整保留表结构，完美适配你的需求。

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