screen-locker/screen_locker/tests/test_shutdown_base.py

"""Tests for _shutdown_base module (daily shutdown base reset)."""

from __future__ import annotations

from contextlib import nullcontext
from datetime import UTC, datetime
import json
from typing import TYPE_CHECKING
from unittest.mock import MagicMock, patch

from screen_locker._shutdown_base import (
    _load_state,
    base_minutes,
    reset_to_base_if_new_day,
)

if TYPE_CHECKING:
    from pathlib import Path


class TestLoadState:
    """Tests for the state-file reader behind both entry points."""

    def test_missing_file_is_empty(self, tmp_path: Path) -> None:
        """Missing file → {} without errors."""
        assert _load_state(tmp_path / "nonexistent.json") == {}

    def test_returns_stored_stamps(self, tmp_path: Path) -> None:
        """Valid file → exact contents."""
        f = tmp_path / "state.json"
        f.write_text(json.dumps({"last_reset_date": "2000-01-01"}))
        assert _load_state(f) == {"last_reset_date": "2000-01-01"}

    def test_corrupt_json_is_empty(self, tmp_path: Path) -> None:
        """Corrupt JSON → {} via except."""
        f = tmp_path / "state.json"
        f.write_text("not-json")
        assert _load_state(f) == {}

    def test_oserror_is_empty(self) -> None:
        """OSError on open → {} via except."""
        mock_path = MagicMock()
        mock_path.exists.return_value = True
        mock_path.open.side_effect = OSError("read fail")
        assert _load_state(mock_path) == {}

    def test_stale_base_keys_are_ignored(self, tmp_path: Path) -> None:
        """A file installed before the base became a constant must not revive 21."""
        f = tmp_path / "state.json"
        f.write_text(
            json.dumps(
                {
                    "last_reset_date": "2000-01-01",
                    "base_mon_wed_hour": 21,
                    "base_thu_sun_hour": 21,
                }
            )
        )
        mixin = MagicMock()
        mixin._read_shutdown_config.return_value = (21 * 60, 21 * 60, 300)
        mixin._write_shutdown_config.return_value = True
        assert reset_to_base_if_new_day(f, mixin) is True
        mixin._write_shutdown_config.assert_called_once_with(
            base_minutes(), base_minutes(), 300, restore=True
        )
        assert "base_mon_wed_hour" not in json.loads(f.read_text())


class TestResetToBaseIfNewDay:
    """Tests for reset_to_base_if_new_day."""

    def _make_mixin(self, write_ok: bool = True) -> MagicMock:
        """Build a minimal mixin mock."""
        mixin = MagicMock()
        mixin._read_shutdown_config.return_value = (21 * 60, 21 * 60, 300)
        mixin._write_shutdown_config.return_value = write_ok
        return mixin

    def test_returns_false_when_already_reset_today(self, tmp_path: Path) -> None:
        """Same-day last_reset_date → early False return (line 63)."""
        today = datetime.now(tz=UTC).strftime("%Y-%m-%d")
        f = tmp_path / "state.json"
        f.write_text(json.dumps({"last_reset_date": today}))
        assert reset_to_base_if_new_day(f, self._make_mixin()) is False

    def test_resets_when_new_day(self, tmp_path: Path) -> None:
        """Different last_reset_date → reset performed, returns True (lines 67-100)."""
        f = tmp_path / "state.json"
        f.write_text(json.dumps({"last_reset_date": "2000-01-01"}))
        mixin = self._make_mixin()
        assert reset_to_base_if_new_day(f, mixin) is True
        base = base_minutes()
        mixin._write_shutdown_config.assert_called_once_with(
            base, base, 300, restore=True
        )

    def test_resets_when_no_state_file(self, tmp_path: Path) -> None:
        """No state file → treated as new day, reset performed (lines 67-100)."""
        f = tmp_path / "nonexistent.json"
        mixin = self._make_mixin()
        assert reset_to_base_if_new_day(f, mixin) is True
        mixin._write_shutdown_config.assert_called_once()

    def test_returns_false_when_write_config_fails(self, tmp_path: Path) -> None:
        """_write_shutdown_config returns False → reset fails (lines 74-76)."""
        f = tmp_path / "state.json"
        f.write_text(json.dumps({"last_reset_date": "2000-01-01"}))
        mixin = self._make_mixin(write_ok=False)
        assert reset_to_base_if_new_day(f, mixin) is False

    def test_uses_default_morning_end_when_config_is_none(self, tmp_path: Path) -> None:
        """_read_shutdown_config returns None → morning_end defaults to 05:00 (300 minutes)."""
        f = tmp_path / "state.json"
        f.write_text(json.dumps({"last_reset_date": "2000-01-01"}))
        mixin = MagicMock()
        mixin._read_shutdown_config.return_value = None
        mixin._write_shutdown_config.return_value = True
        reset_to_base_if_new_day(f, mixin)
        base = base_minutes()
        mixin._write_shutdown_config.assert_called_once_with(
            base, base, 300, restore=True
        )

    def test_handles_corrupt_state_file_gracefully(self, tmp_path: Path) -> None:
        """Corrupt state file treated as no date → reset runs (lines 64-65 except branch)."""
        f = tmp_path / "state.json"
        f.write_text("not-json")
        mixin = self._make_mixin()
        assert reset_to_base_if_new_day(f, mixin) is True

    def test_handles_oserror_on_state_write(self, tmp_path: Path) -> None:
        """An OSError writing the new state file is caught; the reset still returns True."""
        # Use a mock path that fails only on "w" opens.
        state_mock = MagicMock()
        state_mock.exists.return_value = False  # triggers fresh-reset path
        state_mock.open.side_effect = OSError("disk full")
        mixin = self._make_mixin()
        # _write_shutdown_config succeeds, so True is returned even if state write fails.
        # The lock file would come from the same failing mock; lock elsewhere.
        with patch(
            "screen_locker._shutdown_base.bonus_lock", return_value=nullcontext()
        ):
            assert reset_to_base_if_new_day(state_mock, mixin) is True