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

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 数据库，在测试完成后自动销毁。

以上就是如何使用 Flask 创建临时 MySQL 数据库进行单元测试？的详细内容，更多请关注硕下网其它相关文章！