如何为 alembic 迁移配置 pytest

Question

我正在使用 PostgreSQL 数据库和 SQLAlchemy 运行 FastAPI 应用程序。我曾经sessionmaker创建和使用数据库会话，并且有一个需要在数据库上执行的自定义 alembic 迁移脚本。

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)

但它似乎没有做任何事情。我是测试新手，我找不到关于如何正确设置它的合适指南。因此，任何形式的帮助将不胜感激。

Answer 1

我不确定是否要创建apply_migrations()数据库 - 因为这是由 Alembic 驱动的，您在这里尝试设置运行 Alembic 的环境（即，这对我来说看起来是循环的），但只是设置数据库：

@pytest.fixture(scope="session")
def alembic_engine():
    set_up_empty_test_db()
    ...

哪里set_up_empty_test_db()有一个函数，其功能正如它的名字所暗示的那样。就我而言，该函数运行一个启动 Postgres 的 docker 容器，然后创建一个空白/空数据库。

请参阅https://pytest-alembic.readthedocs.io/en/latest/setup.html#setting-up-fixtures