python - 我需要 Python 中的数据库清理器吗？

Question

我有 Rails 背景，是 Python 的新手。不幸的是，我的谷歌搜索和搜索没有成功。

我们正在构建一个没有框架的 Python 应用程序。我们使用 pytest 进行测试，使用 factory-boy 进行工厂测试。

我想知道我们的应用程序是否需要数据库清理器。我找到了 pytest_sessionstart 和 pytest_sessionfinish Hook ，它们似乎是进行清理的好地方。但我不知道我们是否需要像在 Rails 测试中使用的数据库清理器这样的东西。

非常感谢任何信息或帮助!如果我确实需要某种更简洁的方法，明确的说明或文档链接将非常有用。

Answer 1

当我有类似的需求时，我使用 Pytest fixtures 来完成这项工作(或 yield fixtures，它更适合现代 python)。

例如

# without editor, so with errors
@pytest.yield_fixture()
def db():
    db = create_db()
try:
    yield db
finally:
    cleanup_db(db)

def test_something(db):
   do_ugly_things(db)
   assert false