使用Peewee与Pytest时，如何自动搭建干净的测试数据库/表环境？

除了每次创建/删除表，还有几种更高效的方式实现测试间的数据库环境隔离：

1. 利用事务回滚（推荐） #

Peewee原生支持事务操作，我们可以在每个测试前开启事务，测试结束后直接回滚，这样所有测试产生的数据都不会被持久化，比反复建删表性能高很多。

示例代码：

import pytest
from peewee import SqliteDatabase

# 假设你的数据库实例为db
db = SqliteDatabase('test.db')

@pytest.fixture(autouse=True)
def transactional_test():
    # 开启事务
    with db.atomic() as txn:
        yield  # 执行测试逻辑
        # 测试结束后强制回滚事务，丢弃所有临时数据
        txn.rollback()

这种方式的优势：

• 无需频繁建删表，测试执行速度更快
• 自动处理测试异常，哪怕测试失败也能保证数据回滚

2. 使用内存数据库 #

如果测试不需要持久化数据，可以改用SQLite内存数据库，每个测试重新初始化连接，每次都是完全干净的全新环境：

import pytest
from peewee import SqliteDatabase
from your_models import MyTable, MyOtherTable

@pytest.fixture(autouse=True)
def in_memory_db():
    # 创建内存数据库实例
    db = SqliteDatabase(':memory:')
    # 将模型临时绑定到内存数据库
    MyTable._meta.database = db
    MyOtherTable._meta.database = db
    # 一次性创建所有需要的表
    db.create_tables([MyTable, MyOtherTable])
    yield db
    # 测试结束后关闭连接
    db.close()

注意：如果你的模型原本绑定了其他数据库，需要在测试中临时切换到内存库，确保测试环境完全独立。

3. 批量清空表数据 #

如果不想用事务或内存库，也可以在每个测试前后清空表数据，比删表重建更高效：

@pytest.fixture(autouse=True)
def clean_tables():
    # 测试前清空所有表数据，cascade处理外键关联
    MyTable.truncate_table(cascade=True)
    MyOtherTable.truncate_table(cascade=True)
    yield
    # 测试后再次清空，避免残留数据影响后续测试
    MyTable.truncate_table(cascade=True)
    MyOtherTable.truncate_table(cascade=True)

这里的cascade=True很关键，能自动处理外键关联的数据清空，避免因关联约束导致的报错。

内容的提问来源于stack exchange，提问作者Jimson James