mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-11 13:47:39 +01:00
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
68 lines
2.8 KiB
Python
68 lines
2.8 KiB
Python
"""The status window shows the workout streak and early-bird state it tracks."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from screen_locker.tests._status_view_helpers import (
|
|
_make_window,
|
|
_snapshot,
|
|
_texts,
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
from unittest.mock import MagicMock
|
|
|
|
|
|
def _fg_of(mock_tk: MagicMock, needle: str) -> list[object]:
|
|
return [
|
|
c.kwargs.get("fg")
|
|
for c in mock_tk.Label.call_args_list
|
|
if needle in c.kwargs.get("text", "")
|
|
]
|
|
|
|
|
|
class TestSectionStreak:
|
|
"""Tests for StatusWindow._section_streak (display only)."""
|
|
|
|
def test_shows_the_real_week_count_in_success_color(
|
|
self, mock_tk: MagicMock
|
|
) -> None:
|
|
"""The streak line shows the real week count in the success colour."""
|
|
window = _make_window(mock_tk, _snapshot(streak=3))
|
|
assert any("3-week streak" in t for t in _texts(mock_tk))
|
|
assert window._colors.palette.success in _fg_of(mock_tk, "3-week streak")
|
|
|
|
def test_zero_streak_is_stated_plainly_not_padded(self, mock_tk: MagicMock) -> None:
|
|
"""A zero streak reads No streak in the muted colour, with no week count."""
|
|
window = _make_window(mock_tk, _snapshot(streak=0))
|
|
texts = _texts(mock_tk)
|
|
assert any("No streak" in t for t in texts)
|
|
assert not any("-week streak" in t for t in texts)
|
|
assert window._colors.palette.muted in _fg_of(mock_tk, "No streak")
|
|
|
|
def test_early_bird_is_never_rendered(self, mock_tk: MagicMock) -> None:
|
|
"""The 09:00 early-bird extension is gone (wake-alarm owns it)."""
|
|
_make_window(mock_tk, _snapshot(streak=2, early_bird_extended=True))
|
|
assert not any("arly-bird" in t for t in _texts(mock_tk))
|
|
|
|
def test_week_reward_reaches_the_window_in_success_color(
|
|
self, mock_tk: MagicMock
|
|
) -> None:
|
|
"""A week reward message is shown in the success colour."""
|
|
msg = "5 workouts in 2026-W41! 4-week streak"
|
|
window = _make_window(mock_tk, _snapshot(streak=4, week_rewards=(msg,)))
|
|
assert msg in _texts(mock_tk)
|
|
assert window._colors.palette.success in _fg_of(mock_tk, msg)
|
|
|
|
def test_no_reward_text_when_the_week_has_none(self, mock_tk: MagicMock) -> None:
|
|
"""No reward line appears when the week has no reward."""
|
|
_make_window(mock_tk, _snapshot(streak=4))
|
|
assert not any("workouts in 20" in t for t in _texts(mock_tk))
|
|
|
|
def test_a_streak_reset_is_not_painted_as_a_win(self, mock_tk: MagicMock) -> None:
|
|
"""A streak-reset message is shown in the warning colour, not as a win."""
|
|
msg = "Streak reset (was 3 weeks of 5+ workouts)"
|
|
window = _make_window(mock_tk, _snapshot(streak=0, week_rewards=(msg,)))
|
|
assert window._colors.palette.warning in _fg_of(mock_tk, msg)
|