mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 16:53:58 +01:00
Krzysztof kuhy Rudnicki
5c27769642
Some checks are pending
pre-commit / pre-commit (push) Waiting to run
Tests / test (3.10) (push) Waiting to run
Tests / test (3.11) (push) Waiting to run
Tests / test (3.12) (push) Waiting to run
Tests / lock-popup-safety (push) Waiting to run
release-apk / release (APK) (push) Waiting to run
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
"""Tests for post-sick-day workout verification (--verify-workout)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
from unittest.mock import MagicMock
|
|
|
|
from screen_locker.tests.conftest import create_locker
|
|
|
|
if TYPE_CHECKING:
|
|
from pathlib import Path
|
|
|
|
|
|
class TestHandleVerifyWorkoutResult:
|
|
"""Tests for _handle_verify_workout_result."""
|
|
|
|
def test_verified_adjusts_shutdown_and_saves(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""On verified: adjust shutdown, save log, show success."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
locker.log_file = tmp_path / "workout_log.json"
|
|
object.__setattr__(
|
|
locker,
|
|
"_adjust_shutdown_time_later",
|
|
MagicMock(return_value=True),
|
|
)
|
|
|
|
locker._handle_verify_workout_result("verified", "1 session found")
|
|
|
|
assert locker.workout_data["type"] == "phone_verified"
|
|
assert locker.workout_data["after_sick_day"] == "true"
|
|
locker._adjust_shutdown_time_later.assert_called_once()
|
|
locker.root.after.assert_called()
|
|
|
|
def test_verified_without_adjustment(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""On verified but adjustment fails: still saves and shows success."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
locker.log_file = tmp_path / "workout_log.json"
|
|
object.__setattr__(
|
|
locker,
|
|
"_adjust_shutdown_time_later",
|
|
MagicMock(return_value=False),
|
|
)
|
|
|
|
locker._handle_verify_workout_result("verified", "1 session found")
|
|
|
|
assert locker.workout_data["type"] == "phone_verified"
|
|
locker.root.after.assert_called()
|
|
|
|
def test_not_verified_shows_retry(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""On not_verified: show retry screen."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
object.__setattr__(
|
|
locker,
|
|
"_show_verify_retry",
|
|
MagicMock(),
|
|
)
|
|
|
|
locker._handle_verify_workout_result(
|
|
"not_verified",
|
|
"No workout today",
|
|
)
|
|
|
|
locker._show_verify_retry.assert_called_once_with(
|
|
"No workout today",
|
|
)
|
|
|
|
def test_error_shows_retry(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""On error: show retry screen."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
object.__setattr__(
|
|
locker,
|
|
"_show_verify_retry",
|
|
MagicMock(),
|
|
)
|
|
|
|
locker._handle_verify_workout_result("error", "ADB failed")
|
|
|
|
locker._show_verify_retry.assert_called_once_with("ADB failed")
|
|
|
|
|
|
class TestShowVerifyRetry:
|
|
"""Tests for _show_verify_retry."""
|
|
|
|
def test_shows_retry_and_close_buttons(
|
|
self,
|
|
mock_tk: MagicMock,
|
|
mock_sys_exit: MagicMock,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Show TRY AGAIN and Close buttons."""
|
|
locker = create_locker(mock_tk, tmp_path)
|
|
|
|
locker._show_verify_retry("No workout found")
|
|
|
|
# Verify container was cleared and buttons were packed
|
|
locker.container.first.winfo_children.return_value = []
|