mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 17:53:56 +01:00
Krzysztof kuhy Rudnicki
aa3e290391
Real one-line docstrings stating what each test pins down. Files that would pass 250 lines moved whole classes into _part2 siblings (sick_panel, earned, poke_server, shutdown_reading, manual_workout_dialog). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01WzxdWYdwRAtMg7ki6TKxuU
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
"""Tests for the heat-skip confirmation dialog and log entry."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from screen_locker._heat_skip import HeatSkipMixin
|
|
|
|
|
|
class _HeatSkipHost(HeatSkipMixin):
|
|
"""Minimal host exposing just what HeatSkipMixin needs from ScreenLocker."""
|
|
|
|
def __init__(self) -> None:
|
|
self.workout_data: dict[str, str] = {}
|
|
self.save_workout_log = MagicMock()
|
|
|
|
|
|
def _button_command(mock_tk: MagicMock, text: str):
|
|
for call in mock_tk.Button.call_args_list:
|
|
if call.kwargs.get("text") == text:
|
|
return call.kwargs["command"]
|
|
msg = f"No button with text {text!r} was created"
|
|
raise AssertionError(msg)
|
|
|
|
|
|
class TestShowHeatSkipDialog:
|
|
"""_show_heat_skip_dialog blocks on mainloop() until a button fires destroy()."""
|
|
|
|
def test_returns_true_when_skip_clicked(self, mock_tk: MagicMock) -> None:
|
|
"""Clicking Skip workout makes the dialog return True and closes the window."""
|
|
host = _HeatSkipHost()
|
|
|
|
def _simulate_click() -> None:
|
|
_button_command(mock_tk, "Skip workout")()
|
|
|
|
mock_tk.Tk.return_value.mainloop.side_effect = _simulate_click
|
|
assert host._show_heat_skip_dialog(35.0) is True
|
|
mock_tk.Tk.return_value.destroy.assert_called_once()
|
|
|
|
def test_returns_false_when_declined(self, mock_tk: MagicMock) -> None:
|
|
"""Declining makes the dialog return False."""
|
|
host = _HeatSkipHost()
|
|
|
|
def _simulate_click() -> None:
|
|
_button_command(mock_tk, "No, I'll workout")()
|
|
|
|
mock_tk.Tk.return_value.mainloop.side_effect = _simulate_click
|
|
assert host._show_heat_skip_dialog(35.0) is False
|
|
|
|
def test_dialog_shows_temperature_and_threshold(self, mock_tk: MagicMock) -> None:
|
|
"""The dialog text shows the rounded temperature and mentions the threshold."""
|
|
host = _HeatSkipHost()
|
|
mock_tk.Tk.return_value.mainloop.side_effect = None
|
|
host._show_heat_skip_dialog(36.4)
|
|
texts = [c.kwargs.get("text", "") for c in mock_tk.Label.call_args_list]
|
|
assert any("36°C" in t for t in texts)
|
|
assert any("threshold" in t for t in texts)
|
|
|
|
|
|
class TestSaveHeatSkipLog:
|
|
"""_save_heat_skip_log builds workout_data and delegates to save_workout_log."""
|
|
|
|
def test_saves_expected_workout_data(self) -> None:
|
|
"""A heat skip is saved as a heat_skip entry with the rounded temperature and a city."""
|
|
host = _HeatSkipHost()
|
|
host._save_heat_skip_log(34.6)
|
|
assert host.workout_data["type"] == "heat_skip"
|
|
assert host.workout_data["temperature_celsius"] == "35"
|
|
assert host.workout_data["city"]
|
|
host.save_workout_log.assert_called_once()
|