test_alarm_controller.py

import sys
import unittest
from unittest.mock import MagicMock, patch


class TestAlarmController(unittest.TestCase):
    def setUp(self):
        # Clear any cached import of alarm_controller between tests
        if "alarm_controller" in sys.modules:
            del sys.modules["alarm_controller"]

    @patch.dict("sys.modules", {"vlc": MagicMock()})
    def test_play_starts_video(self):
        import alarm_controller

        mock_player = MagicMock()
        mock_player.is_playing.return_value = 0  # Not currently playing
        alarm_controller.vlc.MediaPlayer.return_value = mock_player

        controller = alarm_controller.AlarmController("meme.mp4")
        controller.play()

        mock_player.play.assert_called_once()

    @patch.dict("sys.modules", {"vlc": MagicMock()})
    def test_stop_stops_and_rewinds(self):
        import alarm_controller

        mock_player = MagicMock()
        alarm_controller.vlc.MediaPlayer.return_value = mock_player

        controller = alarm_controller.AlarmController("meme.mp4")
        controller.stop()

        mock_player.stop.assert_called_once()

    @patch.dict("sys.modules", {"vlc": MagicMock()})
    def test_is_playing_returns_state(self):
        import alarm_controller

        mock_player = MagicMock()
        mock_player.is_playing.return_value = 1
        alarm_controller.vlc.MediaPlayer.return_value = mock_player

        controller = alarm_controller.AlarmController("meme.mp4")

        self.assertTrue(controller.is_playing())

    @patch.dict("sys.modules", {"vlc": MagicMock()})
    def test_sets_loop_mode(self):
        import alarm_controller

        mock_player = MagicMock()
        mock_media = MagicMock()
        mock_player.get_media.return_value = mock_media
        alarm_controller.vlc.MediaPlayer.return_value = mock_player

        controller = alarm_controller.AlarmController("meme.mp4")

        mock_media.add_option.assert_called_with("input-repeat=65535")


if __name__ == "__main__":
    unittest.main()