mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 20:33:56 +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
120 lines
4.6 KiB
Python
120 lines
4.6 KiB
Python
"""The poke credit's day reset, workout hand-off and the two reply snapshots."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import TYPE_CHECKING
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from screen_locker import _poke_credit
|
|
from screen_locker._poke_credit import PokeLocker
|
|
from screen_locker._workout_credit import WorkoutCreditResult
|
|
|
|
if TYPE_CHECKING:
|
|
from pathlib import Path
|
|
|
|
|
|
@pytest.fixture(name="log_file")
|
|
def log_file_fixture(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> Path:
|
|
"""A fresh PokeLocker on a tmp log; the day reset stubbed."""
|
|
log = tmp_path / "poke_log.json"
|
|
|
|
def headless(locker_cls: type[PokeLocker]) -> PokeLocker:
|
|
"""Build a PokeLocker without a display, pointed at the temporary log."""
|
|
locker = object.__new__(locker_cls)
|
|
locker.log_file = log
|
|
locker.workout_data = {}
|
|
return locker
|
|
|
|
monkeypatch.setattr(_poke_credit, "_headless_locker", headless)
|
|
monkeypatch.setattr(_poke_credit, "process_week_transition", lambda *_: [])
|
|
monkeypatch.setattr(_poke_credit, "reset_to_base_if_new_day", lambda *_a, **_k: 0)
|
|
return log
|
|
|
|
|
|
@pytest.fixture(name="credit")
|
|
def credit_fixture(monkeypatch: pytest.MonkeyPatch) -> MagicMock:
|
|
"""Replace the shared credit step."""
|
|
stub = MagicMock(
|
|
return_value=WorkoutCreditResult(
|
|
shutdown_adjusted=True,
|
|
extra_bonus_delta=0,
|
|
weekly_count=1,
|
|
already_counted_today=False,
|
|
)
|
|
)
|
|
monkeypatch.setattr(PokeLocker, "_apply_credit_for_written_entry", stub)
|
|
return stub
|
|
|
|
|
|
class TestStartTheDay:
|
|
"""Groups the day-start steps run before a poked session is credited."""
|
|
|
|
def test_new_day_runs_the_week_transition_then_the_reset(
|
|
self, log_file: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
"""Starting the day runs the week transition first, then the base reset."""
|
|
calls: list[str] = []
|
|
|
|
def week(*_args: object) -> list[str]:
|
|
"""Record the week transition and return a streak message."""
|
|
calls.append("week")
|
|
return ["+1h streak"]
|
|
|
|
def reset(*_args: object, **_kwargs: object) -> bool:
|
|
"""Record the base reset and report it ran."""
|
|
calls.append("reset")
|
|
return True
|
|
|
|
monkeypatch.setattr(_poke_credit, "process_week_transition", week)
|
|
monkeypatch.setattr(_poke_credit, "reset_to_base_if_new_day", reset)
|
|
_poke_credit.fresh_locker().start_the_day()
|
|
assert calls == ["week", "reset"]
|
|
|
|
def test_credit_written_sets_the_entry_and_returns_the_result(
|
|
self, log_file: Path, credit: MagicMock
|
|
) -> None:
|
|
"""credit_written stores the entry as workout data and returns the shared credit result."""
|
|
locker = _poke_credit.fresh_locker()
|
|
assert locker.credit_written({"type": "x"}, [{"a": 1}]) == credit.return_value
|
|
assert locker.workout_data == {"type": "x"}
|
|
credit.assert_called_once_with([{"a": 1}])
|
|
|
|
|
|
class TestSnapshots:
|
|
"""Groups the shutdown and gaming-minutes values snapshotted into poke replies."""
|
|
|
|
def test_current_shutdown_renders_tonights_time(
|
|
self, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
"""current_shutdown renders tonight's shutdown minutes as HH:MM."""
|
|
monkeypatch.setattr(_poke_credit, "read_shutdown_config", lambda _p: (1, 2, 3))
|
|
monkeypatch.setattr(_poke_credit, "tonight_minutes", lambda _c, _d: 20 * 60 + 5)
|
|
assert _poke_credit.current_shutdown() == "20:05"
|
|
|
|
def test_unreadable_config_is_none_and_warned(
|
|
self, monkeypatch: pytest.MonkeyPatch, caplog: pytest.LogCaptureFixture
|
|
) -> None:
|
|
"""An unreadable config gives a None shutdown and a shutdown=null warning."""
|
|
monkeypatch.setattr(_poke_credit, "read_shutdown_config", lambda _p: None)
|
|
with caplog.at_level(logging.WARNING):
|
|
assert _poke_credit.current_shutdown() is None
|
|
assert "shutdown=null" in caplog.text
|
|
|
|
def test_current_gaming_minutes_delegates(
|
|
self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
"""current_gaming_minutes returns what gaming_minutes_today reports for the log."""
|
|
seen: list[Path] = []
|
|
|
|
def minutes(log: Path) -> int:
|
|
"""Record the log path asked about and return 90 minutes."""
|
|
seen.append(log)
|
|
return 90
|
|
|
|
monkeypatch.setattr(_poke_credit, "gaming_minutes_today", minutes)
|
|
assert _poke_credit.current_gaming_minutes(tmp_path / "log.json") == 90
|
|
assert seen == [tmp_path / "log.json"]
|