From 09beca6af74cc87b3c6d2eb057338b7356f8be4d Mon Sep 17 00:00:00 2001 From: Raivis Dejus Date: Sat, 7 Mar 2026 11:47:46 +0200 Subject: [PATCH] Fixed test --- buzz/widgets/recording_transcriber_widget.py | 5 ++++- tests/widgets/recording_transcriber_widget_test.py | 4 ++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/buzz/widgets/recording_transcriber_widget.py b/buzz/widgets/recording_transcriber_widget.py index 10c3fd22..1c7bfcd2 100644 --- a/buzz/widgets/recording_transcriber_widget.py +++ b/buzz/widgets/recording_transcriber_widget.py @@ -11,7 +11,7 @@ import sounddevice from enum import auto from typing import Optional, Tuple, Any -from PyQt6.QtCore import QThread, Qt, QThreadPool, QTimer +from PyQt6.QtCore import QThread, Qt, QThreadPool, QTimer, pyqtSignal from PyQt6.QtGui import QTextCursor, QCloseEvent, QColor from PyQt6.QtWidgets import ( QWidget, @@ -71,6 +71,8 @@ class RecordingTranscriberWidget(QWidget): recording_amplitude_listener: Optional[RecordingAmplitudeListener] = None device_sample_rate: Optional[int] = None + transcription_stopped = pyqtSignal() + class RecordingStatus(enum.Enum): STOPPED = auto() RECORDING = auto() @@ -1045,6 +1047,7 @@ class RecordingTranscriberWidget(QWidget): self.reset_record_button() # Restart amplitude listener now that the transcription stream is closed self.reset_recording_amplitude_listener() + self.transcription_stopped.emit() def on_transcriber_error(self, error: str): self.reset_record_button() diff --git a/tests/widgets/recording_transcriber_widget_test.py b/tests/widgets/recording_transcriber_widget_test.py index 36512f12..564664ae 100644 --- a/tests/widgets/recording_transcriber_widget_test.py +++ b/tests/widgets/recording_transcriber_widget_test.py @@ -58,7 +58,7 @@ class TestRecordingTranscriberWidget: widget.record_button.click() qtbot.wait_until(callback=assert_text_box_contains_text, timeout=60 * 1000) - with qtbot.wait_signal(widget.transcription_thread.finished, timeout=60 * 1000): + with qtbot.wait_signal(widget.transcription_stopped, timeout=60 * 1000): widget.stop_recording() assert len(widget.transcription_text_box.toPlainText()) > 0 @@ -104,7 +104,7 @@ class TestRecordingTranscriberWidget: widget.record_button.click() qtbot.wait_until(callback=assert_text_box_contains_text, timeout=60 * 1000) - with qtbot.wait_signal(widget.transcription_thread.finished, timeout=60 * 1000): + with qtbot.wait_signal(widget.transcription_stopped, timeout=60 * 1000): widget.stop_recording() assert len(widget.transcription_text_box.toPlainText()) > 0