wake-alarm/wake_alarm/tests/test_trigger.py

"""Tests for _trigger.py — the per-minute check-and-fire tick."""

from __future__ import annotations

from datetime import datetime, timezone
from unittest.mock import MagicMock, patch

import pytest

from wake_alarm._sync import Alarm
from wake_alarm._trigger import (
    _decide,
    _fire_window,
    _now,
    _parse_args,
    _skip_reason,
    _start_target_unit,
    main,
)


def _at(hour: int, minute: int) -> datetime:
    """Return a Sunday at the given local wall-clock time, timezone-aware.

    Built the same way production does — naive wall clock, then astimezone()
    to attach the system offset — so the tests exercise the real comparison
    semantics rather than a hand-picked fixed offset.
    """
    return datetime(2026, 7, 19, hour, minute, tzinfo=None).astimezone()


@pytest.fixture(autouse=True)
def _ungated() -> None:
    """Open both "already handled today" gates unless a test closes them."""
    with (
        patch("wake_alarm._trigger.was_alarm_dismissed_today", return_value=False),
        patch("wake_alarm._trigger.was_workout_logged_today", return_value=False),
    ):
        yield


class TestNow:
    """Tests for the _now seam."""

    def test_returns_aware_local_time(self) -> None:
        """The clock is timezone-aware and in local time, not UTC."""
        now = _now()
        assert now.tzinfo is not None
        # Local, not UTC: the wall-clock hour must match the system clock's.
        assert now.hour == datetime.now(tz=timezone.utc).astimezone().hour


class TestFireWindow:
    """Tests for _fire_window."""

    def test_window_starts_at_the_alarm_minute(self) -> None:
        """The fire instant is today at the alarm's H:M, seconds zeroed."""
        fire_at, _ = _fire_window(Alarm(6, 45, "remote"), _at(3, 12))
        assert fire_at == _at(6, 45)
        assert fire_at.second == 0

    def test_window_is_the_catchup_length(self) -> None:
        """The window ends TRIGGER_CATCHUP_MINUTES after the alarm."""
        fire_at, end = _fire_window(Alarm(6, 45, "remote"), _at(3, 12))
        assert (end - fire_at).total_seconds() == 60 * 60

    def test_fire_instant_is_aware(self) -> None:
        """The fire instant keeps a timezone, so comparisons are unambiguous."""
        fire_at, end = _fire_window(Alarm(6, 45, "remote"), _at(3, 12))
        assert fire_at.tzinfo is not None
        assert end.tzinfo is not None

    def test_offset_is_derived_from_the_target_wall_time(self) -> None:
        """The offset is re-derived, not inherited from *now*.

        Guards the DST case: deciding a morning alarm from a tick taken on the
        other side of a clock change must not carry *now*'s stale UTC offset.
        """
        fire_at, _ = _fire_window(Alarm(6, 45, "remote"), _at(3, 12))
        expected = datetime(2026, 7, 19, 6, 45, tzinfo=None).astimezone()
        assert fire_at.utcoffset() == expected.utcoffset()
        assert fire_at == expected


class TestSkipReason:
    """Tests for the clock-independent gates."""

    def test_open_when_nothing_blocks(self) -> None:
        """With every gate open there is no reason to skip."""
        assert _skip_reason(_at(7, 0)) is None

    def test_not_an_alarm_day(self) -> None:
        """A weekday outside ALARM_DAYS blocks firing."""
        with patch("wake_alarm._trigger.ALARM_DAYS", frozenset()):
            assert _skip_reason(_at(7, 0)) == "not an alarm day"

    def test_already_dismissed(self) -> None:
        """A dismissal recorded today blocks a re-fire."""
        with patch(
            "wake_alarm._trigger.was_alarm_dismissed_today",
            return_value=True,
        ):
            assert _skip_reason(_at(7, 0)) == "alarm already dismissed today"

    def test_workout_already_logged(self) -> None:
        """A logged workout means the alarm is no longer needed."""
        with patch("wake_alarm._trigger.was_workout_logged_today", return_value=True):
            assert _skip_reason(_at(7, 0)) == "workout already logged today"


