跳转到主内容
新派编程网 - 新潮编程理念,全新技术课堂!

如何使用 Flask 创建临时 MySQL 数据库进行单元测试?

flask 单元测试创建临时mysql数据库 在 flask 项目的单元测试中,经常需要使用临时数据库来隔离测试数据。然而,flask 并没有内置创建临时 mysql 数据库的功能。 解决方案: 可以使用 sqlalchemy 库手动创建临时 mysql 数据库,步骤如下: 在测试套件中创建临时数据库: MySQL(Linux) MySQL 9.6.0是面向Linux平台的2026年创新版本,核心架构迎来重大革新。其将外键约束与级联操作从InnoDB引擎层上移至SQL层,确保所有数据变更均被完整记录至Binlog,彻底解决了CDC(变更数据捕获)与主从复制中的数据不一致难题。此外,该版本引入container_aware启动选项以原生适配容器环境,并对审计日志进行了组件化重构,为追求极致数据一致性与云原生体验的开发者提供了全新选择。 下载
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()
通过以上步骤,可以在 flask 单元测试中使用临时的 mysql 数据库,在测试完成后自动销毁。

相关文章