Function two_databases

tests/test_two_databases.py:19–50

Fixture that sets up two separate databases for testing.

()

@pytest_asyncio.fixture(scope="function")
async def two_databases():
    """Fixture that sets up two separate databases for testing."""
    db_url = os.getenv("TORTOISE_TEST_DB", "sqlite://:memory:")

    from tortoise.backends.base.config_generator import expand_db_url

    # Expand the URL with uniqueness if it's a template
    # This ensures "models" and "events" get different DB names if TORTOISE_TEST_DB has {}
    db1_config = expand_db_url(db_url, testing=True)
    db2_config = expand_db_url(db_url, testing=True)

    ctx = TortoiseContext()
    async with ctx:
        await ctx.init(
            config={
                "connections": {
                    "models": db1_config,
                    "events": db2_config,
                },
                "apps": {
                    "models": {"models": ["tests.testmodels"], "default_connection": "models"},
                    "events": {"models": ["tests.testmodels"], "default_connection": "events"},
                },
            },
            _create_db=True,
        )
        await ctx.generate_schemas()

        db = ctx.connections.get("models")
        second_db = ctx.connections.get("events")

        yield db, second_db


def build_select_sql(db) -> str: