Как настроить pytest для миграции перегонного куба ⇐ Python

[Anonymous]

Я запускаю приложение FastAPI с базой данных PostgreSQL и SQLAlchemy. Я использовал sessionmaker для создания и использования сеансов базы данных, и у меня есть специальный сценарий миграции перегонного куба, который необходимо выполнить в базе данных.

Код: Выделить всё

from app.core.config import DATABASE_URL
from app.models import User, Passwords
from app.db import Base

from alembic import context

config = context.config
config.set_main_option('sqlalchemy.url', DATABASE_URL)

fileConfig(config.config_file_name)

target_metadata = Base.metadata

def run_migrations_offline():

url = config.get_main_option("sqlalchemy.url")
context.configure(
url=url,
target_metadata=target_metadata,
literal_binds=True,
dialect_opts={"paramstyle": "named"},
)

with context.begin_transaction():
context.run_migrations()

Я хочу иметь возможность создать тестовую базу данных, применить ту же миграцию и удалить ее после теста. Я попробовал следующее в conftest.py

Код: Выделить всё

# Apply migrations at beginning and end of testing session
@pytest.fixture(scope="session")
def apply_migrations():
warnings.filterwarnings("ignore", category=DeprecationWarning)
environ["TEST"] = "True"
config = Config("alembic.ini")
alembic.command.upgrade(config, "head")
yield
alembic.command.downgrade(config, "base")

@pytest.fixture(scope="session")
def create_test_db():
url = f"{str(DATABASE_URL)}_test"

if database_exists(url):
drop_database(url)
create_database(url)
apply_migrations()
engine = create_engine(url, echo=True)

yield
drop_database(url)

Но, похоже, это ничего не делает.
Я новичок в тестировании и не могу найти достойного руководства о том, как правильно это настроить. Поэтому любая помощь будет принята с благодарностью.

Подробнее здесь: https://stackoverflow.com/questions/690 ... migrations