wecker/tests/test_wecker.py

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()