NixReaperandClaude Sonnet 5
606d2532c4
CI / lint (push) Canceled after 0s
CI / typecheck (push) Canceled after 0s
CI / unit-tests (push) Canceled after 0s
CI / integration-tests (push) Canceled after 0s
CI / container-build (push) Canceled after 0s
CI / dependency-audit (push) Canceled after 0s
FilmographyCreditOut already aggregated a guest star's episode appearances into one row per series with an episode_count, but never exposed which episodes those were even though the individual episode Credit rows exist in the database. Adds an `episodes` list (season/ episode number, title, air date, sorted chronologically) alongside the existing count -- closes a known limitation tracked in docs/api.md. Co-Authored-By: Claude Sonnet 5 <[email protected]>
122 lines
4.0 KiB
Python
122 lines
4.0 KiB
Python
"""Integration tests for person filmography assembly
|
|
(app/services/person_service.py:get_person_credits) — movies, regular
|
|
series roles, and guest-star episode credits aggregated per series."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import date
|
|
|
|
import pytest
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.ids import aurora_id
|
|
from app.models.credit import Credit
|
|
from app.models.movie import Movie
|
|
from app.models.person import Person
|
|
from app.models.tv import Episode, Season, Series
|
|
from app.services.person_service import get_person_credits
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_person_credits_include_movies_series_and_aggregated_guest_spots(
|
|
db_session: AsyncSession,
|
|
):
|
|
person = Person(name="Guest Actor")
|
|
db_session.add(person)
|
|
await db_session.flush()
|
|
|
|
movie = Movie(title="A Movie", sort_title="Movie", primary_release_date=date(2020, 1, 1))
|
|
db_session.add(movie)
|
|
await db_session.flush()
|
|
db_session.add(
|
|
Credit(
|
|
person_id=person.id,
|
|
entity_type="movie",
|
|
entity_id=movie.id,
|
|
department="Acting",
|
|
character="Hero",
|
|
)
|
|
)
|
|
|
|
series = Series(
|
|
title="A Series", sort_title="Series", first_air_date=date(2019, 5, 1)
|
|
)
|
|
db_session.add(series)
|
|
await db_session.flush()
|
|
db_session.add(
|
|
Credit(
|
|
person_id=person.id,
|
|
entity_type="series",
|
|
entity_id=series.id,
|
|
department="Acting",
|
|
character="Regular Character",
|
|
)
|
|
)
|
|
|
|
season = Season(series_id=series.id, season_number=1)
|
|
db_session.add(season)
|
|
await db_session.flush()
|
|
|
|
# Inserted out of episode-number order deliberately, to prove the
|
|
# aggregated breakdown sorts by (season, episode) rather than insertion
|
|
# or credit-creation order.
|
|
episode1 = Episode(
|
|
series_id=series.id,
|
|
season_id=season.id,
|
|
season_number=1,
|
|
episode_number=2,
|
|
title="Second Appearance",
|
|
air_date=date(2019, 5, 15),
|
|
)
|
|
episode2 = Episode(
|
|
series_id=series.id,
|
|
season_id=season.id,
|
|
season_number=1,
|
|
episode_number=1,
|
|
title="First Appearance",
|
|
air_date=date(2019, 5, 8),
|
|
)
|
|
db_session.add_all([episode1, episode2])
|
|
await db_session.flush()
|
|
|
|
for episode in (episode1, episode2):
|
|
db_session.add(
|
|
Credit(
|
|
person_id=person.id,
|
|
entity_type="episode",
|
|
entity_id=episode.id,
|
|
department="Acting",
|
|
character="Guest Character",
|
|
is_guest=True,
|
|
)
|
|
)
|
|
await db_session.commit()
|
|
|
|
credits_out = await get_person_credits(db_session, aurora_id("person", str(person.id)))
|
|
|
|
titles = {c.title for c in credits_out.cast}
|
|
assert titles == {"A Movie", "A Series"}
|
|
|
|
movie_credit = next(c for c in credits_out.cast if c.media_type == "movie")
|
|
assert movie_credit.character == "Hero"
|
|
assert movie_credit.episode_count is None
|
|
|
|
# Both the regular-cast series credit and the two guest-star episode
|
|
# credits point at the same series, but must NOT collapse into one
|
|
# entry -- they have different characters/roles.
|
|
series_entries = [c for c in credits_out.cast if c.media_type == "series"]
|
|
assert len(series_entries) == 2
|
|
|
|
regular = next(c for c in series_entries if c.character == "Regular Character")
|
|
assert regular.episode_count is None
|
|
|
|
guest = next(c for c in series_entries if c.character == "Guest Character")
|
|
assert guest.episode_count == 2
|
|
assert len(guest.episodes) == 2
|
|
assert [e.episode_number for e in guest.episodes] == [1, 2]
|
|
assert [e.title for e in guest.episodes] == ["First Appearance", "Second Appearance"]
|
|
assert guest.episodes[0].season_number == 1
|
|
assert guest.episodes[0].air_date == date(2019, 5, 8)
|
|
assert guest.episodes[0].id == aurora_id("episode", str(episode2.id))
|
|
assert guest.episodes[1].id == aurora_id("episode", str(episode1.id))
|