使用 pytest 进行测试时如何使用夹具正确清除数据库

Question

我有一个测试数据库。假设其中创建了三个表：

create table users (
    id serial constraint users_pkey primary key,
    name text not null
);
create table roles (
    id serial constraint roles_pkey primary key,
    name text not null
);
create table users_roles (
    user_id int constraint users_roles_users_fkey references users(id),
    role_id int constraint users_roles_roles_fkey references roles(id)
);

每个测试都从使用 factory_boy 工厂的数据填充数据库开始。典型的测试如下所示：

def test_get_user_roles(api_client):
    user = UserFactory.create(id=1)
    role = RolesFactory.create(id=1)
    users_roles = UsersRolesFactory.create(user_id=1, role_id=1)
    response = api_client.get(f"/users/{user.id}/roles")
    assert ...

我可以使用 @pytest.mark.usefixtures("clear_users_roles", "clear_users", "clear_roles") 清除表，其中 "clear_users_roles"、"clear_users"、"clear_roles" 是如果没有明显清除表的装置表之间的关系。 但是在上面的示例中，表 users_roles 和用户之间以及 users_roles 和角色之间存在关系（外键），问题是固定装置无序运行。 因此，当我运行我的测试时，我得到一个完整性错误，因为，例如，夹具“clear_users”在“clear_users_roles”之前执行，并且我的 RDBMS 无法删除记录，因为该记录仍然引用表“users”。有没有按特定顺序运行固定装置的正确方法？或者对于这样的案例可能有一些模式/最佳实践？

Answer 1

您是否尝试过在带有 yield 关键字的夹具中使用拆卸代码？

来自pytest documentation的示例：

# content of conftest.py

import smtplib
import pytest


@pytest.fixture(scope="module")
def smtp_connection():
    smtp_connection = smtplib.SMTP("smtp.gmail.com", 587, timeout=5)
    yield smtp_connection  # provide the fixture value
    print("teardown smtp")
    smtp_connection.close()