1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
from collections.abc import AsyncIterator
import pytest
from sqlalchemy import create_engine
from sqlalchemy.engine import Engine
from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
from tests.sqlalchemy_factory.models import Base
@pytest.fixture()
def engine() -> Engine:
return create_engine("sqlite:///:memory:")
@pytest.fixture()
def async_engine() -> AsyncEngine:
return create_async_engine("sqlite+aiosqlite:///:memory:")
@pytest.fixture(autouse=True)
async def fx_drop_create_meta(async_engine: AsyncEngine) -> AsyncIterator[None]:
async with async_engine.begin() as conn:
await conn.run_sync(Base.metadata.drop_all)
await conn.run_sync(Base.metadata.create_all)
yield
|
