mirror of
https://github.com/kuhyx/wake-alarm.git
synced 2026-11-10 18:07:05 +01:00
Krzysztof kuhy Rudnicki
c3c129b068
On 2026-09-19 the phone was powered off after the bathroom tag and nothing on the PC noticed: every finished session was published and nothing read it. Now wake-alarm distils the phone's morning into an HMAC-signed ~/.local/state/wake_alarm/morning_session.json (wake-alarm-session.timer, 07:00-11:00 every 2 min) that screen-locker and leetcode-guard read to defer their locks -- provisionally until min(start + 90 min, 09:30) while the session is live, until 11:00 once it completed inside that cutoff, and not at all for a session finished later (the phone switched on at noon completes the legs then; that must un-lock nothing). The phone publishes the session at start as well as at finish (current_session, cleared with the finish stamp), so the PC can tell "started, not finished" from "never started". A required 1-5 energy tap joins the weigh-in and ships as latest_energy with a PC reader mirroring _weight (the shared half is _field_reader). Also fixes the vmbox harness, red since the 2026-09-11 home-tidy: the guest tree lives at ~/wake-alarm, and the INSTALL-OK grep matched vm's echo of the command line. Takes Flutter 3.47.5. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Tp54ewdXRLaimUuT8LgYyn
222 lines
7.8 KiB
Python
222 lines
7.8 KiB
Python
"""The signed morning-session file the lock consumers read."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime, timedelta
|
|
import json
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING
|
|
from unittest.mock import patch
|
|
|
|
from wake_alarm import _session
|
|
from wake_alarm._session import (
|
|
describe,
|
|
exempt_until,
|
|
main,
|
|
read_entry,
|
|
refresh,
|
|
to_entry,
|
|
write_entry,
|
|
)
|
|
from wake_alarm._session_payload import SessionRecord
|
|
|
|
if TYPE_CHECKING:
|
|
import pytest
|
|
|
|
NOW = datetime(2026, 9, 19, 8, 30).astimezone()
|
|
|
|
|
|
def _record(
|
|
outcome: str = "completed",
|
|
started: str = "2026-09-19T08:01:00.000",
|
|
finished: str | None = "2026-09-19T08:40:00.000",
|
|
date: str = "2026-09-19",
|
|
) -> SessionRecord:
|
|
return SessionRecord(
|
|
date=date,
|
|
outcome=outcome,
|
|
started_at=started,
|
|
finished_at=finished,
|
|
energy=3,
|
|
weight_kg=73.8,
|
|
re_rings=0,
|
|
device_id="p",
|
|
published_at=1.0,
|
|
)
|
|
|
|
|
|
def _local(iso: str) -> datetime:
|
|
return datetime.fromisoformat(iso).astimezone()
|
|
|
|
|
|
class TestExemptUntil:
|
|
"""The policy, signed in once so no consumer re-derives it."""
|
|
|
|
def test_completed_in_time_earns_eleven(self) -> None:
|
|
assert exempt_until(_record()) == _local("2026-09-19T11:00:00")
|
|
|
|
def test_live_is_start_plus_grace_capped(self) -> None:
|
|
assert exempt_until(_record("in_progress", finished=None)) == _local(
|
|
"2026-09-19T09:30:00"
|
|
)
|
|
early = _record("in_progress", started="2026-09-19T06:00:00.000", finished=None)
|
|
assert exempt_until(early) == _local("2026-09-19T07:30:00")
|
|
|
|
def test_completed_after_the_cutoff_earns_nothing(self) -> None:
|
|
"""A phone switched on at noon completes the legs then; no carrot."""
|
|
assert exempt_until(_record(finished="2026-09-19T12:33:00.000")) is None
|
|
# An aborted 06:00 alarm "completed" at 10:00 locks no later than today.
|
|
late_early = _record(
|
|
started="2026-09-19T06:05:00.000", finished="2026-09-19T10:00:00.000"
|
|
)
|
|
assert exempt_until(late_early) is None
|
|
|
|
def test_failed_and_unreadable_instants_earn_nothing(self) -> None:
|
|
assert exempt_until(_record("failed")) is None
|
|
assert exempt_until(_record(started="when?")) is None
|
|
assert exempt_until(_record(finished=None)) is None
|
|
assert exempt_until(_record(finished="later")) is None
|
|
|
|
|
|
class TestEntryFile:
|
|
"""Writing, signing, reading back."""
|
|
|
|
def test_round_trip_is_verified(self) -> None:
|
|
entry = to_entry(_record(), today="2026-09-19", refreshed_at=5.0)
|
|
assert entry["exempt_until"] == _local("2026-09-19T11:00:00").isoformat()
|
|
assert write_entry(entry)
|
|
assert read_entry() == entry
|
|
assert "hmac" in entry
|
|
|
|
def test_none_heartbeat_and_failed_have_no_exemption(self) -> None:
|
|
entry = to_entry(None, today="2026-09-19", refreshed_at=5.0)
|
|
assert entry["outcome"] == "none"
|
|
assert "exempt_until" not in entry
|
|
failed = to_entry(_record("failed"), today="2026-09-19", refreshed_at=5.0)
|
|
assert failed["outcome"] == "failed"
|
|
assert "exempt_until" not in failed
|
|
|
|
def test_tampering_is_rejected(self, tmp_path: Path) -> None:
|
|
write_entry(to_entry(_record(), today="2026-09-19", refreshed_at=5.0))
|
|
path = _session.MORNING_SESSION_FILE
|
|
tampered = json.loads(path.read_text())
|
|
tampered["exempt_until"] = _local("2026-09-19T23:00:00").isoformat()
|
|
path.write_text(json.dumps(tampered))
|
|
assert read_entry() is None
|
|
path.write_text("[]")
|
|
assert read_entry() is None
|
|
path.write_text("{")
|
|
assert read_entry() is None
|
|
path.unlink()
|
|
assert read_entry() is None
|
|
assert (tmp_path / "morning_session.json") == path
|
|
|
|
def test_unsigned_when_the_key_is_missing(self) -> None:
|
|
with patch("wake_alarm._session.compute_entry_hmac", return_value=None):
|
|
entry = to_entry(None, today="2026-09-19", refreshed_at=5.0)
|
|
assert write_entry(entry)
|
|
assert "hmac" not in entry
|
|
|
|
def test_write_failure_is_false(self) -> None:
|
|
with patch.object(Path, "write_text", side_effect=OSError("ro")):
|
|
assert not write_entry(to_entry(None, today="2026-09-19", refreshed_at=5.0))
|
|
|
|
|
|
class TestRefresh:
|
|
"""Fetch, select, write."""
|
|
|
|
def _logs(self, record_json: dict[str, object]) -> list[tuple[str, str]]:
|
|
raw = json.dumps(
|
|
{
|
|
"alarm": {
|
|
"fields": {
|
|
"morning_sessions": [
|
|
[record_json],
|
|
"2026-09-19T06:20:00.000Z-0000-p",
|
|
]
|
|
}
|
|
}
|
|
}
|
|
)
|
|
return [("p", raw)]
|
|
|
|
def test_todays_session_is_written(self) -> None:
|
|
session = {
|
|
"date": "2026-09-19",
|
|
"started_at": "2026-09-19T08:01:00.000",
|
|
"updated_at": "2026-09-19T08:40:00.000",
|
|
"outcome": "completed",
|
|
}
|
|
with patch(
|
|
"wake_alarm._session.fetch_firebase_logs", return_value=self._logs(session)
|
|
):
|
|
entry = refresh(NOW)
|
|
assert entry is not None
|
|
assert entry["outcome"] == "completed"
|
|
assert read_entry() == entry
|
|
|
|
def test_yesterdays_session_becomes_none(self) -> None:
|
|
session = {
|
|
"date": "2026-09-18",
|
|
"started_at": "2026-09-18T08:01:00.000",
|
|
"outcome": "completed",
|
|
}
|
|
with patch(
|
|
"wake_alarm._session.fetch_firebase_logs", return_value=self._logs(session)
|
|
):
|
|
entry = refresh(NOW)
|
|
assert entry is not None
|
|
assert (entry["date"], entry["outcome"]) == ("2026-09-19", "none")
|
|
|
|
def test_unreadable_backend_leaves_the_file_alone(self) -> None:
|
|
write_entry(to_entry(_record(), today="2026-09-19", refreshed_at=5.0))
|
|
with patch("wake_alarm._session.fetch_firebase_logs", return_value=None):
|
|
assert refresh(NOW) is None
|
|
assert read_entry() is not None
|
|
|
|
def test_write_failure_is_none(self) -> None:
|
|
with (
|
|
patch("wake_alarm._session.fetch_firebase_logs", return_value=[]),
|
|
patch("wake_alarm._session.write_entry", return_value=False),
|
|
):
|
|
assert refresh(NOW) is None
|
|
|
|
|
|
class TestCli:
|
|
"""--show and --refresh, and the verdict line."""
|
|
|
|
def test_describe_covers_every_verdict(self) -> None:
|
|
entry = to_entry(_record(), today="2026-09-19", refreshed_at=5.0)
|
|
assert "exempt until" in describe(entry, NOW)
|
|
assert "exemption ended" in describe(entry, NOW + timedelta(hours=4))
|
|
assert "no exemption" in describe(
|
|
to_entry(None, today="x", refreshed_at=5.0), NOW
|
|
)
|
|
|
|
def test_show_prints_the_verified_entry(
|
|
self, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
write_entry(to_entry(_record(), today="2026-09-19", refreshed_at=5.0))
|
|
assert main(["--show"]) == 0
|
|
out = capsys.readouterr().out
|
|
assert '"outcome": "completed"' in out
|
|
assert "2026-09-19 completed" in out
|
|
|
|
def test_show_without_a_file_exits_1(
|
|
self, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
assert main(["--show"]) == 1
|
|
assert "no verified" in capsys.readouterr().out
|
|
|
|
def test_refresh_uses_the_backend(self) -> None:
|
|
with patch("wake_alarm._session.fetch_firebase_logs", return_value=[]) as fetch:
|
|
assert main(["--refresh"]) == 0
|
|
fetch.assert_called_once()
|
|
entry = read_entry()
|
|
assert entry is not None
|
|
assert entry["outcome"] == "none"
|
|
|
|
def test_argv_defaults_to_sys_argv(self) -> None:
|
|
with patch("wake_alarm._session.sys.argv", ["prog", "--show"]):
|
|
assert main() == 1
|