Pytest и очистка базы данных перед запуском тестов

Question

я использую Flask для создания веб-службы и pytest для тестирования

я использую приспособления pytest для настройки и разрыва ресурсов тестирования, но мне нужно протестировать конечную точку POST, которая создаст несколько записей вбаза данных

Как мы можем очистить эти записи?

Answer 1

Вы можете использовать прибор для очистки .

@pytest.fixture
def cleanup():
    yield
    # This is executed when the test using the fixture is done
    db_cleanup()

def test_records_created(cleanup):     # pylint: disable=redefined-outer-name,unused-argument
    response = app.test_client().post('/path', json=payload)
    assert response.status_code == 200
    assert ...