85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
from datetime import UTC, datetime
|
|
|
|
import pytest
|
|
|
|
from backend.app.models import (
|
|
Artist,
|
|
ImageProvenance,
|
|
ImageRecord,
|
|
Post,
|
|
Source,
|
|
Tag,
|
|
TagKind,
|
|
)
|
|
from backend.app.models.tag import image_tag
|
|
from backend.app.services.artist_service import ArtistService
|
|
|
|
pytestmark = pytest.mark.integration
|
|
|
|
|
|
async def _fixture(db):
|
|
artist = Artist(name="Nadia", slug="nadia")
|
|
db.add(artist)
|
|
await db.flush()
|
|
src = Source(artist_id=artist.id, platform="web", url="http://x")
|
|
db.add(src)
|
|
await db.flush()
|
|
post = Post(
|
|
source_id=src.id, external_post_id="p1",
|
|
post_date=datetime(2026, 3, 1, tzinfo=UTC),
|
|
)
|
|
db.add(post)
|
|
await db.flush()
|
|
img = ImageRecord(
|
|
path="/images/a/1.jpg", sha256="a" + "0" * 63,
|
|
size_bytes=1, mime="image/jpeg", width=4, height=8,
|
|
origin="downloaded", integrity_status="unknown",
|
|
)
|
|
db.add(img)
|
|
await db.flush()
|
|
db.add(ImageProvenance(
|
|
image_record_id=img.id, post_id=post.id, source_id=src.id))
|
|
tag = Tag(name="forest", kind=TagKind.general)
|
|
db.add(tag)
|
|
await db.flush()
|
|
await db.execute(image_tag.insert().values(
|
|
image_record_id=img.id, tag_id=tag.id, source="manual"))
|
|
await db.flush()
|
|
return artist, src, img
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_overview_aggregates(db):
|
|
artist, src, img = await _fixture(db)
|
|
svc = ArtistService(db)
|
|
ov = await svc.overview("nadia")
|
|
assert ov["name"] == "Nadia"
|
|
assert ov["image_count"] == 1
|
|
assert ov["date_range"]["min"].startswith("2026-03-01")
|
|
assert ov["date_range"]["max"].startswith("2026-03-01")
|
|
assert any(t["name"] == "forest" for t in ov["cooccurring_tags"])
|
|
assert ov["sources"][0]["image_count"] == 1
|
|
assert ov["activity"][0]["count"] == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_overview_unknown_slug_returns_none(db):
|
|
svc = ArtistService(db)
|
|
assert await svc.overview("ghost") is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_paged_images(db):
|
|
artist, src, img = await _fixture(db)
|
|
svc = ArtistService(db)
|
|
page = await svc.images("nadia", cursor=None, limit=10)
|
|
assert page is not None
|
|
assert len(page.images) == 1
|
|
assert page.images[0]["thumbnail_url"].startswith("/images/thumbs/")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_paged_images_unknown_slug_none(db):
|
|
svc = ArtistService(db)
|
|
assert await svc.images("ghost", cursor=None, limit=10) is None
|