如何在一些测试后明确指示 PyTest 删除数据库？

Question

I am writing unit-tests with pytest-django for a django app. I want to make my tests more performant and doing so is requiring me to keep data saved in database for a certain time and not drop it after one test only. For example:

@pytest.mark.django_db
def test_save():
    p1 = MyModel.objects.create(description="some description") # this object has the id 1
    p1.save()

@pytest.mark.django_db
def test_modify():
    p1 = MyModel.objects.get(id=1)
    p1.description = "new description"
    

What I want is to know how to keep both tests separated but both will use the same test database for some time and drop it thereafter.

Answer 1

I think what you need are pytest fixtures. They allow you yo create objects (stored in database if needed) that will be used during tests. You can have a look at pytest fixtures scope that you can set so that the fixture is not deleted from database and reloading for each test that requires it but instead is created once for a bunch of tests and deleted afterwards.

You should read the documentation of pytest fixtures ( https://docs.pytest.org/en/6.2.x/fixture.html ) and the section dedicated to fixtures' scope ( https://docs.pytest.org/en/6.2.x/fixture.html#scope-sharing-fixtures-across-classes-modules-packages-or-session ).