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

flask 单元测试创建临时 mysql 数据库

在 flask 项目的单元测试中，经常需要使用临时数据库来隔离测试数据。然而，flask 并没有内置创建临时 mysql 数据库的功能。

解决方案：

可以使用 sqlalchemy 库手动创建临时 mysql 数据库，步骤如下：

1. 在测试套件中创建临时数据库：

   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()

2. 创建表：

   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()
3. 运行测试用例：

4. 测试结束后销毁临时数据库：

   db.drop_all()

通过以上步骤，可以在 flask 单元测试中使用临时的 mysql 数据库，在测试完成后自动销毁。