diff --git a/wake_alarm/tests/test_alarm_display.py b/wake_alarm/tests/test_alarm_display.py index 6dc4900..a71f74c 100644 --- a/wake_alarm/tests/test_alarm_display.py +++ b/wake_alarm/tests/test_alarm_display.py @@ -127,3 +127,20 @@ class TestDisplayHelpers: ): _restore_display() mock_run.assert_not_called() + + def test_restore_display_runs_xset_s_on_when_present(self) -> None: + """_restore_display re-enables the screensaver via xset when present.""" + with ( + patch( + "python_pkg.wake_alarm._alarm_display.shutil.which", + return_value="/usr/bin/xset", + ), + patch("python_pkg.wake_alarm._alarm_display.subprocess.run") as mock_run, + ): + _restore_display() + mock_run.assert_called_once_with( + ["/usr/bin/xset", "s", "on"], + check=False, + capture_output=True, + timeout=5, + ) diff --git a/wake_alarm/tests/test_alarm_part2.py b/wake_alarm/tests/test_alarm_part2.py index e7a7181..81b60c8 100644 --- a/wake_alarm/tests/test_alarm_part2.py +++ b/wake_alarm/tests/test_alarm_part2.py @@ -59,6 +59,7 @@ def _block_extra_devices() -> Generator[MagicMock]: patch("python_pkg.wake_alarm._alarm._restore_fans"), patch("python_pkg.wake_alarm._alarm._set_max_brightness"), patch("python_pkg.wake_alarm._alarm._wake_display"), + patch("python_pkg.wake_alarm._alarm._restore_display"), patch("python_pkg.wake_alarm._alarm._warn_if_no_real_sink"), patch("python_pkg.wake_alarm._alarm._activate_alarm_audio", return_value=None), patch("python_pkg.wake_alarm._alarm._restore_alarm_audio"), diff --git a/wake_alarm/tests/test_alarm_part3.py b/wake_alarm/tests/test_alarm_part3.py index 249c4d5..f2772f4 100644 --- a/wake_alarm/tests/test_alarm_part3.py +++ b/wake_alarm/tests/test_alarm_part3.py @@ -58,6 +58,7 @@ def _block_extra_devices() -> Generator[MagicMock]: patch("python_pkg.wake_alarm._alarm._restore_fans"), patch("python_pkg.wake_alarm._alarm._set_max_brightness"), patch("python_pkg.wake_alarm._alarm._wake_display"), + patch("python_pkg.wake_alarm._alarm._restore_display"), patch("python_pkg.wake_alarm._alarm._warn_if_no_real_sink"), patch("python_pkg.wake_alarm._alarm._activate_alarm_audio", return_value=None), patch("python_pkg.wake_alarm._alarm._restore_alarm_audio"), diff --git a/wake_alarm/tests/test_alarm_part4.py b/wake_alarm/tests/test_alarm_part4.py index b091264..31e52a5 100644 --- a/wake_alarm/tests/test_alarm_part4.py +++ b/wake_alarm/tests/test_alarm_part4.py @@ -56,6 +56,7 @@ def _block_extra_devices() -> Generator[MagicMock]: patch("python_pkg.wake_alarm._alarm._restore_fans"), patch("python_pkg.wake_alarm._alarm._set_max_brightness"), patch("python_pkg.wake_alarm._alarm._wake_display"), + patch("python_pkg.wake_alarm._alarm._restore_display"), patch("python_pkg.wake_alarm._alarm._warn_if_no_real_sink"), patch("python_pkg.wake_alarm._alarm._activate_alarm_audio", return_value=None), patch("python_pkg.wake_alarm._alarm._restore_alarm_audio"),