screen-locker/screen_locker/tests/test_shutdown_flat_earners.py

"""Tests for the pass over every flat earner, and for the base it projects.

The registry, not this repo, lists the earners: a newly registered flat
earner must be a term of the reset and an hour of the live pass with no code
here. Also covers one real signed ledger end to end, and the status
projection following the 2026-10-01 reading cut.
"""

from __future__ import annotations

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

import earned_time
import pytest

from screen_locker import _earned
from screen_locker._day import today_str
from screen_locker._shutdown_base import (
    apply_flat_bonuses_if_new,
    base_minutes,
    reset_to_base_if_new_day,
)
from screen_locker._status_data import gather_status
from screen_locker.tests._earned_fixtures import (
    EXTRA,
    answering,
    credit,
    register,
    signing_key,
    write_ledger,
)
from screen_locker.tests.test_status_data import _files

if TYPE_CHECKING:
    from collections.abc import Iterator
    from pathlib import Path

    from screen_locker._status_types import ShutdownProjection


def _mixin() -> MagicMock:
    mixin = MagicMock()
    mixin._read_shutdown_config.return_value = (base_minutes(), base_minutes(), 300)
    mixin._write_shutdown_config.return_value = True
    mixin._adjust_shutdown_time_by.return_value = True
    return mixin


@pytest.fixture
def answers() -> Iterator[dict[str, bool | None]]:
    """Every flat earner says no until a test sets it."""
    given: dict[str, bool | None] = {"leetcode": False, "reading": False}
    with answering(given):
        yield given


@pytest.fixture
def with_extra(monkeypatch: pytest.MonkeyPatch) -> None:
    """Register EXTRA after the existing earners, as a new gate would be."""
    register(monkeypatch, EXTRA)


class TestFlatBonuses:
    def test_both_earned_apply_both_and_keep_both_stamps(
        self, tmp_path: Path, answers: dict[str, bool | None]
    ) -> None:
        """The second save must not overwrite the first one's stamp."""
        answers["leetcode"] = True
        answers["reading"] = True
        state = tmp_path / "state.json"
        mixin = _mixin()
        apply_flat_bonuses_if_new(state, mixin)
        assert mixin._adjust_shutdown_time_by.call_count == 2
        assert json.loads(state.read_text()) == {
            "leetcode_bonus_date": today_str(),
            "reading_bonus_date": today_str(),
        }
        apply_flat_bonuses_if_new(state, mixin)
        assert mixin._adjust_shutdown_time_by.call_count == 2

    @pytest.mark.usefixtures("pre_ladder")
    def test_only_reading_earned_applies_only_reading(
        self, tmp_path: Path, answers: dict[str, bool | None]
    ) -> None:
        answers["reading"] = True
        state = tmp_path / "state.json"
        mixin = _mixin()
        apply_flat_bonuses_if_new(state, mixin)
        mixin._adjust_shutdown_time_by.assert_called_once_with(60, at=None)
        assert json.loads(state.read_text()) == {"reading_bonus_date": today_str()}


@pytest.mark.usefixtures("with_extra")
class TestANewlyRegisteredEarner:
    def test_reset_includes_and_stamps_it(
        self, tmp_path: Path, answers: dict[str, bool | None]
    ) -> None:
        answers["extra"] = True
        state = tmp_path / "state.json"
        mixin = _mixin()
        assert reset_to_base_if_new_day(state, mixin) is True
        mixin._write_shutdown_config.assert_called_once_with(
            base_minutes() + 60, base_minutes() + 60, 300, restore=True
        )
        assert json.loads(state.read_text()) == {
            "last_reset_date": today_str(),
            "extra_bonus_date": today_str(),
        }

    def test_live_pass_applies_it_once(
        self, tmp_path: Path, answers: dict[str, bool | None]
    ) -> None:
        answers["extra"] = True
        state = tmp_path / "state.json"
        mixin = _mixin()
        apply_flat_bonuses_if_new(state, mixin)
        apply_flat_bonuses_if_new(state, mixin)
        mixin._adjust_shutdown_time_by.assert_called_once_with(60, at=None)
        assert json.loads(state.read_text()) == {"extra_bonus_date": today_str()}

    def test_its_ledger_is_read_like_any_other(self, tmp_path: Path) -> None:
        write_ledger(EXTRA, [credit(EXTRA)])
        mixin = _mixin()
        with signing_key(tmp_path):
            apply_flat_bonuses_if_new(tmp_path / "state.json", mixin)
        mixin._adjust_shutdown_time_by.assert_called_once_with(60, at=None)

    def test_a_penalised_earner_lowers_the_base_from_its_day(
        self, monkeypatch: pytest.MonkeyPatch
    ) -> None:
        """The base is read from the registry at call time, never a constant."""
        eve, cut = date(2026, 10, 2), date(2026, 10, 3)
        penalised = earned_time.Earner(
            name="cut",
            label="cut",
            gaming_minutes=60,
            shutdown_minutes=60,
            penalty_from=cut,
        )
        before = (base_minutes(eve), base_minutes(cut))
        register(monkeypatch, penalised)
        assert (base_minutes(eve), base_minutes(cut)) == (before[0], before[1] - 60)