class TestDecide:
    """Tests for the fire/don't-fire decision."""

    def test_before_the_alarm(self) -> None:
        """A tick before the alarm minute does not fire."""
        fire, reason = _decide(_at(6, 44), Alarm(6, 45, "remote"))
        assert not fire
        assert "still ahead" in reason

    def test_exactly_on_the_alarm_minute(self) -> None:
        """The alarm minute itself fires."""
        fire, _ = _decide(_at(6, 45), Alarm(6, 45, "remote"))
        assert fire

    def test_within_the_catchup_window(self) -> None:
        """A tick inside the catch-up window still fires (e.g. after resume)."""
        fire, _ = _decide(_at(7, 30), Alarm(6, 45, "remote"))
        assert fire

    def test_at_the_window_edge(self) -> None:
        """The window is half-open: its end instant no longer fires."""
        fire, reason = _decide(_at(7, 45), Alarm(6, 45, "remote"))
        assert not fire
        assert "window closed" in reason

    def test_long_after_the_alarm(self) -> None:
        """Hours later, the window is closed."""
        fire, reason = _decide(_at(20, 0), Alarm(6, 45, "remote"))
        assert not fire
        assert "window closed" in reason

    def test_gate_beats_the_clock(self) -> None:
        """A closed gate wins even at the exact alarm minute."""
        with patch(
            "wake_alarm._trigger.was_alarm_dismissed_today",
            return_value=True,
        ):
            fire, reason = _decide(_at(6, 45), Alarm(6, 45, "remote"))
        assert not fire
        assert reason == "alarm already dismissed today"

    def test_alarm_moved_earlier_fires_immediately(self) -> None:
        """A phone alarm moved back before now fires on the very next tick."""
        fire, _ = _decide(_at(6, 50), Alarm(6, 40, "remote"))
        assert fire


class TestStartTargetUnit:
    """Tests for the systemctl invocation."""

    def test_starts_the_orchestrator_not_the_alarm(self) -> None:
        """The trigger must start morning-routine, never wake-alarm."""
        completed = MagicMock(returncode=0)
        with patch("subprocess.run", return_value=completed) as run:
            assert _start_target_unit()
        assert run.call_args[0][0] == [
            "systemctl",
            "--user",
            "start",
            "morning-routine.service",
        ]

    def test_nonzero_exit_is_reported(self) -> None:
        """A systemctl failure returns False rather than raising."""
        completed = MagicMock(returncode=1, stderr=b"no such unit")
        with patch("subprocess.run", return_value=completed):
            assert not _start_target_unit()

    def test_missing_systemctl_is_swallowed(self) -> None:
        """An OSError (no systemctl) returns False rather than raising."""
        with patch("subprocess.run", side_effect=OSError("not found")):
            assert not _start_target_unit()

    def test_does_not_raise_on_failure(self) -> None:
        """check=False: a failing start never raises CalledProcessError."""
        completed = MagicMock(returncode=1, stderr=b"")
        with patch("subprocess.run", return_value=completed) as run:
            _start_target_unit()
        assert run.call_args.kwargs["check"] is False


class TestMain:
    """Tests for the tick entry point."""

    def test_parse_args_default(self) -> None:
        """--dry-run defaults to off."""
        assert not _parse_args([]).dry_run

    def _run_main(self, argv: list[str], now: datetime, alarm: Alarm) -> MagicMock:
        with (
            patch("sys.argv", ["_trigger", *argv]),
            patch("wake_alarm._trigger._now", return_value=now),
            patch("wake_alarm._trigger.get_alarm", return_value=alarm),
            patch("wake_alarm._trigger._start_target_unit") as start,
        ):
            main()
        return start

    def test_fires_at_the_alarm_minute(self) -> None:
        """On the alarm minute the target unit is started."""
        start = self._run_main([], _at(6, 45), Alarm(6, 45, "remote"))
        start.assert_called_once()

    def test_does_not_fire_early(self) -> None:
        """Before the alarm nothing is started."""
        start = self._run_main([], _at(5, 0), Alarm(6, 45, "remote"))
        start.assert_not_called()

    def test_dry_run_never_starts_anything(
        self, capsys: pytest.CaptureFixture[str]
    ) -> None:
        """--dry-run reports WOULD FIRE without touching systemd."""
        start = self._run_main(["--dry-run"], _at(6, 45), Alarm(6, 45, "remote"))
        start.assert_not_called()
        assert "WOULD FIRE" in capsys.readouterr().out

    def test_dry_run_reports_not_firing(
        self, capsys: pytest.CaptureFixture[str]
    ) -> None:
        """--dry-run explains why it is not firing."""
        self._run_main(["--dry-run"], _at(5, 0), Alarm(6, 45, "remote"))
        out = capsys.readouterr().out
        assert "not firing" in out
        assert "still ahead" in out

    @pytest.mark.parametrize("source", ["fallback", "stale-cache"])
    def test_warns_when_firing_on_a_degraded_source(
        self, caplog: pytest.LogCaptureFixture, source: str
    ) -> None:
        """Firing on a non-phone alarm time is logged loudly."""
        with caplog.at_level("WARNING"):
            self._run_main([], _at(6, 45), Alarm(6, 45, source))
        assert "degraded alarm source" in caplog.text

    @pytest.mark.parametrize("source", ["remote", "cache"])
    def test_quiet_when_firing_on_the_phone_alarm(
        self, caplog: pytest.LogCaptureFixture, source: str
    ) -> None:
        """A normal fire does not emit the degraded-source warning."""
        with caplog.at_level("WARNING"):
            self._run_main([], _at(6, 45), Alarm(6, 45, source))
        assert "degraded alarm source" not in caplog.text