Flask+SQLAlchemy测试中临时文件无法删除问题求助

问题出在SQLAlchemy的连接池仍持有临时SQLite文件的句柄——即使调用db.session.remove()，也不会主动释放连接池里的所有连接，导致os.unlink()执行时文件被占用。以下是针对性的解决方案：

核心修复：彻底关闭SQLAlchemy连接池 #

在fixture的清理阶段，除了删除表和移除会话，必须调用db.engine.dispose()来关闭连接池中的所有连接，释放文件句柄。修改后的fixture代码如下：

import os
import tempfile

import pytest
from flask.testing import FlaskClient

from flaskr import create_app
from flaskr.models import db

@pytest.fixture
def app():
    db_fd, db_path = tempfile.mkstemp(dir="test")

    app = create_app(
        {
            "TESTING": True,
            "SQLALCHEMY_DATABASE_URI": f"sqlite:///{db_path}",
        }
    )

    # 初始化数据库无需嵌套test_client，直接用app_context即可
    with app.app_context():
        db.create_all()

    yield app

    # 清理阶段：先操作数据库，再断开所有连接
    with app.app_context():
        db.drop_all()
        db.session.remove()
        # 关键：关闭连接池内的所有连接，释放文件句柄
        db.engine.dispose()

    os.close(db_fd)
    os.unlink(db_path)

额外优化：规范测试用例的上下文使用 #

测试用例中如果只需要操作数据库，直接使用app.app_context()即可，无需嵌套test_client（test_client用于模拟HTTP请求）。上下文退出时会自动清理会话：

def test_example(app):
    with app.app_context():
        stmt = ... # 你的SQL语句或ORM查询
        result = db.session.execute(stmt)
        assert ...

原代码失败的原因 #

• db.session.remove()仅移除当前上下文的会话实例，但连接池中的底层数据库连接并未关闭，仍然持有SQLite文件的句柄。
• 原代码中嵌套的test_client上下文是多余的，反而可能导致额外的连接未被正确清理。
• os.unlink()执行时，连接池的连接仍在占用文件，因此报错。

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