screen-locker/screen_locker/tests/test_compliance_state_part2.py
Krzysztof kuhy Rudnicki 9ab7c09c79
Some checks are pending
dependency freshness / dependencies on newest stable (push) Waiting to run
md-naming / markdown naming convention (push) Waiting to run
pre-commit / pre-commit (push) Waiting to run
Tests / test (push) Waiting to run
Tests / lock-popup-safety (push) Waiting to run
release-apk / release (APK) (push) Waiting to run
The early-bird window is the wake-alarm carrot, not a clock
05:00-08:30 (09:00 with the extended reward) locked nobody but did hold the
workout lock off on a clock that knew nothing about whether the user was
up, next to the morning-session rung that did. EarlyBirdMixin._is_early_bird_time
is now that rung's verdict: open before the alarm, during the session, and
until 11:00 once it completed in time -- exactly what wake-alarm signs. The
wake_alarm_skip rung banks the pending marker so the carrot's end still tries
a phone/RunnerUp workout before locking; early-bird-workout-check.timer moves
to 09:30:30 and 11:00:30, the two possible ends. The consumer's boot-retry
window starts at 05:00 with the refresher.

Also: the sick screen lists four past justifications, one line each -- six
full-length ones pushed it 7 px past 1366x768, which the screens-fit gate
only sees with a real history on disk. Takes @types/node 26.6.2, eslint 10.11.0.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01XTdfXjbGYYrY7Quxcfz6mg
2026-09-20 08:39:44 +02:00

220 lines
7.9 KiB
Python

"""Tests for _compliance_state.explain_lock_decision's full branch trace."""
from __future__ import annotations
from datetime import UTC, datetime
import json
from typing import TYPE_CHECKING
from unittest.mock import patch
import freedays
from screen_locker._compliance_state import explain_lock_decision
from screen_locker._sick_tracker import SickHistory
if TYPE_CHECKING:
from pathlib import Path
def _today() -> str:
return datetime.now(tz=UTC).strftime("%Y-%m-%d")
class TestExplainLockDecision:
"""Full branch trace of the read-only lock-decision chain."""
def _files(self, tmp_path: Path) -> dict[str, Path]:
return {
"log_file": tmp_path / "log.json",
"early_bird_pending_file": tmp_path / "early_bird_pending.json",
}
def test_scheduled_skip_short_circuits(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
freedays.mark(
freedays.today(),
paths=freedays.Paths.under(tmp_path / "freedays"),
)
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
)
assert result.fired is False
assert result.stage == "scheduled_skip"
assert result.trace[0].fired is True
def test_early_bird_window_still_open_skips(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
files["early_bird_pending_file"].write_text(
json.dumps({"date": _today(), "hmac": "sig"})
)
now = datetime.now(tz=UTC).astimezone().replace(hour=6, minute=0)
with patch(
"screen_locker._compliance_predicates.verify_entry_hmac", return_value=True
):
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
wake_skip=True,
now=now,
)
assert result.fired is False
assert result.stage == "early_bird_window_open"
def test_expired_early_bird_falls_through_to_full_lock(
self, tmp_path: Path
) -> None:
files = self._files(tmp_path)
files["early_bird_pending_file"].write_text(
json.dumps({"date": _today(), "hmac": "sig"})
)
now = datetime.now(tz=UTC).astimezone().replace(hour=10, minute=0)
with patch(
"screen_locker._compliance_predicates.verify_entry_hmac", return_value=True
):
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
now=now,
)
assert result.fired is True
assert result.stage == "would_lock"
assert result.auto_upgrade.via == "early_bird_expired"
expired_step = next(
t for t in result.trace if t.name == "early_bird_pending_expired"
)
assert expired_step.fired is True
def test_sick_day_skips(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
result = explain_lock_decision(
**files,
sick_history=SickHistory(sick_days=[_today()]),
weekly_minimum_met=False,
relaxed_day=False,
now=datetime.now(tz=UTC).astimezone().replace(hour=12, minute=0),
)
assert result.fired is False
assert result.stage == "sick_day"
def test_already_logged_skips(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
files["log_file"].write_text(json.dumps({_today(): {"hmac": "sig"}}))
with patch(
"screen_locker._compliance_predicates.verify_entry_hmac", return_value=True
):
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
now=datetime.now(tz=UTC).astimezone().replace(hour=12, minute=0),
)
assert result.fired is False
assert result.stage == "already_logged"
def test_wake_alarm_skip_skips(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
wake_skip=True,
now=datetime.now(tz=UTC).astimezone().replace(hour=12, minute=0),
)
assert result.fired is False
assert result.stage == "wake_alarm_skip"
def test_carrot_in_force_skips_as_wake_alarm(self, tmp_path: Path) -> None:
"""Before the alarm, during the session, until 11:00 -- one stage, one reason."""
files = self._files(tmp_path)
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
wake_skip=True,
)
assert result.fired is False
assert result.stage == "wake_alarm_skip"
assert "early-bird" not in result.reason
def test_relaxed_day_skips(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=True,
now=datetime.now(tz=UTC).astimezone().replace(hour=12, minute=0),
)
assert result.fired is False
assert result.stage == "relaxed_day"
def test_relaxed_day_already_skipped_short_circuits_before_relaxed_day(
self, tmp_path: Path
) -> None:
"""A same-day relaxed_day_skip entry reports its own accurate stage.
Without this, the trace would report the generic "relaxed_day" stage
even after the user already dismissed today's prompt, which is
misleading — the popup is not about to be shown again.
"""
from screen_locker._log_mixin import write_signed_entry
files = self._files(tmp_path)
write_signed_entry(files["log_file"], _today(), {"type": "relaxed_day_skip"})
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=True,
now=datetime.now(tz=UTC).astimezone().replace(hour=12, minute=0),
)
assert result.fired is False
assert result.stage == "relaxed_day_already_skipped"
def test_weekly_minimum_met_skips(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=True,
relaxed_day=False,
now=datetime.now(tz=UTC).astimezone().replace(hour=12, minute=0),
)
assert result.fired is False
assert result.stage == "weekly_minimum_met"
def test_full_lock_when_nothing_applies(self, tmp_path: Path) -> None:
files = self._files(tmp_path)
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
now=datetime.now(tz=UTC).astimezone().replace(hour=12, minute=0),
)
assert result.fired is True
assert result.stage == "would_lock"
assert result.auto_upgrade.via == "none"
assert result.heat_skip_evaluated is False
def test_defaults_now_to_current_time(self, tmp_path: Path) -> None:
"""Covers the ``now is None`` branch — just needs to not raise."""
files = self._files(tmp_path)
result = explain_lock_decision(
**files,
sick_history=SickHistory(),
weekly_minimum_met=False,
relaxed_day=False,
)
assert isinstance(result.fired, bool)