from unittest.mock import Mock, patch import pytest from improvisation_lab.application.interval_practice.console_interval_app import \ ConsoleIntervalPracticeApp from improvisation_lab.config import Config from improvisation_lab.domain.music_theory import Notes from improvisation_lab.infrastructure.audio import DirectAudioProcessor from improvisation_lab.presentation.interval_practice.console_interval_view import \ ConsoleIntervalPracticeView from improvisation_lab.service import IntervalPracticeService class TestConsoleIntervalPracticeApp: @pytest.fixture def init_module(self): """Initialize ConsoleIntervalPracticeApp for testing.""" config = Config() service = IntervalPracticeService(config) self.app = ConsoleIntervalPracticeApp(service, config) self.app.ui = Mock(spec=ConsoleIntervalPracticeView) self.app.audio_processor = Mock(spec=DirectAudioProcessor) self.app.audio_processor.is_recording = False @pytest.mark.usefixtures("init_module") @patch.object(DirectAudioProcessor, "start_recording", return_value=None) @patch("time.sleep", side_effect=KeyboardInterrupt) def test_launch(self, mock_start_recording, mock_sleep): """Test launching the application.""" self.app.launch() assert self.app.is_running assert self.app.current_phrase_idx == 0 assert self.app.current_note_idx == 0 self.app.ui.launch.assert_called_once() self.app.ui.display_phrase_info.assert_called_once_with(0, self.app.phrases) mock_start_recording.assert_called_once() @pytest.mark.usefixtures("init_module") def test_process_audio_callback(self): """Test processing audio callback.""" audio_data = Mock() self.app.phrases = [ [Notes.C, Notes.C_SHARP, Notes.C], [Notes.D, Notes.D_SHARP, Notes.D], ] self.app.current_phrase_idx = 0 self.app.current_note_idx = 2 with patch.object( self.app.service, "process_audio", return_value=Mock(remaining_time=0) ) as mock_process_audio: self.app._process_audio_callback(audio_data) mock_process_audio.assert_called_once_with(audio_data, "C") self.app.ui.display_pitch_result.assert_called_once() self.app.ui.display_phrase_info.assert_called_once_with(1, self.app.phrases) @pytest.mark.usefixtures("init_module") def test_advance_to_next_note(self): """Test advancing to the next note.""" self.app.phrases = [[Notes.C, Notes.C_SHARP, Notes.C]] self.app.current_phrase_idx = 0 self.app.current_note_idx = 2 self.app._advance_to_next_note() assert self.app.current_note_idx == 0 assert self.app.current_phrase_idx == 0 self.app.ui.display_phrase_info.assert_called_once_with(0, self.app.phrases)