complete restructure

vibe_bot/tests/test_tts.py

@@ -22,7 +22,7 @@ def test_tts_engine_init(mock_kokoro_tts: MagicMock) -> None:
 
 
 def test_generate_audio(mock_kokoro_tts: MagicMock) -> None:
-    """Test audio generation returns a BytesIO object."""
+    """Test audio generation returns a full (non-partial) AudioResult."""
     from io import BytesIO
 
     from vibe_bot.tts import TTSEngine
@@ -30,9 +30,11 @@ def test_generate_audio(mock_kokoro_tts: MagicMock) -> None:
     engine = TTSEngine("/tmp/test-model.onnx", "/tmp/test-voices.bin")
     result = engine.generate_audio("hello world this is a test")
 
-    assert isinstance(result, BytesIO)
-    result.seek(0)
-    data = result.read()
+    assert isinstance(result.audio, BytesIO)
+    assert result.partial is False
+    assert result.failed_chunks == 0
+    result.audio.seek(0)
+    data = result.audio.read()
     assert len(data) > 0
 
 
@@ -57,7 +59,8 @@ def test_generate_audio_single_chunk(mock_kokoro_tts: MagicMock) -> None:
     engine = TTSEngine("/tmp/test-model.onnx", "/tmp/test-voices.bin")
     result = engine.generate_audio("single chunk text")
 
-    assert isinstance(result, BytesIO)
+    assert isinstance(result.audio, BytesIO)
+    assert result.partial is False
     mock_kokoro_tts["process_chunk_sequential"].assert_called_once()
 
 
@@ -77,7 +80,9 @@ def test_generate_audio_multiple_chunks(mock_kokoro_tts: MagicMock) -> None:
         "this text is long enough to be split into multiple chunks",
     )
 
-    assert isinstance(result, BytesIO)
+    assert isinstance(result.audio, BytesIO)
+    assert result.partial is False
+    assert result.failed_chunks == 0
     assert mock_kokoro_tts["process_chunk_sequential"].call_count == 3
 
 
@@ -108,7 +113,12 @@ def test_generate_audio_chunk_failure(mock_kokoro_tts: MagicMock) -> None:
     engine = TTSEngine("/tmp/test-model.onnx", "/tmp/test-voices.bin")
     result = engine.generate_audio("good chunk bad chunk another good")
 
-    assert isinstance(result, BytesIO)
+    # Audio is still produced for the good chunks, but flagged as partial.
+    assert isinstance(result.audio, BytesIO)
+    assert result.partial is True
+    assert result.failed_chunks == 1
+    result.audio.seek(0)
+    assert len(result.audio.read()) > 0
 
 
 def test_generate_audio_all_chunks_fail(mock_kokoro_tts: MagicMock) -> None:
@@ -145,13 +155,13 @@ def test_generate_audio_returns_seekable(mock_kokoro_tts: MagicMock) -> None:
     engine = TTSEngine("/tmp/test-model.onnx", "/tmp/test-voices.bin")
     result = engine.generate_audio("hello world")
 
-    result.seek(0)
-    data = result.read()
+    result.audio.seek(0)
+    data = result.audio.read()
     assert len(data) > 0
 
     # Should be able to seek and read again
-    result.seek(0)
-    data2 = result.read()
+    result.audio.seek(0)
+    data2 = result.audio.read()
     assert data == data2