54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from pathlib import Path
|
|
|
|
from sqlalchemy import select
|
|
|
|
from app.db.engine import create_engine
|
|
from app.db.models.base import Base
|
|
from app.db.models.images import Image
|
|
from app.db.models.imports import Import
|
|
from app.db.session import create_sessionmaker
|
|
|
|
|
|
def test_models_images_insert_select(tmp_path: Path) -> None:
|
|
db_path = tmp_path / "orm_images.db"
|
|
engine = create_engine("sqlite+aiosqlite:///" + db_path.as_posix())
|
|
|
|
async def _run() -> None:
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(Base.metadata.create_all)
|
|
|
|
Session = create_sessionmaker(engine)
|
|
|
|
async with Session() as session:
|
|
imp = Import(created_by="tester", source="manual")
|
|
session.add(imp)
|
|
await session.flush()
|
|
|
|
img = Image(
|
|
illust_id=1,
|
|
page_index=0,
|
|
ext="jpg",
|
|
original_url="https://example.test/original.jpg",
|
|
proxy_path="/i/1.jpg",
|
|
random_key=0.5,
|
|
created_import_id=imp.id,
|
|
)
|
|
session.add(img)
|
|
await session.commit()
|
|
await session.refresh(img)
|
|
assert img.id == 1
|
|
|
|
async with Session() as session:
|
|
fetched = (await session.execute(select(Image))).scalars().first()
|
|
assert fetched is not None
|
|
assert fetched.illust_id == 1
|
|
assert fetched.page_index == 0
|
|
|
|
await engine.dispose()
|
|
|
|
asyncio.run(_run())
|
|
|