Function db

conftest.py:68–108

Function-scoped fixture with transaction rollback cleanup. Each test runs inside a transaction that gets rolled back at the end, providing isolation without the overhead of schema recreation. For databases that don't support transactions (e.g., MySQL MyISAM), falls back to tru

(db_module)

@pytest_asyncio.fixture(scope="function")
async def db(db_module):
    """
    Function-scoped fixture with transaction rollback cleanup.

    Each test runs inside a transaction that gets rolled back at the end,
    providing isolation without the overhead of schema recreation.

    For databases that don't support transactions (e.g., MySQL MyISAM),
    falls back to truncation cleanup.

    This is the FASTEST isolation method - use for most tests.

    Usage:
        @pytest.mark.asyncio
        async def test_something(db):
            obj = await Model.create(name="test")
            assert obj.id is not None
            # Changes are rolled back after test
    """
    # Get connection from the context using its default connection
    conn = db_module.db()

    # Check if the database supports transactions
    if conn.capabilities.supports_transactions:
        # Start a savepoint/transaction
        transaction = conn._in_transaction()
        await transaction.__aenter__()

        try:
            yield db_module
        finally:
            # Rollback the transaction (discards all changes made during test)
            class _RollbackException(Exception):
                pass

            await transaction.__aexit__(_RollbackException, _RollbackException(), None)
    else:
        # For databases without transaction support (e.g., MyISAM),
        # fall back to truncation cleanup
        yield db_module
        await _truncate_all_tables(db_module)


@pytest_asyncio.fixture(scope="function")