Josh Knapp
67ed69df00
Send each segment to the frontend immediately after transcription via a new pipeline.segment IPC message, then send speaker assignments as a batch pipeline.speaker_update message after diarization completes. This lets the UI display segments progressively instead of waiting for the entire pipeline to finish. Changes: - Add partial_segment_message and speaker_update_message IPC factories - Add on_segment callback parameter to TranscribeService.transcribe() - Emit partial segments and speaker updates from PipelineService.run() - Add send_and_receive_with_progress to SidecarManager (Rust) - Route pipeline.segment/speaker_update events in run_pipeline command - Listen for streaming events in Svelte frontend (+page.svelte) - Add tests for new message types, callback signature, and update logic Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
"""Tests for IPC message types."""
|
|
|
|
from voice_to_notes.ipc.messages import (
|
|
IPCMessage,
|
|
error_message,
|
|
partial_segment_message,
|
|
progress_message,
|
|
ready_message,
|
|
speaker_update_message,
|
|
)
|
|
|
|
|
|
def test_ipc_message_to_dict():
|
|
msg = IPCMessage(id="req-1", type="ping", payload={"key": "value"})
|
|
d = msg.to_dict()
|
|
assert d == {"id": "req-1", "type": "ping", "payload": {"key": "value"}}
|
|
|
|
|
|
def test_ipc_message_from_dict():
|
|
data = {"id": "req-1", "type": "ping", "payload": {"key": "value"}}
|
|
msg = IPCMessage.from_dict(data)
|
|
assert msg.id == "req-1"
|
|
assert msg.type == "ping"
|
|
assert msg.payload == {"key": "value"}
|
|
|
|
|
|
def test_ipc_message_from_dict_missing_fields():
|
|
msg = IPCMessage.from_dict({})
|
|
assert msg.id == ""
|
|
assert msg.type == ""
|
|
assert msg.payload == {}
|
|
|
|
|
|
def test_progress_message():
|
|
msg = progress_message("req-1", 50, "transcribing", "Processing...")
|
|
assert msg.type == "progress"
|
|
assert msg.payload["percent"] == 50
|
|
assert msg.payload["stage"] == "transcribing"
|
|
|
|
|
|
def test_error_message():
|
|
msg = error_message("req-1", "not_found", "File not found")
|
|
assert msg.type == "error"
|
|
assert msg.payload["code"] == "not_found"
|
|
assert msg.payload["message"] == "File not found"
|
|
|
|
|
|
def test_ready_message():
|
|
msg = ready_message()
|
|
assert msg.type == "ready"
|
|
assert msg.id == "system"
|
|
assert "version" in msg.payload
|
|
|
|
|
|
def test_partial_segment_message():
|
|
msg = partial_segment_message("req-1", {"index": 0, "text": "hello"})
|
|
assert msg.type == "pipeline.segment"
|
|
assert msg.payload["index"] == 0
|
|
assert msg.payload["text"] == "hello"
|
|
|
|
|
|
def test_speaker_update_message():
|
|
msg = speaker_update_message("req-1", [{"index": 0, "speaker": "SPEAKER_00"}])
|
|
assert msg.type == "pipeline.speaker_update"
|
|
assert msg.payload["updates"][0]["speaker"] == "SPEAKER_00"
|