test: add delays in WebSocket streaming tests to prevent SSL errors

Author: twangodev

tests/integration/test_tts_websocket_integration.py

@@ -36,6 +36,8 @@ def text_stream():
 
     def test_websocket_streaming_with_different_models(self, client, save_audio):
         """Test WebSocket streaming with different models."""
+        import time
+
         models = get_args(Model)
 
         for model in models:
@@ -49,6 +51,9 @@ def text_stream():
             # Write to output directory
             save_audio(audio_chunks, f"test_websocket_model_{model}.mp3")
 
+            # Brief delay to avoid SSL errors when opening next WebSocket connection
+            time.sleep(0.3)
+
     def test_websocket_streaming_with_wav_format(self, client, save_audio):
         """Test WebSocket streaming with WAV format."""
         config = TTSConfig(format="wav", chunk_length=200)
@@ -218,6 +223,8 @@ async def test_async_websocket_streaming_with_different_models(
         self, async_client, save_audio
     ):
         """Test async WebSocket streaming with different models."""
+        import asyncio
+
         models = get_args(Model)
 
         for model in models:
@@ -236,6 +243,9 @@ async def text_stream():
             # Write to output directory
             save_audio(audio_chunks, f"test_async_websocket_model_{model}.mp3")
 
+            # Brief delay to avoid SSL errors when opening next WebSocket connection
+            await asyncio.sleep(0.3)
+
     @pytest.mark.asyncio
     async def test_async_websocket_streaming_with_format(
         self, async_client, save_audio
@@ -326,6 +336,8 @@ async def test_async_websocket_streaming_multiple_calls(
         self, async_client, save_audio
     ):
         """Test multiple async WebSocket streaming calls in sequence."""
+        import asyncio
+
         for i in range(3):
 
             async def text_stream():
@@ -338,6 +350,9 @@ async def text_stream():
             assert len(audio_chunks) > 0, f"Call {i + 1} should return audio"
             save_audio(audio_chunks, f"test_async_websocket_call_{i + 1}.mp3")
 
+            # Brief delay to avoid SSL errors when opening next WebSocket connection
+            await asyncio.sleep(0.3)
+
     @pytest.mark.asyncio
     async def test_async_websocket_streaming_empty_text(self, async_client, save_audio):
         """Test async WebSocket streaming with empty text stream raises error."""