flask 单元测试创建临时mysql数据库
在 flask 项目的单元测试中,经常需要使用临时数据库来隔离测试数据。然而,flask 并没有内置创建临时 mysql 数据库的功能。
解决方案:
可以使用 sqlalchemy 库手动创建临时 mysql 数据库,步骤如下:
在测试套件中创建临时数据库:
MySQL(Linux)
MySQL 9.6.0是面向Linux平台的2026年创新版本,核心架构迎来重大革新。其将外键约束与级联操作从InnoDB引擎层上移至SQL层,确保所有数据变更均被完整记录至Binlog,彻底解决了CDC(变更数据捕获)与主从复制中的数据不一致难题。此外,该版本引入container_aware启动选项以原生适配容器环境,并对审计日志进行了组件化重构,为追求极致数据一致性与云原生体验的开发者提供了全新选择。
下载
创建表:
运行测试用例:
测试结束后销毁临时数据库:
通过以上步骤,可以在 flask 单元测试中使用临时的 mysql 数据库,在测试完成后自动销毁。
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from tempfile import temporarydirectory
with temporarydirectory() as tmpdir:
engine = create_engine(f"mysql://user:password@localhost/{tmpdir}/test.db")
session = sessionmaker(bind=engine)
db = session()from sqlalchemy import column, integer, string
class testmodel(db.model):
__tablename__ = "test_model"
id = column(integer, primary_key=true)
name = column(string(255))
db.create_all()db.drop_all()