test_wrapped.py

"""Reading Wrapped: year scoping, standout reads, sourced theme breakdown."""

from __future__ import annotations

import datetime

from app.wrapped import compute_wrapped, year_bounds
from ingest.models import (
    Author,
    Book,
    DailyActivity,
    ReadingStat,
    ReadingState,
    ReadingStatus,
    Source,
    SourceKind,
    ThemeTag,
)

_EPOCH = datetime.date(1970, 1, 1).toordinal()


def _ordinal(y: int, m: int, d: int) -> int:
    return datetime.date(y, m, d).toordinal() - _EPOCH


def _finished_state(title: str, theme: str, read_seconds: int, ts: int) -> ReadingState:
    src = Source(SourceKind.CURATED_LIST, "curated-list:x", "2026-06-05", theme)
    book = Book(
        book_id=title, title=title, authors=(Author("A"),), theme_tags=(ThemeTag(theme, src),)
    )
    stat = ReadingStat(
        key=title,
        title=title,
        authors=("A",),
        pages_read=100,
        total_pages=100,
        read_time_seconds=read_seconds,
        last_read_ts=ts,
        sessions=3,
    )
    return ReadingState(
        title=title, authors=("A",), status=ReadingStatus.FINISHED, book=book, stat=stat
    )


def test_year_bounds() -> None:
    lo, hi = year_bounds(2024)
    assert lo == _ordinal(2024, 1, 1)
    assert hi == _ordinal(2025, 1, 1)


def test_wrapped_scopes_to_year() -> None:
    ts_2024 = int(datetime.datetime(2024, 6, 1, tzinfo=datetime.UTC).timestamp())
    ts_2023 = int(datetime.datetime(2023, 6, 1, tzinfo=datetime.UTC).timestamp())
    states = [
        _finished_state("In Year", "trans", 7200, ts_2024),
        _finished_state("Last Year", "trans", 9999, ts_2023),
    ]
    day = _ordinal(2024, 6, 1)
    activity = [DailyActivity(day_ordinal=day, seconds=3600, pages=40)]
    wrapped = compute_wrapped(states, activity, 2024)
    assert wrapped.books_finished == 1
    assert wrapped.standout_reads[0].title == "In Year"
    assert dict(wrapped.theme_breakdown).get("trans") == 1
    assert wrapped.days_read == 1


def test_standouts_ranked_by_time() -> None:
    ts = int(datetime.datetime(2024, 3, 1, tzinfo=datetime.UTC).timestamp())
    states = [
        _finished_state("Short", "a", 1000, ts),
        _finished_state("Long", "a", 9000, ts),
    ]
    wrapped = compute_wrapped(states, [], 2024)
    assert [r.title for r in wrapped.standout_reads] == ["Long", "Short"]
    assert wrapped.standout_reads[0].total_read_time_hours == 2.5


def test_standout_hours_are_all_time_and_the_year_knows_it() -> None:
    """A book started before the year keeps its earlier hours, and Wrapped says so.

    KOReader has one cumulative total per book, so the standout hours are not a
    partition of the year's hours. The flag exists so the renderer can name the
    difference instead of leaving a reader to find it.
    """
    ts = int(datetime.datetime(2024, 3, 1, tzinfo=datetime.UTC).timestamp())
    # 10 hours of lifetime reading on a book finished in 2024, of which only one
    # hour happened in 2024.
    states = [_finished_state("Long Haul", "a", 36_000, ts)]
    activity = [DailyActivity(day_ordinal=_ordinal(2024, 3, 1), seconds=3600, pages=40)]

    wrapped = compute_wrapped(states, activity, 2024)

    assert wrapped.read_time_hours == 1.0
    assert wrapped.standout_reads[0].total_read_time_hours == 10.0
    assert wrapped.standouts_exceed_the_year is True


def test_standouts_inside_the_year_do_not_trip_the_flag() -> None:
    ts = int(datetime.datetime(2024, 3, 1, tzinfo=datetime.UTC).timestamp())
    states = [_finished_state("Quick", "a", 3600, ts)]
    activity = [DailyActivity(day_ordinal=_ordinal(2024, 3, 1), seconds=7200, pages=90)]

    wrapped = compute_wrapped(states, activity, 2024)

    assert wrapped.standouts_exceed_the_year is False


def test_the_demo_world_is_the_case_that_exceeds_the_year(
    states: list, daily_activity: list
) -> None:
    """The shipped demo is exactly the shape the caption has to explain.

    Its five standouts total more than the year they sit inside, so the demo
    dashboard — the artifact the a11y gate audits and the one a visitor sees —
    is the page that must carry the reconciliation, not a hypothetical library.
    """
    wrapped = compute_wrapped(states, daily_activity, 2024)
    standout_hours = sum(r.total_read_time_hours for r in wrapped.standout_reads)

    assert standout_hours > wrapped.read_time_hours
    assert wrapped.standouts_exceed_the_year is True


def test_wrapped_with_real_demo(states: list, daily_activity: list) -> None:
    # The demo canon was read in 2024 (timestamps ~1.71e9).
    wrapped = compute_wrapped(states, daily_activity, 2024)
    assert wrapped.books_finished >= 1
    assert wrapped.read_time_hours > 0