screen-locker/screen_locker/tests/test_shutdown_target.py

"""Tests for ``_shutdown_target``: the one formula behind reset and preview."""

from __future__ import annotations

from datetime import date, datetime, time
import json
from typing import TYPE_CHECKING
from unittest.mock import patch

import earned_time
import sickdays

from screen_locker import _grace_floor
from screen_locker._shutdown_base import base_minutes
from screen_locker._shutdown_target import (
    DayInputs,
    day_credit_count,
    derive,
    gather,
    gather_inputs,
    workout_units,
)
from screen_locker.tests._sick_helpers import mark_sick

if TYPE_CHECKING:
    from pathlib import Path

_DAY = date(2026, 10, 10)
_ISO = _DAY.isoformat()
SICK_FLOOR = earned_time.SICK_SHUTDOWN_FLOOR_MINUTES
_FLAT = {"leetcode": False, "reading": False, "anki": False, "tutor": False}


def at(hhmm: str) -> float:
    """Return the epoch seconds of a local HH:MM on the test day."""
    hours, minutes = map(int, hhmm.split(":"))
    return datetime.combine(_DAY, time(hours, minutes)).astimezone().timestamp()


def log_with(tmp_path: Path, entries: list[dict[str, object]]) -> Path:
    """Write a log file holding the given entries under the test day."""
    log = tmp_path / "log.json"
    log.write_text(json.dumps({_ISO: entries}))
    return log


class TestTarget:
    """The shutdown target derived from the day's inputs, sick or not."""

    def test_a_rest_day_stands_in_for_the_first_workout_only(self) -> None:
        """A rest day fills in only the first workout unit and nothing beyond it."""
        assert workout_units(0, rest_day=True) == 1
        assert workout_units(2, rest_day=True) == 2
        assert not workout_units(0, rest_day=False)

    def test_an_early_first_task_no_longer_lifts_the_sum(self) -> None:
        """The grace floor is gone (2026-10-17): late credits replace it."""
        target = derive(
            DayInputs(_DAY, {**_FLAT, "anki": True}, first_done=at("18:55"))
        )
        assert target.minutes == target.earned
        assert not target.lift

    def test_a_sick_day_has_the_20_00_floor(self) -> None:
        """A sick day sits at the 20:00 floor plus the gaming pay of its terms, with no lift."""
        target = derive(DayInputs(_DAY, _FLAT, first_done=at("18:55"), sick_day=True))
        assert target.resolution.sick is True
        assert target.minutes == SICK_FLOOR + sum(
            t.gaming_minutes for t in target.resolution.terms
        )
        # earned_time 0.13.0: credits stack on 20:00, so there is no lift.
        assert not target.lift

    def test_a_sick_day_stacks_gaming_pay_on_20_00(self) -> None:
        """On a sick day credits stack on 20:00 and the result is capped at 23:00."""
        busy = dict.fromkeys(_FLAT, True)
        target = derive(DayInputs(_DAY, busy, workout_credits=1, sick_day=True))
        assert target.minutes >= target.earned
        assert target.minutes == min(
            23 * 60, SICK_FLOOR + sum(t.gaming_minutes for t in target.resolution.terms)
        )
        assert not target.lift

    def test_gather_keeps_the_sick_floor_when_the_workout_lock_is_kept(self) -> None:
        """Rule 2: the 20:00 floor is the day's, whatever the keep set says."""
        mark_sick(_DAY, keep=[sickdays.LOCKER_WORKOUT])
        assert gather_inputs(_DAY, None).sick_day is True

    def test_base_minutes_keeps_the_sick_floor_when_the_workout_lock_is_kept(
        self,
    ) -> None:
        """Keeping the workout lock on a sick day does not lower the base below the sick floor."""
        plain = base_minutes(_DAY)
        assert plain < SICK_FLOOR  # reading's cut is in force on this date
        mark_sick(_DAY, keep=[sickdays.LOCKER_WORKOUT])
        assert base_minutes(_DAY) == SICK_FLOOR

    def test_not_sick_is_the_plain_sum(self) -> None:
        """A normal day's target is exactly the sum of its earned terms."""
        target = derive(DayInputs(_DAY, _FLAT))
        assert target.resolution.sick is False
        assert target.minutes == target.earned == target.resolution.shutdown_minutes

    def test_no_log_counts_no_workouts(self) -> None:
        """A missing log file counts as zero workout credits."""
        assert not day_credit_count(None, _DAY)

    def test_gather_reads_every_real_source(self, tmp_path: Path) -> None:
        """Gathering reads the log, gate answers and credit time into one target."""
        log = log_with(
            tmp_path,
            [
                {
                    "workout_data": {"type": "manual_workout", "end_time": "18:00"},
                    "timestamp": "2026-10-10T20:00:00+00:00",
                }
            ],
        )
        with (
            patch("screen_locker._shutdown_target.gate_answers", return_value=_FLAT),
            patch.object(_grace_floor, "first_credit_time", return_value=None),
        ):
            target = gather(_DAY, log)
        assert target.resolution.term("workout").answer == 1
        assert target.first_done_at == at("18:00")
        assert target.minutes == target.earned
        assert not target.rest_day