gurix
92eba762ce
Move audio ownership out of the shared runner and into each style via an injected Sound capability (play_music(path) / play_beep()). The runner now only owns button sampling and cleanup; the style kicks off its own tone in start() and drives the LED. - simple: a repeating square-wave beep (ordinary alarm-clock tone), no music file required. The beep is synthesised in memory as signed-16-bit stereo PCM (stdlib array+math) and played via pygame.mixer.Sound — no shipped audio asset, no new dependency. - blink: unchanged behaviour — music on an endless loop from --music-file / MUSIC_FILE / the default track, via pygame.mixer.music. - AlarmStyle contract gains sound + music_file in __init__ and a start() lifecycle hook; AlarmSound Protocol documents the audio seam for future styles that handle their own tone. - setup() no longer loads music (that is the style's job now). README updated: per-style ringing tone, the --music-file note (blink only), and the extended plugin contract.
127 lines
4.4 KiB
Python
127 lines
4.4 KiB
Python
import pytest
|
|
import sys
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
# Mock RPi.GPIO and pygame before importing wecker
|
|
sys.modules["RPi"] = MagicMock()
|
|
sys.modules["RPi.GPIO"] = MagicMock()
|
|
sys.modules["pygame"] = MagicMock()
|
|
|
|
import wecker # noqa: E402
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_gpio():
|
|
with patch.object(wecker, "GPIO") as mock:
|
|
mock.LOW = 0
|
|
mock.HIGH = 1
|
|
yield mock
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_pygame():
|
|
with patch.object(wecker, "pygame") as mock:
|
|
yield mock
|
|
|
|
|
|
def test_set_led(mock_gpio):
|
|
wecker.set_led(True)
|
|
mock_gpio.output.assert_called_with(wecker.LED_PIN, mock_gpio.LOW)
|
|
|
|
wecker.set_led(False)
|
|
mock_gpio.output.assert_called_with(wecker.LED_PIN, mock_gpio.HIGH)
|
|
|
|
|
|
def test_setup_inits_mixer_and_gpio_not_music(mock_gpio, mock_pygame):
|
|
wecker.setup()
|
|
assert mock_pygame.mixer.pre_init.called
|
|
assert mock_pygame.mixer.init.called
|
|
assert mock_gpio.setmode.called
|
|
assert mock_gpio.setup.called
|
|
# setup() no longer loads music — that is now the style's job.
|
|
assert not mock_pygame.mixer.music.load.called
|
|
|
|
|
|
def test_run_alarm_unknown_style_raises(mock_gpio, mock_pygame):
|
|
"""An unknown style is rejected before any hardware is touched."""
|
|
with pytest.raises(ValueError, match="Unknown alarm style"):
|
|
wecker.run_alarm(style_name="nope", test_mode=True)
|
|
mock_pygame.mixer.init.assert_not_called()
|
|
|
|
|
|
def test_run_alarm_uses_selected_style(mock_gpio, mock_pygame):
|
|
"""run_alarm looks up the style, injects a Sound, and calls start()."""
|
|
fake = MagicMock()
|
|
fake.update.return_value = True # keep ringing
|
|
fake_style_cls = MagicMock(return_value=fake)
|
|
|
|
with patch("wecker.get_style", return_value=fake_style_cls) as get_style:
|
|
wecker.run_alarm(style_name="whatever", test_mode=True)
|
|
|
|
get_style.assert_called_once_with("whatever")
|
|
args = fake_style_cls.call_args.args
|
|
assert args[0] is wecker.set_led
|
|
assert isinstance(args[1], wecker.Sound)
|
|
fake.start.assert_called_once()
|
|
fake.update.assert_called()
|
|
|
|
|
|
def test_run_alarm_default_music_file(mock_gpio, mock_pygame, monkeypatch):
|
|
monkeypatch.delenv("MUSIC_FILE", raising=False)
|
|
fake = MagicMock()
|
|
fake.update.return_value = True
|
|
fake_style_cls = MagicMock(return_value=fake)
|
|
with patch("wecker.get_style", return_value=fake_style_cls):
|
|
wecker.run_alarm(test_mode=True)
|
|
assert fake_style_cls.call_args.args[2] == wecker.DEFAULT_MUSIC_FILE
|
|
|
|
|
|
def test_run_alarm_env_music_file(mock_gpio, mock_pygame, monkeypatch):
|
|
monkeypatch.setenv("MUSIC_FILE", "/env/track.mp3")
|
|
fake = MagicMock()
|
|
fake.update.return_value = True
|
|
fake_style_cls = MagicMock(return_value=fake)
|
|
with patch("wecker.get_style", return_value=fake_style_cls):
|
|
wecker.run_alarm(test_mode=True)
|
|
assert fake_style_cls.call_args.args[2] == "/env/track.mp3"
|
|
|
|
|
|
def test_run_alarm_music_file_arg_overrides_env(mock_gpio, mock_pygame, monkeypatch):
|
|
monkeypatch.setenv("MUSIC_FILE", "/env/track.mp3")
|
|
fake = MagicMock()
|
|
fake.update.return_value = True
|
|
fake_style_cls = MagicMock(return_value=fake)
|
|
with patch("wecker.get_style", return_value=fake_style_cls):
|
|
wecker.run_alarm(music_file="/arg/track.mp3", test_mode=True)
|
|
assert fake_style_cls.call_args.args[2] == "/arg/track.mp3"
|
|
|
|
|
|
@patch("wecker.time.time")
|
|
def test_run_alarm_blink_plays_music(mock_time, mock_gpio, mock_pygame):
|
|
"""The default (blink) style plays music on an endless loop."""
|
|
wecker.run_alarm(test_mode=True)
|
|
assert mock_pygame.mixer.music.play.called
|
|
assert mock_pygame.mixer.music.play.call_args.args[0] == -1 # endless loop
|
|
|
|
|
|
def test_run_alarm_simple_beeps_and_plays_no_music(mock_gpio, mock_pygame):
|
|
"""The simple style beeps and never touches the music stream."""
|
|
wecker.run_alarm(style_name="simple", test_mode=True)
|
|
assert mock_pygame.mixer.Sound.return_value.play.called # beep played
|
|
assert not mock_pygame.mixer.music.play.called # no music
|
|
|
|
|
|
def test_sound_play_music_loads_and_loops(mock_pygame):
|
|
s = wecker.Sound()
|
|
s.play_music("/track.mp3")
|
|
mock_pygame.mixer.music.load.assert_called_with("/track.mp3")
|
|
mock_pygame.mixer.music.set_volume.assert_called_with(1.0)
|
|
mock_pygame.mixer.music.play.assert_called_with(-1)
|
|
|
|
|
|
def test_sound_play_beep_plays_buffer(mock_pygame):
|
|
s = wecker.Sound()
|
|
s.play_beep()
|
|
mock_pygame.mixer.Sound.assert_called_once_with(wecker._BEEP_BUFFER)
|
|
mock_pygame.mixer.Sound.return_value.play.assert_called_once()
|