测试夹具中多次创建Flask应用时SQLAlchemy报错问题

绝对不是框架的问题，我之前也踩过一模一样的坑——本质是你的测试夹具在数据库初始化、会话管理或者环境隔离上没处理到位，导致后续测试的数据库环境是“残缺”的。我来给你拆解问题，再直接上正确的夹具写法。

为什么会出现「no such table: document」？ #

你说create_app每次测试都会被调用，那大概率是这两个原因：

1. 内存SQLite的连接隔离问题：SQLite的内存库是「每个连接对应一个独立数据库」，如果你的应用初始化和测试数据填充用了不同连接，后续测试的会话就会连到一个空的新内存库，自然找不到表。
2. 数据库操作不在应用上下文内：SQLAlchemy通过init_app和Flask集成时，是延迟初始化的——所有数据库操作（建表、插入数据、commit）必须在Flask的应用上下文里执行，否则会使用默认的独立连接，和应用本身的数据库完全脱节。
3. 测试后清理不彻底：如果会话没正确关闭、表没销毁，后续测试的数据库状态会混乱，甚至出现连接池复用旧连接的情况。

正确的预填充数据库夹具写法 #

假设你的应用工厂和模型结构是常规的（比如db通过init_app绑定到Flask），直接用下面的模板改就行：

第一步：定义function级别的隔离夹具 #

import pytest
from your_app import create_app
from your_app.models import db, Document

@pytest.fixture(scope="function")  # 关键：每个测试都全新生成，完全隔离
def app_with_documents():
    # 测试配置：用内存SQLite，关闭不必要的追踪，解决线程问题
    test_config = {
        "TESTING": True,
        "SQLALCHEMY_DATABASE_URI": "sqlite:///:memory:",
        "SQLALCHEMY_TRACK_MODIFICATIONS": False,
        "SQLALCHEMY_ENGINE_OPTIONS": {"connect_args": {"check_same_thread": False}}
    }

    # 创建应用实例
    app = create_app(test_config)

    # 所有数据库操作必须在应用上下文内执行！
    with app.app_context():
        # 1. 创建所有表
        db.create_all()
        
        # 2. 预填充Document数据
        test_docs = [
            Document(title="测试文档1", content="内容1"),
            Document(title="测试文档2", content="内容2")
        ]
        db.session.add_all(test_docs)
        db.session.commit()

        # 把应用和db实例传给测试用例
        yield app, db

    # 测试结束后彻底清理
    with app.app_context():
        db.session.remove()  # 先关闭会话
        db.drop_all()        # 销毁所有表

第二步：测试用例的使用方式 #

def test_document_list(app_with_documents):
    app, db = app_with_documents
    # 可以用测试客户端发请求，也可以直接操作数据库
    with app.test_client() as client:
        response = client.get("/documents")
        assert response.status_code == 200
    
    # 直接验证数据库数据
    docs = Document.query.all()
    assert len(docs) == 2

进阶优化：用事务回滚提升速度 #

如果你的测试数据量很大，每次建表删表会很慢，可以用嵌套事务+自动回滚代替创建销毁表：

@pytest.fixture(scope="function")
def app_with_documents():
    test_config = {
        "TESTING": True,
        "SQLALCHEMY_DATABASE_URI": "sqlite:///:memory:",
        "SQLALCHEMY_TRACK_MODIFICATIONS": False,
        "SQLALCHEMY_ENGINE_OPTIONS": {"connect_args": {"check_same_thread": False}}
    }
    app = create_app(test_config)

    with app.app_context():
        db.create_all()
        # 开启嵌套事务，测试结束后自动回滚
        db.session.begin_nested()
        
        # 填充测试数据
        test_docs = [
            Document(title="测试文档1", content="内容1"),
            Document(title="测试文档2", content="内容2")
        ]
        db.session.add_all(test_docs)
        db.session.commit()

        # 注册请求销毁钩子：每次测试后回滚事务
        @app.teardown_request
        def teardown_transaction(exc):
            db.session.rollback()
            db.session.begin_nested()

        yield app, db

    with app.app_context():
        db.session.remove()
        db.drop_all()

这种方式不需要每次删表，只回滚事务，测试速度会快很多。

核心要点总结 #

1. 必须用function级作用域：确保每个测试都有独立的应用和数据库环境，完全隔离。
2. 所有数据库操作在应用上下文内：db.create_all()、数据插入、commit都要放在app.app_context()里，保证和应用使用同一个数据库连接。
3. 内存SQLite要加check_same_thread=False：避免线程隔离导致的连接不一致问题。
4. 测试后彻底清理：关闭会话+销毁表（或回滚事务），防止后续测试复用旧状态。

内容的提问来源于stack exchange，提问作者W.P. McNeill