flask 单元测试创建临时 mysql 数据库
在 flask 项目的单元测试中,经常需要使用临时数据库来隔离测试数据。然而,flask 并没有内置创建临时 mysql 数据库的功能。
解决方案:
可以使用 sqlalchemy 库手动创建临时 mysql 数据库,步骤如下:
在测试套件中创建临时数据库:
from sqlalchemy import create_enginefrom sqlalchemy.orm import sessionmakerfrom tempfile import temporarydirectorywith 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, stringclass testmodel(db.model): __tablename__ = "test_model" id = column(integer, primary_key=true) name = column(string(255))db.create_all()
运行测试用例:
测试结束后销毁临时数据库:
db.drop_all()
通过以上步骤,可以在 flask 单元测试中使用临时的 mysql 数据库,在测试完成后自动销毁。
以上就是如何使用 Flask 创建临时 MySQL 数据库进行单元测试?的详细内容,更多请关注范的资源库其它相关文章!
转载请注明:范的资源库 » 如何使用Flask创建临时MySQL数据库进行单元测试?