class TestRealLedgers:
    @pytest.mark.usefixtures("pre_ladder")
    def test_signed_credits_on_disk_reach_the_reset(self, tmp_path: Path) -> None:
        """No mock between the ledgers and the written hour."""
        # Pinned to local midnight, so the credits are never late ones.
        midnight = datetime.now(tz=UTC).astimezone().replace(hour=0, minute=0, second=0)
        write_ledger(earned_time.LEETCODE, [credit(earned_time.LEETCODE, midnight)])
        write_ledger(earned_time.READING, [credit(earned_time.READING, midnight)])
        state = tmp_path / "state.json"
        mixin = _mixin()
        with signing_key(tmp_path):
            assert reset_to_base_if_new_day(state, mixin) is True
            # Under the same key: the base reads the ledgers too (maturity).
            base = base_minutes()
        mixin._write_shutdown_config.assert_called_once_with(
            base + 120, base + 120, 300, restore=True
        )
        assert json.loads(state.read_text()) == {
            "last_reset_date": today_str(),
            "leetcode_bonus_date": today_str(),
            "reading_bonus_date": today_str(),
        }

    def test_the_live_pass_hands_on_the_credit_s_own_time(self, tmp_path: Path) -> None:
        """A late credit counts from its ledger row, never the pass's clock."""
        midnight = datetime.now(tz=UTC).astimezone().replace(hour=0, minute=0, second=0)
        write_ledger(earned_time.LEETCODE, [credit(earned_time.LEETCODE, midnight)])
        mixin = _mixin()
        with signing_key(tmp_path), answering({"leetcode": True}):
            apply_flat_bonuses_if_new(tmp_path / "state.json", mixin)
        mixin._adjust_shutdown_time_by.assert_called_once_with(
            _earned.first_minutes(earned_time.LEETCODE, midnight.date()),
            at=float(int(midnight.timestamp())),
        )


class TestProjectionFollowsTheCut:
    """Every projected row is priced on its own date, not on today's base."""

    def _shutdown(self, tmp_path: Path, now: datetime) -> ShutdownProjection:
        with patch(
            "screen_locker._status_data.has_workout_skip_today", return_value=False
        ):
            return gather_status(**_files(tmp_path), now=now).shutdown

    def test_the_reading_cut_lands_mid_week(self, tmp_path: Path) -> None:
        """Wed 2026-09-30 is the last 20:00 day; Thu 2026-10-01 is the first 19:00."""
        shutdown = self._shutdown(tmp_path, datetime(2026, 9, 30, 12, 0, tzinfo=UTC))
        assert [d.minutes for d in shutdown.rest_of_week] == [20 * 60] * 3 + [
            19 * 60
        ] * 4

    def test_next_week_preview_includes_a_cut_dated_after_today(
        self, tmp_path: Path
    ) -> None:
        """On Mon 2026-10-05, the cuts from Tue already show; next week has them."""
        shutdown = self._shutdown(tmp_path, datetime(2026, 10, 5, 12, 0, tzinfo=UTC))
        tuesday = date(2026, 10, 6)
        lowered = 19 * 60 - sum(
            e.shutdown_minutes for e in earned_time.EARNERS if e.penalty_from == tuesday
        )

        def base(day: date) -> int:
            # From the sleep ladder (Sat 2026-10-10) the registry prices the
            # day itself; before it, the cut from Tuesday holds.
            # The registry in force that day (the tutor cutover switches it),
            # and from today on its first credits: a never-paid tutor spares
            # the floor (earned_time 0.6.1).
            earners = _earned.registry(day)
            starts = _earned.first_credits(earners, day)
            return earned_time.base_for(
                day, earners, first_credits=starts
            ).shutdown_minutes

        week = [date(2026, 10, 5 + i) for i in range(7)]
        assert [base(d) for d in week[1:4]] == [lowered] * 3
        # Fri 2026-10-09 the Anki waiver drops Tuesday's cuts again.
        assert week[4] == earned_time.ANKI_WAIVED_FROM
        assert base(week[4]) == 19 * 60
        assert [d.minutes for d in shutdown.rest_of_week] == [19 * 60] + [
            base(d) for d in week[1:]
        ]
        assert [d.minutes for d in shutdown.next_week_preview] == [
            base(date(2026, 10, 12 + i)) for i in range(7)
        ]