mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 16:53:58 +01:00
Krzysztof kuhy Rudnicki
aa3e290391
Real one-line docstrings stating what each test pins down. Files that would pass 250 lines moved whole classes into _part2 siblings (sick_panel, earned, poke_server, shutdown_reading, manual_workout_dialog). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01WzxdWYdwRAtMg7ki6TKxuU
128 lines
4.9 KiB
Python
128 lines
4.9 KiB
Python
"""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
|