如何在Flask+pytest+SQLAlchemy测试中回滚数据库会话变更？

你的问题核心在于测试中如果主动调用了db.session.commit()，原来的fixture里的rollback无法撤销已经提交到数据库的变更——因为commit会把session的变更持久化到磁盘。下面给你几种可行的解决方案：

方案一：Mock Commit方法+嵌套事务 #

这个方案能拦截测试代码里的commit操作，用嵌套事务（保存点）替代真实提交，测试结束后一次性回滚所有变更：

import pytest
from your_app import app, db

@pytest.fixture(scope="function")
def db_session():
    with app.app_context():
        # 开启嵌套事务（创建保存点）
        db.session.begin_nested()

        # 替换commit方法，改成回滚到保存点再重新创建保存点
        original_commit = db.session.commit
        def mock_commit():
            db.session.rollback()
            db.session.begin_nested()
        
        db.session.commit = mock_commit

        yield db.session

        # 恢复原始commit方法
        db.session.commit = original_commit
        # 回滚最外层事务，清理所有变更
        db.session.rollback()

使用时，测试函数直接依赖db_session fixture即可，不需要修改原有业务代码里的commit逻辑。

方案二：使用临时数据库文件 #

每个测试用例创建独立的临时数据库，测试结束后自动清理，彻底隔离测试数据：

import pytest
from your_app import Flask, db

@pytest.fixture(scope="function")
def app(tmp_path):
    # 在临时目录创建新的测试数据库
    temp_db = tmp_path / "temp_test.db"
    test_app = Flask(__name__)
    test_app.config.update(
        SQLALCHEMY_DATABASE_URI=f"sqlite:///{temp_db}",
        SQLALCHEMY_TRACK_MODIFICATIONS=False,
        TESTING=True
    )

    db.init_app(test_app)

    with test_app.app_context():
        # 创建表结构
        db.create_all()
        yield test_app

    # pytest会自动清理tmp_path下的临时文件

如果你的测试需要依赖数据库初始数据，可以在fixture里添加数据初始化逻辑，比如db.session.add(User(name="test"))后commit——因为是临时数据库，不会影响其他测试。

方案三：连接级事务隔离 #

直接操作数据库连接，绑定独立事务到测试session，确保所有操作都在未提交的事务中：

import pytest
from your_app import app, db

@pytest.fixture(scope="function")
def db_session():
    with app.app_context():
        # 获取数据库连接
        connection = db.engine.connect()
        # 开启连接级事务
        transaction = connection.begin()

        # 创建绑定到该连接的scoped session
        session = db.create_scoped_session(
            options={"bind": connection, "binds": {}}
        )

        # 替换全局session
        original_session = db.session
        db.session = session

        yield session

        # 回滚事务并关闭连接
        transaction.rollback()
        connection.close()
        # 恢复全局session
        db.session = original_session

这个方案适合复杂场景，比如测试中涉及多个session操作，能确保所有变更都在同一个未提交的事务里，测试结束后一次性回滚。

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