mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 21:53:54 +01:00
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
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
190 lines
6.7 KiB
Python
190 lines
6.7 KiB
Python
"""Tests for early bird auto-upgrade, has_logged_today, and init flow."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Any
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from screen_locker.screen_lock import ScreenLocker
|
|
from screen_locker.tests.conftest import (
|
|
create_locker,
|
|
create_locker_early_bird,
|
|
)
|
|
|
|
|
|
class TestTryAutoUpgradeEarlyBird:
|
|
"""Tests for _try_auto_upgrade_early_bird method."""
|
|
|
|
def test_upgrade_succeeds_when_verified(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Returns True, saves phone_verified entry, adjusts shutdown."""
|
|
log_file = tmp_path / "log.json"
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
locker.log_file = log_file
|
|
object.__setattr__(
|
|
locker,
|
|
"_verify_phone_workout",
|
|
MagicMock(return_value=("verified", "Workout verified! (67 min)")),
|
|
)
|
|
object.__setattr__(
|
|
locker,
|
|
"_adjust_shutdown_time_later",
|
|
MagicMock(return_value=True),
|
|
)
|
|
with patch(
|
|
"screen_locker._log_mixin.compute_entry_hmac",
|
|
return_value=None,
|
|
):
|
|
result = locker._try_auto_upgrade_early_bird()
|
|
|
|
assert result is True
|
|
with log_file.open() as f:
|
|
data: dict[str, Any] = json.load(f)
|
|
today = datetime.now(tz=UTC).strftime("%Y-%m-%d")
|
|
assert data[today][0]["workout_data"]["type"] == "phone_verified"
|
|
assert data[today][0]["workout_data"]["after_early_bird"] == "true"
|
|
|
|
def test_upgrade_fails_when_not_verified(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Returns False when phone shows no workout."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
object.__setattr__(
|
|
locker,
|
|
"_verify_phone_workout",
|
|
MagicMock(return_value=("no_phone", "No phone connected")),
|
|
)
|
|
assert locker._try_auto_upgrade_early_bird() is False
|
|
|
|
def test_upgrade_fails_on_os_error(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Returns False when _verify_phone_workout raises OSError."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
object.__setattr__(
|
|
locker,
|
|
"_verify_phone_workout",
|
|
MagicMock(side_effect=OSError("adb fail")),
|
|
)
|
|
assert locker._try_auto_upgrade_early_bird() is False
|
|
|
|
def test_upgrade_fails_on_runtime_error(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Returns False when _verify_phone_workout raises RuntimeError."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
object.__setattr__(
|
|
locker,
|
|
"_verify_phone_workout",
|
|
MagicMock(side_effect=RuntimeError("unexpected")),
|
|
)
|
|
assert locker._try_auto_upgrade_early_bird() is False
|
|
|
|
|
|
class TestInitEarlyBirdFlow:
|
|
"""Integration tests for early bird branches in __init__."""
|
|
|
|
def test_init_saves_log_and_exits_during_early_bird_window(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""First login during 5-8:30 window: save early_bird log, exit."""
|
|
mock_sys_exit.side_effect = SystemExit(0)
|
|
with (
|
|
patch.object(Path, "resolve", return_value=tmp_path),
|
|
patch.object(ScreenLocker, "has_logged_today", return_value=False),
|
|
patch.object(ScreenLocker, "_is_sick_day_today", return_value=False),
|
|
patch.object(ScreenLocker, "_is_early_bird_pending", return_value=False),
|
|
patch.object(ScreenLocker, "_is_early_bird_time", return_value=True),
|
|
patch.object(
|
|
ScreenLocker,
|
|
"_try_auto_upgrade_early_bird",
|
|
return_value=False,
|
|
),
|
|
patch.object(ScreenLocker, "_save_early_bird_pending") as mock_save,
|
|
patch.object(ScreenLocker, "_start_phone_check"),
|
|
patch.object(ScreenLocker, "_start_verify_workout_check"),
|
|
patch(
|
|
"screen_locker._early_bird.morning_skip_today",
|
|
return_value=None,
|
|
),
|
|
pytest.raises(SystemExit),
|
|
):
|
|
ScreenLocker(demo_mode=True)
|
|
|
|
mock_save.assert_called_once()
|
|
mock_sys_exit.assert_called_with(0)
|
|
|
|
def test_init_exits_when_early_bird_log_still_in_window(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Early bird log exists and window still active: skip lock, exit."""
|
|
mock_sys_exit.side_effect = SystemExit(0)
|
|
|
|
with pytest.raises(SystemExit):
|
|
create_locker_early_bird(mock_tk, tmp_path, state="log_active")
|
|
|
|
mock_sys_exit.assert_called_with(0)
|
|
|
|
def test_init_exits_when_early_bird_log_upgrades_successfully(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Early bird log + past 8:30 + workout done: upgrade, exit."""
|
|
mock_sys_exit.side_effect = SystemExit(0)
|
|
with (
|
|
patch.object(Path, "resolve", return_value=tmp_path),
|
|
patch.object(ScreenLocker, "has_logged_today", return_value=False),
|
|
patch.object(ScreenLocker, "_is_sick_day_today", return_value=False),
|
|
patch.object(ScreenLocker, "_is_early_bird_pending", return_value=True),
|
|
patch.object(ScreenLocker, "_is_early_bird_time", return_value=False),
|
|
patch.object(
|
|
ScreenLocker, "_try_auto_upgrade_early_bird", return_value=True
|
|
),
|
|
patch.object(ScreenLocker, "_start_phone_check"),
|
|
patch.object(ScreenLocker, "_start_verify_workout_check"),
|
|
pytest.raises(SystemExit),
|
|
):
|
|
ScreenLocker(demo_mode=True)
|
|
|
|
mock_sys_exit.assert_called_with(0)
|
|
|
|
def test_init_shows_lock_when_early_bird_log_no_workout(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Early bird log + past 8:30 + no workout: show lock, no early exit."""
|
|
locker = create_locker_early_bird(mock_tk, tmp_path, state="log_expired")
|
|
|
|
# _try_auto_upgrade_early_bird returns False (default in create_locker)
|
|
# so __init__ falls through to show the lock without calling sys.exit
|
|
mock_sys_exit.assert_not_called()
|
|
assert locker.demo_mode is True
|