"""Tests for sick-budget UI integration, finalize, debt-clear, and dialogs.""" # pylint: disable=protected-access from __future__ import annotations from typing import TYPE_CHECKING from unittest.mock import MagicMock, patch from screen_locker import _sick_tracker from screen_locker._sick_tracker import SickHistory from screen_locker.tests.conftest import create_locker if TYPE_CHECKING: from pathlib import Path # --------------------------------------------------------------------------- # _ui_flows.py — branches added for sick budget + finalize # --------------------------------------------------------------------------- class TestShowRetryAndSickBudget: """Tests for budget-aware _show_retry_and_sick.""" def test_shows_sick_button_when_budget_available( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) with patch.object(_sick_tracker, "load_history", return_value=SickHistory()): locker._show_retry_and_sick("nope") button_texts = { call.args[1] for call in mock_tk.Button.call_args_list if len(call.args) > 1 } # Buttons are created via the helper which sets text via kwarg "text". button_texts |= { call.kwargs.get("text") for call in mock_tk.Button.call_args_list } assert "I'm sick" in button_texts def test_hides_sick_button_when_budget_exhausted( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) full = SickHistory(sick_days=["2026-05-09"] * 99) with ( patch.object(_sick_tracker, "load_history", return_value=full), patch.object(_sick_tracker, "is_budget_exhausted", return_value=True), ): locker._show_retry_and_sick("nope") button_texts: set[str] = set() for call in mock_tk.Button.call_args_list: button_texts.add(call.kwargs.get("text", "")) assert "I'm sick" not in button_texts class TestProceedToSickCountdownLoadsHistory: """Covers the no-cache branch of _proceed_to_sick_countdown.""" def test_loads_history_when_cache_missing( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) object.__setattr__(locker, "clear_container", MagicMock()) object.__setattr__( locker, "_sick_mode_used_today", MagicMock(return_value=False) ) object.__setattr__( locker, "_adjust_shutdown_time_earlier", MagicMock(return_value=True), ) with patch.object( _sick_tracker, "load_history", return_value=SickHistory() ) as mock_load: locker._proceed_to_sick_countdown() mock_load.assert_called_once() assert hasattr(locker, "_sick_history_cache") class TestFinalizeSickDay: """Covers _finalize_sick_day branches including commitment penalty.""" def test_marks_commitment_broken_and_writes_debt( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {} history = SickHistory(commitments={"2026-05-10": True}) locker._sick_history_cache = history object.__setattr__(locker, "unlock_screen", MagicMock()) with ( patch.object(_sick_tracker, "had_commitment_for_today", return_value=True), patch.object(_sick_tracker, "save_history", return_value=True), ): locker._finalize_sick_day() assert locker.workout_data["broke_commitment"] == "true" assert locker.workout_data["type"] == "sick_day" assert "debt" in locker.workout_data locker.unlock_screen.assert_called_once() def test_loads_history_when_cache_missing( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {} object.__setattr__(locker, "unlock_screen", MagicMock()) with ( patch.object( _sick_tracker, "load_history", return_value=SickHistory() ) as mock_load, patch.object(_sick_tracker, "save_history", return_value=True), ): locker._finalize_sick_day() mock_load.assert_called_once() locker.unlock_screen.assert_called_once() # --------------------------------------------------------------------------- # screen_lock.py — _clear_debt_on_verified_workout branches # --------------------------------------------------------------------------- class TestClearDebtOnVerifiedWorkout: """Tests for _clear_debt_on_verified_workout.""" def test_returns_none_when_not_phone_verified( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {"type": "sick_day"} assert locker._clear_debt_on_verified_workout() is None def test_returns_zero_when_no_debt( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {"type": "phone_verified"} with patch.object( _sick_tracker, "load_history", return_value=SickHistory(debt=0) ): assert locker._clear_debt_on_verified_workout() == 0 def test_decrements_when_debt_positive( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {"type": "phone_verified"} history = SickHistory(debt=2) with ( patch.object(_sick_tracker, "load_history", return_value=history), patch.object(_sick_tracker, "save_history", return_value=True) as mock_save, ): assert locker._clear_debt_on_verified_workout() == 1 mock_save.assert_called_once() class TestUnlockScreenCommitmentPrompt: """Tests for unlock_screen branches around commitment prompt + debt label.""" def test_phone_verified_schedules_commitment_prompt( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {"type": "phone_verified"} locker.log_file = tmp_path / "log.json" object.__setattr__(locker, "save_workout_log", MagicMock()) object.__setattr__( locker, "_try_adjust_shutdown_for_workout", MagicMock(return_value=False), ) object.__setattr__( locker, "_clear_debt_on_verified_workout", MagicMock(return_value=0), ) locker.unlock_screen() # The last after() call schedules the commitment prompt closure. last_call = locker.root.after.call_args_list[-1] assert last_call.args[0] == 1500 def test_non_verified_schedules_close_directly( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {"type": "sick_day"} locker.log_file = tmp_path / "log.json" object.__setattr__(locker, "save_workout_log", MagicMock()) object.__setattr__( locker, "_try_adjust_shutdown_for_workout", MagicMock(return_value=False), ) object.__setattr__( locker, "_clear_debt_on_verified_workout", MagicMock(return_value=None), ) locker.unlock_screen() # close() goes through root.after directly. locker.root.after.assert_called_with(1500, locker.close) def test_renders_debt_label_when_positive( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker.workout_data = {"type": "phone_verified"} locker.log_file = tmp_path / "log.json" object.__setattr__(locker, "save_workout_log", MagicMock()) object.__setattr__( locker, "_try_adjust_shutdown_for_workout", MagicMock(return_value=True), ) object.__setattr__( locker, "_clear_debt_on_verified_workout", MagicMock(return_value=2), ) locker.unlock_screen() # _text was called via mock_tk.Label; just assert a Label call mentions debt. labels = [call.kwargs.get("text", "") for call in mock_tk.Label.call_args_list] assert any("Workout debt: 2" in t for t in labels) # --------------------------------------------------------------------------- # _sick_dialog.py — UI mixin # --------------------------------------------------------------------------- class TestShowSickJustification: """Tests for the structured sick justification dialog.""" def test_renders_form_without_commitment( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) with patch.object(_sick_tracker, "load_history", return_value=SickHistory()): locker._show_sick_justification() assert locker._sick_history_cache.sick_days == [] assert hasattr(locker, "_sick_submit_button") # Submit button starts enabled (no commitment). # config(state="disabled") only called for commitment path. for call in locker._sick_submit_button.config.call_args_list: assert call.kwargs.get("state") != "disabled" def test_renders_form_with_commitment_disables_submit( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) history = SickHistory(commitments={"2026-05-10": True}) with ( patch.object(_sick_tracker, "load_history", return_value=history), patch.object(_sick_tracker, "had_commitment_for_today", return_value=True), ): locker._show_sick_justification() # Submit button was disabled and forced-delay started. states = [ call.kwargs.get("state") for call in locker._sick_submit_button.config.call_args_list ] assert "disabled" in states def test_renders_recent_history_when_present( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) history = SickHistory( justifications=[ {"date": "2026-05-01", "symptom": "fever", "severity": 7}, ], ) with patch.object(_sick_tracker, "load_history", return_value=history): locker._show_sick_justification() labels = [call.kwargs.get("text", "") for call in mock_tk.Label.call_args_list] assert any("Recent sick days" in t for t in labels) class TestUpdateCommitmentForcedDelay: """Tests for _update_commitment_forced_delay.""" def test_ticks_down( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker._sick_submit_button = MagicMock() locker._commitment_forced_remaining = 3 locker._update_commitment_forced_delay() assert locker._commitment_forced_remaining == 2 locker.root.after.assert_called() def test_enables_when_done( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path ) -> None: locker = create_locker(mock_tk, tmp_path) locker._sick_submit_button = MagicMock() locker._commitment_forced_remaining = 0 locker._update_commitment_forced_delay() locker._sick_submit_button.config.assert_called_with( text="SUBMIT", state="normal" )