Function db_postgres

tests/contrib/postgres/conftest.py:43–60

Function-scoped fixture with transaction rollback for postgres tests. Equivalent to: test.TestCase with standard testmodels.

(db_module_postgres)

@pytest_asyncio.fixture(scope="function")
async def db_postgres(db_module_postgres):
    """
    Function-scoped fixture with transaction rollback for postgres tests.

    Equivalent to: test.TestCase with standard testmodels.
    """
    conn = db_module_postgres.db()
    transaction = conn._in_transaction()
    await transaction.__aenter__()

    try:
        yield db_module_postgres
    finally:

        class _RollbackException(Exception):
            pass

        await transaction.__aexit__(_RollbackException, _RollbackException(), None)


@pytest_asyncio.fixture(scope="function")