DataSage12 commited on
Commit
513f551
·
1 Parent(s): 218be2f

Clean up: remove redundant test_audio_websocket.py script

Browse files
Files changed (3) hide show
  1. Dockerfile +0 -1
  2. app.py +0 -12
  3. test_audio_websocket.py +0 -85
Dockerfile CHANGED
@@ -66,7 +66,6 @@ COPY app.py .
66
 
67
  # Copy diagnostic scripts
68
  COPY debug_websocket.py .
69
- COPY test_audio_websocket.py .
70
  COPY test_services_detailed.py .
71
 
72
  # Create necessary directories and set permissions
 
66
 
67
  # Copy diagnostic scripts
68
  COPY debug_websocket.py .
 
69
  COPY test_services_detailed.py .
70
 
71
  # Create necessary directories and set permissions
app.py CHANGED
@@ -128,18 +128,6 @@ def start_services():
128
  except Exception as e:
129
  logger.warning(f"Impossible d'exécuter les tests WebSocket: {e}")
130
 
131
- # Test audio WebSocket
132
- logger.info("🔍 Test audio WebSocket...")
133
- try:
134
- result = subprocess.run([sys.executable, "test_audio_websocket.py"],
135
- capture_output=True, text=True, timeout=60)
136
- if result.stdout:
137
- logger.info(f"Résultats des tests audio WebSocket:\n{result.stdout}")
138
- if result.stderr:
139
- logger.error(f"Erreurs des tests audio WebSocket:\n{result.stderr}")
140
- except Exception as e:
141
- logger.warning(f"Impossible d'exécuter les tests audio WebSocket: {e}")
142
-
143
  # Test détaillé des services
144
  logger.info("🔍 Test détaillé des services...")
145
  try:
 
128
  except Exception as e:
129
  logger.warning(f"Impossible d'exécuter les tests WebSocket: {e}")
130
 
 
 
 
 
 
 
 
 
 
 
 
 
131
  # Test détaillé des services
132
  logger.info("🔍 Test détaillé des services...")
133
  try:
test_audio_websocket.py DELETED
@@ -1,85 +0,0 @@
1
- #!/usr/bin/env python3
2
- """
3
- Script de test pour diagnostiquer le problème audio WebSocket
4
- """
5
-
6
- import asyncio
7
- import aiohttp
8
- import json
9
- import sys
10
- import wave
11
- import tempfile
12
- import os
13
-
14
- async def test_audio_websocket():
15
- """Teste l'envoi d'audio via WebSocket"""
16
- print("🔍 Test d'envoi audio via WebSocket...")
17
-
18
- try:
19
- # Créer un fichier audio de test
20
- with tempfile.NamedTemporaryFile(suffix='.wav', delete=False) as f:
21
- # Créer un fichier WAV de 2 secondes de silence
22
- with wave.open(f.name, 'w') as wav_file:
23
- wav_file.setnchannels(1)
24
- wav_file.setsampwidth(2)
25
- wav_file.setframerate(16000)
26
- wav_file.writeframes(b'\x00' * 64000) # 2 secondes de silence
27
-
28
- # Lire le fichier audio
29
- with open(f.name, 'rb') as audio_file:
30
- audio_data = audio_file.read()
31
-
32
- print(f"✅ Fichier audio créé: {len(audio_data)} bytes")
33
-
34
- # Test de connexion WebSocket avec audio
35
- async with aiohttp.ClientSession() as session:
36
- async with session.ws_connect("ws://localhost:5003/live_stream/test_user?lang=fr&room_id=test") as ws:
37
- print("✅ Connexion WebSocket établie")
38
-
39
- # Envoyer un message de démarrage
40
- start_message = {"type": "start", "user_id": "test_user"}
41
- await ws.send_str(json.dumps(start_message))
42
- print("✅ Message de démarrage envoyé")
43
-
44
- # Attendre un peu
45
- await asyncio.sleep(1)
46
-
47
- # Envoyer l'audio par chunks
48
- chunk_size = 1024
49
- for i in range(0, len(audio_data), chunk_size):
50
- chunk = audio_data[i:i+chunk_size]
51
- await ws.send_bytes(chunk)
52
- print(f"✅ Chunk audio envoyé: {len(chunk)} bytes")
53
- await asyncio.sleep(0.1) # Petite pause entre les chunks
54
-
55
- # Envoyer un message d'arrêt
56
- stop_message = {"type": "stop", "user_id": "test_user"}
57
- await ws.send_str(json.dumps(stop_message))
58
- print("✅ Message d'arrêt envoyé")
59
-
60
- # Attendre une réponse
61
- try:
62
- async with asyncio.timeout(10):
63
- msg = await ws.receive()
64
- print(f"✅ Réponse reçue: {msg}")
65
- except asyncio.TimeoutError:
66
- print("⚠️ Aucune réponse reçue dans les 10 secondes")
67
-
68
- await ws.close()
69
- print("✅ Connexion WebSocket fermée")
70
-
71
- # Nettoyer le fichier temporaire
72
- os.unlink(f.name)
73
-
74
- except Exception as e:
75
- print(f"❌ Erreur: {e}")
76
- import traceback
77
- print(f"Traceback: {traceback.format_exc()}")
78
-
79
- async def main():
80
- print("🚀 Démarrage du test audio WebSocket...")
81
- await test_audio_websocket()
82
- print("✅ Test terminé")
83
-
84
- if __name__ == "__main__":
85
- asyncio.run(main())