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

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

解决方案：

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

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

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

2. 创建表：

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

3. 运行测试用例：

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

   db.drop_all()

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