Function multi_db

tests/test_query_api.py:317–335

Fixture that sets up multiple databases for testing.

()

@pytest_asyncio.fixture
async def multi_db():
    """Fixture that sets up multiple databases for testing."""
    from tortoise.context import TortoiseContext

    ctx = TortoiseContext()
    async with ctx:
        await ctx.init(
            config={
                "connections": {
                    "first": "sqlite://:memory:",
                    "second": "sqlite://:memory:",
                },
                "apps": {
                    "models": {"models": [__name__], "default_connection": "first"},
                },
            }
        )
        await ctx.generate_schemas()
        yield ctx


@pytest.mark.asyncio