Как реализовать фикстуру с setup/teardown для БД?

Question

Используйте фикстуру с yield для разделения кода setup и teardown. Перед тестом создаётся соединение с БД, после — закрывается. Для этого подходят библиотеки pytest-postgresql или кастомные решения с SQLAlchemy.

YeaHub · Accepted Answer

1. Фикстура с yield:import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

@pytest.fixture
def db_session():
    # Setup: создание соединения
    engine = create_engine("postgresql://user:pass@localhost/test_db")
    Session = sessionmaker(bind=engine)
    session = Session()

yield session  # Тест использует session здесь

# Teardown: закрытие соединения
    session.close()
    engine.dispose()2. Пример теста:def test_user_creation(db_session):
    user = User(name="Alice")
    db_session.add(user)
    db_session.commit()
    assert db_session.query(User).count() == 13. Альтернатива с contextlib:from contextlib import contextmanager

@contextmanager
def temp_db():
    engine = create_engine("sqlite:///:memory:")
    Base.metadata.create_all(engine)
    yield engine
    Base.metadata.drop_all(engine)Вывод:Фикстуры с yield — самый чистый способ управлять ресурсами в pytest.