pytest 中的刷新数据库

在编写测试时，有一个重要的概念叫做“工作单元”，它指出每个测试应该只专注于测试一个特定的东西，通常是一个函数或一个类。此外，每个测试都应该是独立的，这意味着一个测试的结果不应影响另一个测试的结果。

因此，在每次测试之前确保数据库是干净的是至关重要的。

在本文中，我将演示如何使用 pytest 实现数据库刷新。

pytest 中有 5 个作用域：

function: 默认作用域，fixture 在测试结束时被销毁。
class：夹具teardown在班级的最后一次测试期间被销毁。
module：夹具在模块（single_test_file.py）中最后一个测试的拆卸过程中被破坏。
package：夹具在拆解包（单个/文件夹）中的最后一个测试期间被破坏。
session：夹具在测试会话结束时被销毁。（直到命令行退出）
因此，我们可以在每次测试后使用function作用域回滚和截断数据库的所有表，并session在所有测试结束后使用作用域删除所有表。

这种方法确保数据库模式保持不变并防止测试失败。

以下是使用 SQLAlchemy 实现它的方法：

# conftest.py

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative <