mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 18:13: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
67 lines
2.9 KiB
Python
67 lines
2.9 KiB
Python
"""The wake-alarm carrot in the lock's startup ladder (``_check_today_state_exits``)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from screen_locker._morning_session import MorningSkip
|
|
from screen_locker.screen_lock import ScreenLocker
|
|
|
|
_SKIP = MorningSkip(outcome="completed", exempt_until=datetime(2099, 1, 1).astimezone())
|
|
|
|
|
|
def _locker(
|
|
tmp_path: Path, *, carrot: MorningSkip | None, has_logged: bool = False
|
|
) -> None:
|
|
"""Build a locker whose early-bird window is the given carrot verdict."""
|
|
with (
|
|
patch.object(Path, "resolve", return_value=tmp_path / "screen_locker"),
|
|
patch.object(ScreenLocker, "has_logged_today", return_value=has_logged),
|
|
patch.object(ScreenLocker, "_is_sick_day_today", return_value=False),
|
|
patch.object(ScreenLocker, "_is_early_bird_pending", return_value=False),
|
|
patch("screen_locker._early_bird.morning_skip_today", return_value=carrot),
|
|
patch.object(ScreenLocker, "_try_auto_upgrade_early_bird", return_value=False),
|
|
patch.object(ScreenLocker, "_start_phone_check"),
|
|
patch.object(ScreenLocker, "_start_relaxed_day_flow"),
|
|
patch.object(ScreenLocker, "_scan_and_fill_week_runnerup", return_value=0),
|
|
):
|
|
ScreenLocker(demo_mode=True)
|
|
|
|
|
|
class TestWakeSkipIntegration:
|
|
"""How the startup ladder reacts to the wake-alarm carrot."""
|
|
|
|
def test_exits_and_banks_the_marker_while_the_carrot_is_in_force(
|
|
self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path
|
|
) -> None:
|
|
"""While the carrot is in force the lock exits, banks the pending marker and records why."""
|
|
with (
|
|
patch.object(ScreenLocker, "_save_early_bird_pending") as save,
|
|
patch("screen_locker._auto_upgrade.record_decision") as record,
|
|
):
|
|
_locker(tmp_path, carrot=_SKIP)
|
|
mock_sys_exit.assert_called_once_with(0)
|
|
save.assert_called_once()
|
|
decision = record.call_args.args[0]
|
|
assert decision.reason == "wake_alarm_skip"
|
|
assert decision.extra["exempt_until"] == _SKIP.exempt_until.isoformat()
|
|
assert "no lock until" in decision.detail
|
|
|
|
def test_does_not_exit_without_a_carrot(
|
|
self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path
|
|
) -> None:
|
|
"""Without a carrot the startup ladder does not exit."""
|
|
_locker(tmp_path, carrot=None)
|
|
mock_sys_exit.assert_not_called()
|
|
|
|
def test_logged_today_takes_precedence(
|
|
self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path
|
|
) -> None:
|
|
"""A workout already logged today wins over the carrot as the exit reason."""
|
|
with patch("screen_locker._auto_upgrade.record_decision") as record:
|
|
_locker(tmp_path, carrot=_SKIP, has_logged=True)
|
|
mock_sys_exit.assert_called_once_with(0)
|
|
assert record.call_args.args[0].reason == "workout_logged_today"
|