test_satellite_streaming.py 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089
  1. # -*- coding: utf-8 -*-
  2. """
  3. Tests für Live-Audio-Streaming zwischen Satellites.
  4. Testet:
  5. - Streaming EventData-Klassen
  6. - Satellite Streaming Properties
  7. - start_stream / stop_stream Methoden
  8. - start_record_stream / stop_record_stream Methoden
  9. - Private Handler (_on_stream_audio_received, etc.)
  10. - Mutual Exclusion (Recording vs. Streaming)
  11. - Disconnect-Guard
  12. - Integration mit echtem EventManager
  13. """
  14. import asyncio
  15. import os
  16. import wave
  17. import pytest
  18. from unittest.mock import AsyncMock, MagicMock, patch
  19. from trixy_core.events.event_data.basic import (
  20. SatelliteStreamStarted,
  21. SatelliteStreamStopped,
  22. SatelliteStreamRecordStarted,
  23. SatelliteStreamRecordStopped,
  24. )
  25. from trixy_core.satellite.satellite import Satellite, ConnectionState
  26. # =============================================================================
  27. # Fixtures
  28. # =============================================================================
  29. @pytest.fixture
  30. def mock_event_manager():
  31. """Erstellt einen Mock-EventManager."""
  32. manager = MagicMock()
  33. manager.trigger = AsyncMock()
  34. manager.register = MagicMock()
  35. manager.unregister = MagicMock(return_value=True)
  36. return manager
  37. @pytest.fixture
  38. def mock_application(mock_event_manager):
  39. """Erstellt eine Mock-Application."""
  40. app = MagicMock()
  41. app.events = mock_event_manager
  42. return app
  43. @pytest.fixture
  44. def source_satellite(mock_application):
  45. """Erstellt einen verbundenen Quell-Satellite (Mikrofon)."""
  46. sat = Satellite(
  47. satellite_id="sat-source-001",
  48. room_id="wohnzimmer",
  49. mac_address="AA:BB:CC:DD:EE:01",
  50. alias="Wohnzimmer",
  51. application=mock_application,
  52. )
  53. sat.set_connected()
  54. return sat
  55. @pytest.fixture
  56. def target_satellite(mock_application):
  57. """Erstellt einen verbundenen Ziel-Satellite (Lautsprecher)."""
  58. sat = Satellite(
  59. satellite_id="sat-target-002",
  60. room_id="buero",
  61. mac_address="AA:BB:CC:DD:EE:02",
  62. alias="Büro",
  63. application=mock_application,
  64. )
  65. sat.set_connected()
  66. # Mock audio_out Socket
  67. mock_writer = MagicMock()
  68. mock_writer.write = MagicMock()
  69. mock_writer.drain = AsyncMock()
  70. sat.sockets.audio_out = mock_writer
  71. return sat
  72. # =============================================================================
  73. # Tests für Streaming EventData-Klassen
  74. # =============================================================================
  75. class TestStreamingEventDataClasses:
  76. """Tests für Streaming Event-Datenklassen."""
  77. def test_stream_started_creation(self):
  78. """Testet SatelliteStreamStarted Erstellung."""
  79. event = SatelliteStreamStarted(
  80. satellite_id="sat-001",
  81. target_satellite_id="sat-002",
  82. stream_id="stream-abc123",
  83. )
  84. assert event.satellite_id == "sat-001"
  85. assert event.target_satellite_id == "sat-002"
  86. assert event.stream_id == "stream-abc123"
  87. def test_stream_stopped_creation(self):
  88. """Testet SatelliteStreamStopped Erstellung."""
  89. event = SatelliteStreamStopped(
  90. satellite_id="sat-001",
  91. target_satellite_id="sat-002",
  92. stream_id="stream-abc123",
  93. reason="manual",
  94. duration_seconds=45.3,
  95. bytes_forwarded=1024000,
  96. )
  97. assert event.reason == "manual"
  98. assert event.duration_seconds == 45.3
  99. assert event.bytes_forwarded == 1024000
  100. def test_stream_record_started_creation(self):
  101. """Testet SatelliteStreamRecordStarted Erstellung."""
  102. event = SatelliteStreamRecordStarted(
  103. satellite_id="sat-001",
  104. stream_id="stream-abc123",
  105. filename="/tmp/live.wav",
  106. )
  107. assert event.filename == "/tmp/live.wav"
  108. def test_stream_record_stopped_creation(self):
  109. """Testet SatelliteStreamRecordStopped Erstellung."""
  110. event = SatelliteStreamRecordStopped(
  111. satellite_id="sat-001",
  112. stream_id="stream-abc123",
  113. filename="/tmp/live.wav",
  114. reason="manual",
  115. duration_seconds=10.5,
  116. file_size_bytes=336000,
  117. )
  118. assert event.reason == "manual"
  119. assert event.file_size_bytes == 336000
  120. def test_stream_started_defaults(self):
  121. """Testet Standardwerte von SatelliteStreamStarted."""
  122. event = SatelliteStreamStarted()
  123. assert event.satellite_id == ""
  124. assert event.target_satellite_id == ""
  125. assert event.stream_id == ""
  126. def test_stream_stopped_defaults(self):
  127. """Testet Standardwerte von SatelliteStreamStopped."""
  128. event = SatelliteStreamStopped()
  129. assert event.duration_seconds == 0.0
  130. assert event.bytes_forwarded == 0
  131. # =============================================================================
  132. # Tests für Streaming Properties
  133. # =============================================================================
  134. class TestStreamingProperties:
  135. """Tests für Satellite Streaming-Properties."""
  136. def test_is_streaming_initial(self, source_satellite):
  137. """Testet initialen Streaming-Status."""
  138. assert source_satellite.is_streaming is False
  139. def test_stream_id_initial(self, source_satellite):
  140. """Testet initiale Stream-ID."""
  141. assert source_satellite.stream_id == ""
  142. def test_stream_target_initial(self, source_satellite):
  143. """Testet initiales Stream-Ziel."""
  144. assert source_satellite.stream_target is None
  145. def test_is_stream_recording_initial(self, source_satellite):
  146. """Testet initialen Stream-Recording-Status."""
  147. assert source_satellite.is_stream_recording is False
  148. # =============================================================================
  149. # Tests für start_stream
  150. # =============================================================================
  151. class TestStartStream:
  152. """Tests für start_stream Methode."""
  153. @pytest.mark.asyncio
  154. async def test_start_stream_success(self, source_satellite, target_satellite, mock_event_manager):
  155. """Testet erfolgreichen Stream-Start."""
  156. # Mock send_command
  157. source_satellite.send_command = AsyncMock(return_value=True)
  158. result = await source_satellite.start_stream(target_satellite)
  159. assert result is True
  160. assert source_satellite.is_streaming is True
  161. assert source_satellite.stream_target is target_satellite
  162. assert source_satellite.stream_id.startswith("stream-")
  163. @pytest.mark.asyncio
  164. async def test_start_stream_triggers_event(self, source_satellite, target_satellite, mock_event_manager):
  165. """Testet, dass start_stream das satellite_stream_started Event auslöst."""
  166. source_satellite.send_command = AsyncMock(return_value=True)
  167. await source_satellite.start_stream(target_satellite)
  168. mock_event_manager.trigger.assert_called_once()
  169. call_args = mock_event_manager.trigger.call_args
  170. assert call_args[0][0] == "satellite_stream_started"
  171. event_data = call_args[0][1]
  172. assert isinstance(event_data, SatelliteStreamStarted)
  173. assert event_data.satellite_id == "sat-source-001"
  174. assert event_data.target_satellite_id == "sat-target-002"
  175. @pytest.mark.asyncio
  176. async def test_start_stream_sends_record_start_command(self, source_satellite, target_satellite):
  177. """Testet, dass SatelliteRecordStart Command gesendet wird."""
  178. source_satellite.send_command = AsyncMock(return_value=True)
  179. await source_satellite.start_stream(target_satellite)
  180. source_satellite.send_command.assert_called_once()
  181. cmd = source_satellite.send_command.call_args[0][0]
  182. from trixy_core.network.cmd import SatelliteRecordStart
  183. assert isinstance(cmd, SatelliteRecordStart)
  184. @pytest.mark.asyncio
  185. async def test_start_stream_registers_listeners(self, source_satellite, target_satellite, mock_event_manager):
  186. """Testet, dass Event-Listener registriert werden."""
  187. source_satellite.send_command = AsyncMock(return_value=True)
  188. await source_satellite.start_stream(target_satellite)
  189. # Zwei Listener: audio_input_received und satellite_record_stopped
  190. assert mock_event_manager.register.call_count == 2
  191. event_names = [call[0][0] for call in mock_event_manager.register.call_args_list]
  192. assert "audio_input_received" in event_names
  193. assert "satellite_record_stopped" in event_names
  194. @pytest.mark.asyncio
  195. async def test_start_stream_no_application(self):
  196. """Testet Fehler ohne Application-Referenz."""
  197. sat = Satellite(satellite_id="no-app")
  198. target = Satellite(satellite_id="target")
  199. result = await sat.start_stream(target)
  200. assert result is False
  201. @pytest.mark.asyncio
  202. async def test_start_stream_not_connected(self, mock_application, target_satellite):
  203. """Testet Fehler wenn Quell-Satellite nicht verbunden."""
  204. sat = Satellite(satellite_id="disconnected", application=mock_application)
  205. result = await sat.start_stream(target_satellite)
  206. assert result is False
  207. @pytest.mark.asyncio
  208. async def test_start_stream_already_streaming(self, source_satellite, target_satellite):
  209. """Testet Fehler wenn bereits gestreamt wird."""
  210. source_satellite.send_command = AsyncMock(return_value=True)
  211. await source_satellite.start_stream(target_satellite)
  212. result = await source_satellite.start_stream(target_satellite)
  213. assert result is False
  214. @pytest.mark.asyncio
  215. async def test_start_stream_active_conversation(self, source_satellite, target_satellite):
  216. """Testet Fehler bei aktiver Konversation."""
  217. source_satellite.conversation_id = "conv-123"
  218. result = await source_satellite.start_stream(target_satellite)
  219. assert result is False
  220. @pytest.mark.asyncio
  221. async def test_start_stream_self_target(self, source_satellite):
  222. """Testet Fehler bei Stream zu sich selbst."""
  223. result = await source_satellite.start_stream(source_satellite)
  224. assert result is False
  225. @pytest.mark.asyncio
  226. async def test_start_stream_target_not_connected(self, source_satellite, mock_application):
  227. """Testet Fehler wenn Ziel-Satellite nicht verbunden."""
  228. target = Satellite(
  229. satellite_id="target-disconnected",
  230. application=mock_application,
  231. )
  232. # Nicht set_connected() aufrufen
  233. result = await source_satellite.start_stream(target)
  234. assert result is False
  235. @pytest.mark.asyncio
  236. async def test_start_stream_target_no_audio_out(self, source_satellite, mock_application):
  237. """Testet Fehler wenn Ziel keinen audio_out Socket hat."""
  238. target = Satellite(
  239. satellite_id="target-no-audio",
  240. application=mock_application,
  241. )
  242. target.set_connected()
  243. # audio_out bleibt None
  244. result = await source_satellite.start_stream(target)
  245. assert result is False
  246. @pytest.mark.asyncio
  247. async def test_start_stream_stops_active_recording(self, source_satellite, target_satellite, mock_event_manager):
  248. """Testet Mutual Exclusion: Stream stoppt laufende Aufnahme."""
  249. source_satellite.send_command = AsyncMock(return_value=True)
  250. # Recording simulieren
  251. source_satellite._recording_active = True
  252. source_satellite._recording_id = "rec-test"
  253. source_satellite._recording_listener_registered = True
  254. result = await source_satellite.start_stream(target_satellite)
  255. assert result is True
  256. assert source_satellite.is_streaming is True
  257. # Recording sollte gestoppt worden sein
  258. assert source_satellite._recording_active is False
  259. @pytest.mark.asyncio
  260. async def test_start_stream_with_timeout(self, source_satellite, target_satellite):
  261. """Testet Stream-Start mit Timeout."""
  262. source_satellite.send_command = AsyncMock(return_value=True)
  263. result = await source_satellite.start_stream(target_satellite, timeout=30.0)
  264. assert result is True
  265. assert source_satellite._stream_timeout_handle is not None
  266. @pytest.mark.asyncio
  267. async def test_start_stream_without_timeout(self, source_satellite, target_satellite):
  268. """Testet Stream-Start ohne Timeout (unbegrenzt)."""
  269. source_satellite.send_command = AsyncMock(return_value=True)
  270. result = await source_satellite.start_stream(target_satellite, timeout=0.0)
  271. assert result is True
  272. assert source_satellite._stream_timeout_handle is None
  273. # =============================================================================
  274. # Tests für stop_stream
  275. # =============================================================================
  276. class TestStopStream:
  277. """Tests für stop_stream Methode."""
  278. @pytest.mark.asyncio
  279. async def test_stop_stream_success(self, source_satellite, target_satellite, mock_event_manager):
  280. """Testet erfolgreiches Stream-Stoppen."""
  281. source_satellite.send_command = AsyncMock(return_value=True)
  282. await source_satellite.start_stream(target_satellite)
  283. mock_event_manager.trigger.reset_mock()
  284. result = await source_satellite.stop_stream()
  285. assert result is True
  286. assert source_satellite.is_streaming is False
  287. assert source_satellite.stream_target is None
  288. assert source_satellite.stream_id == ""
  289. @pytest.mark.asyncio
  290. async def test_stop_stream_triggers_event(self, source_satellite, target_satellite, mock_event_manager):
  291. """Testet, dass stop_stream das satellite_stream_stopped Event auslöst."""
  292. source_satellite.send_command = AsyncMock(return_value=True)
  293. await source_satellite.start_stream(target_satellite)
  294. mock_event_manager.trigger.reset_mock()
  295. await source_satellite.stop_stream(reason="manual")
  296. mock_event_manager.trigger.assert_called_once()
  297. call_args = mock_event_manager.trigger.call_args
  298. assert call_args[0][0] == "satellite_stream_stopped"
  299. event_data = call_args[0][1]
  300. assert isinstance(event_data, SatelliteStreamStopped)
  301. assert event_data.reason == "manual"
  302. assert event_data.satellite_id == "sat-source-001"
  303. assert event_data.target_satellite_id == "sat-target-002"
  304. @pytest.mark.asyncio
  305. async def test_stop_stream_sends_record_stop_command(self, source_satellite, target_satellite):
  306. """Testet, dass SatelliteRecordStop Command gesendet wird."""
  307. source_satellite.send_command = AsyncMock(return_value=True)
  308. await source_satellite.start_stream(target_satellite)
  309. source_satellite.send_command.reset_mock()
  310. await source_satellite.stop_stream()
  311. source_satellite.send_command.assert_called_once()
  312. cmd = source_satellite.send_command.call_args[0][0]
  313. from trixy_core.network.cmd import SatelliteRecordStop
  314. assert isinstance(cmd, SatelliteRecordStop)
  315. @pytest.mark.asyncio
  316. async def test_stop_stream_unregisters_listeners(self, source_satellite, target_satellite, mock_event_manager):
  317. """Testet, dass Event-Listener entfernt werden."""
  318. source_satellite.send_command = AsyncMock(return_value=True)
  319. await source_satellite.start_stream(target_satellite)
  320. mock_event_manager.unregister.reset_mock()
  321. await source_satellite.stop_stream()
  322. assert mock_event_manager.unregister.call_count == 2
  323. event_names = [call[0][0] for call in mock_event_manager.unregister.call_args_list]
  324. assert "audio_input_received" in event_names
  325. assert "satellite_record_stopped" in event_names
  326. @pytest.mark.asyncio
  327. async def test_stop_stream_cancels_timeout(self, source_satellite, target_satellite):
  328. """Testet, dass der Timeout-Timer abgebrochen wird."""
  329. source_satellite.send_command = AsyncMock(return_value=True)
  330. await source_satellite.start_stream(target_satellite, timeout=60.0)
  331. # TimerHandle durch Mock ersetzen um cancel-Aufruf zu prüfen
  332. mock_handle = MagicMock()
  333. source_satellite._stream_timeout_handle = mock_handle
  334. await source_satellite.stop_stream()
  335. assert source_satellite._stream_timeout_handle is None
  336. mock_handle.cancel.assert_called_once()
  337. @pytest.mark.asyncio
  338. async def test_stop_stream_not_streaming(self, source_satellite):
  339. """Testet Fehler wenn kein Stream aktiv."""
  340. result = await source_satellite.stop_stream()
  341. assert result is False
  342. @pytest.mark.asyncio
  343. async def test_stop_stream_with_reason(self, source_satellite, target_satellite, mock_event_manager):
  344. """Testet verschiedene Stop-Gründe."""
  345. source_satellite.send_command = AsyncMock(return_value=True)
  346. for reason in ("manual", "timeout", "wakeword", "disconnect", "target_disconnect"):
  347. await source_satellite.start_stream(target_satellite)
  348. mock_event_manager.trigger.reset_mock()
  349. await source_satellite.stop_stream(reason=reason)
  350. event_data = mock_event_manager.trigger.call_args[0][1]
  351. assert event_data.reason == reason
  352. @pytest.mark.asyncio
  353. async def test_stop_stream_also_stops_recording(self, source_satellite, target_satellite, temp_dir, mock_event_manager):
  354. """Testet, dass stop_stream auch Stream-Aufnahme stoppt."""
  355. source_satellite.send_command = AsyncMock(return_value=True)
  356. await source_satellite.start_stream(target_satellite)
  357. wav_path = os.path.join(str(temp_dir), "test_stream.wav")
  358. await source_satellite.start_record_stream(wav_path)
  359. assert source_satellite.is_stream_recording is True
  360. await source_satellite.stop_stream()
  361. assert source_satellite.is_stream_recording is False
  362. assert source_satellite.is_streaming is False
  363. # =============================================================================
  364. # Tests für start_record_stream / stop_record_stream
  365. # =============================================================================
  366. class TestStreamRecording:
  367. """Tests für Stream-Aufnahme auf Disk."""
  368. @pytest.mark.asyncio
  369. async def test_start_record_stream_success(self, source_satellite, target_satellite, temp_dir, mock_event_manager):
  370. """Testet erfolgreichen Start der Stream-Aufnahme."""
  371. source_satellite.send_command = AsyncMock(return_value=True)
  372. await source_satellite.start_stream(target_satellite)
  373. mock_event_manager.trigger.reset_mock()
  374. wav_path = os.path.join(str(temp_dir), "test_record.wav")
  375. result = await source_satellite.start_record_stream(wav_path)
  376. assert result is True
  377. assert source_satellite.is_stream_recording is True
  378. @pytest.mark.asyncio
  379. async def test_start_record_stream_triggers_event(self, source_satellite, target_satellite, temp_dir, mock_event_manager):
  380. """Testet, dass start_record_stream Event auslöst."""
  381. source_satellite.send_command = AsyncMock(return_value=True)
  382. await source_satellite.start_stream(target_satellite)
  383. mock_event_manager.trigger.reset_mock()
  384. wav_path = os.path.join(str(temp_dir), "test_record.wav")
  385. await source_satellite.start_record_stream(wav_path)
  386. mock_event_manager.trigger.assert_called_once()
  387. call_args = mock_event_manager.trigger.call_args
  388. assert call_args[0][0] == "satellite_stream_record_started"
  389. event_data = call_args[0][1]
  390. assert isinstance(event_data, SatelliteStreamRecordStarted)
  391. assert event_data.filename == wav_path
  392. @pytest.mark.asyncio
  393. async def test_start_record_stream_no_active_stream(self, source_satellite, temp_dir):
  394. """Testet Fehler wenn kein Stream aktiv."""
  395. wav_path = os.path.join(str(temp_dir), "test.wav")
  396. result = await source_satellite.start_record_stream(wav_path)
  397. assert result is False
  398. @pytest.mark.asyncio
  399. async def test_start_record_stream_already_recording(self, source_satellite, target_satellite, temp_dir):
  400. """Testet Fehler wenn bereits aufgenommen wird."""
  401. source_satellite.send_command = AsyncMock(return_value=True)
  402. await source_satellite.start_stream(target_satellite)
  403. wav_path = os.path.join(str(temp_dir), "test.wav")
  404. await source_satellite.start_record_stream(wav_path)
  405. result = await source_satellite.start_record_stream(wav_path)
  406. assert result is False
  407. @pytest.mark.asyncio
  408. async def test_start_record_stream_invalid_path(self, source_satellite, target_satellite):
  409. """Testet Fehler bei ungültigem Dateipfad."""
  410. source_satellite.send_command = AsyncMock(return_value=True)
  411. await source_satellite.start_stream(target_satellite)
  412. result = await source_satellite.start_record_stream("/nonexistent/dir/test.wav")
  413. assert result is False
  414. @pytest.mark.asyncio
  415. async def test_stop_record_stream_success(self, source_satellite, target_satellite, temp_dir, mock_event_manager):
  416. """Testet erfolgreiches Stoppen der Stream-Aufnahme."""
  417. source_satellite.send_command = AsyncMock(return_value=True)
  418. await source_satellite.start_stream(target_satellite)
  419. wav_path = os.path.join(str(temp_dir), "test_stop.wav")
  420. await source_satellite.start_record_stream(wav_path)
  421. mock_event_manager.trigger.reset_mock()
  422. result = await source_satellite.stop_record_stream()
  423. assert result is True
  424. assert source_satellite.is_stream_recording is False
  425. @pytest.mark.asyncio
  426. async def test_stop_record_stream_triggers_event(self, source_satellite, target_satellite, temp_dir, mock_event_manager):
  427. """Testet, dass stop_record_stream Event auslöst."""
  428. source_satellite.send_command = AsyncMock(return_value=True)
  429. await source_satellite.start_stream(target_satellite)
  430. wav_path = os.path.join(str(temp_dir), "test_stop_event.wav")
  431. await source_satellite.start_record_stream(wav_path)
  432. mock_event_manager.trigger.reset_mock()
  433. await source_satellite.stop_record_stream(reason="test_stop")
  434. mock_event_manager.trigger.assert_called_once()
  435. call_args = mock_event_manager.trigger.call_args
  436. assert call_args[0][0] == "satellite_stream_record_stopped"
  437. event_data = call_args[0][1]
  438. assert isinstance(event_data, SatelliteStreamRecordStopped)
  439. assert event_data.reason == "test_stop"
  440. assert event_data.filename == wav_path
  441. @pytest.mark.asyncio
  442. async def test_stop_record_stream_not_recording(self, source_satellite):
  443. """Testet Fehler wenn keine Stream-Aufnahme aktiv."""
  444. result = await source_satellite.stop_record_stream()
  445. assert result is False
  446. @pytest.mark.asyncio
  447. async def test_stop_record_stream_creates_valid_wav(self, source_satellite, target_satellite, temp_dir, mock_event_manager):
  448. """Testet, dass eine gültige WAV-Datei erstellt wird."""
  449. source_satellite.send_command = AsyncMock(return_value=True)
  450. await source_satellite.start_stream(target_satellite)
  451. wav_path = os.path.join(str(temp_dir), "valid_wav.wav")
  452. await source_satellite.start_record_stream(wav_path)
  453. # Audio-Daten simulieren (1s Stille bei 16KHz 16-bit mono)
  454. audio_chunk = b"\x00\x00" * 4096
  455. source_satellite._stream_recording_wav.writeframes(audio_chunk)
  456. source_satellite._stream_recording_bytes += len(audio_chunk)
  457. await source_satellite.stop_record_stream()
  458. # WAV-Datei prüfen
  459. assert os.path.exists(wav_path)
  460. with wave.open(wav_path, "rb") as wf:
  461. assert wf.getnchannels() == 1
  462. assert wf.getsampwidth() == 2
  463. assert wf.getframerate() == 16000
  464. @pytest.mark.asyncio
  465. async def test_stream_continues_after_record_stop(self, source_satellite, target_satellite, temp_dir):
  466. """Testet, dass der Stream nach Recording-Stop weiterläuft."""
  467. source_satellite.send_command = AsyncMock(return_value=True)
  468. await source_satellite.start_stream(target_satellite)
  469. wav_path = os.path.join(str(temp_dir), "continue.wav")
  470. await source_satellite.start_record_stream(wav_path)
  471. await source_satellite.stop_record_stream()
  472. # Stream sollte noch aktiv sein
  473. assert source_satellite.is_streaming is True
  474. assert source_satellite.is_stream_recording is False
  475. # =============================================================================
  476. # Tests für Audio-Handler
  477. # =============================================================================
  478. class TestStreamAudioHandler:
  479. """Tests für _on_stream_audio_received Handler."""
  480. @pytest.mark.asyncio
  481. async def test_audio_forwarded_to_target(self, source_satellite, target_satellite):
  482. """Testet, dass Audio zum Ziel-Satellite weitergeleitet wird."""
  483. source_satellite.send_command = AsyncMock(return_value=True)
  484. await source_satellite.start_stream(target_satellite)
  485. # Audio-Event simulieren
  486. audio_chunk = b"\x01\x02" * 2048
  487. event_data = MagicMock()
  488. event_data.satellite_id = "sat-source-001"
  489. event_data.audio_data = audio_chunk
  490. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  491. # Prüfe, dass Audio zum Ziel geschrieben wurde
  492. target_satellite.sockets.audio_out.write.assert_called_once_with(audio_chunk)
  493. target_satellite.sockets.audio_out.drain.assert_called_once()
  494. assert source_satellite._stream_bytes_forwarded == len(audio_chunk)
  495. @pytest.mark.asyncio
  496. async def test_audio_filtered_by_satellite_id(self, source_satellite, target_satellite):
  497. """Testet, dass Audio von anderen Satellites ignoriert wird."""
  498. source_satellite.send_command = AsyncMock(return_value=True)
  499. await source_satellite.start_stream(target_satellite)
  500. # Event von anderem Satellite
  501. event_data = MagicMock()
  502. event_data.satellite_id = "sat-other-999"
  503. event_data.audio_data = b"\x00" * 4096
  504. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  505. # Kein Write zum Ziel
  506. target_satellite.sockets.audio_out.write.assert_not_called()
  507. @pytest.mark.asyncio
  508. async def test_audio_ignored_when_not_streaming(self, source_satellite, target_satellite):
  509. """Testet, dass Audio ignoriert wird wenn kein Stream aktiv."""
  510. event_data = MagicMock()
  511. event_data.satellite_id = "sat-source-001"
  512. event_data.audio_data = b"\x00" * 4096
  513. # Kein aktiver Stream
  514. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  515. # Kein Crash, kein Write
  516. assert source_satellite._stream_bytes_forwarded == 0
  517. @pytest.mark.asyncio
  518. async def test_audio_empty_data_ignored(self, source_satellite, target_satellite):
  519. """Testet, dass leere Audio-Daten ignoriert werden."""
  520. source_satellite.send_command = AsyncMock(return_value=True)
  521. await source_satellite.start_stream(target_satellite)
  522. event_data = MagicMock()
  523. event_data.satellite_id = "sat-source-001"
  524. event_data.audio_data = b""
  525. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  526. target_satellite.sockets.audio_out.write.assert_not_called()
  527. @pytest.mark.asyncio
  528. async def test_target_disconnect_stops_stream(self, source_satellite, target_satellite, mock_event_manager):
  529. """Testet, dass Stream stoppt wenn Ziel-Socket weg ist."""
  530. source_satellite.send_command = AsyncMock(return_value=True)
  531. await source_satellite.start_stream(target_satellite)
  532. # Simuliere Ziel-Disconnect
  533. target_satellite.sockets.audio_out = None
  534. event_data = MagicMock()
  535. event_data.satellite_id = "sat-source-001"
  536. event_data.audio_data = b"\x00" * 4096
  537. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  538. assert source_satellite.is_streaming is False
  539. @pytest.mark.asyncio
  540. async def test_target_write_error_stops_stream(self, source_satellite, target_satellite, mock_event_manager):
  541. """Testet, dass Stream stoppt bei Socket-Write-Fehler."""
  542. source_satellite.send_command = AsyncMock(return_value=True)
  543. await source_satellite.start_stream(target_satellite)
  544. # Simuliere Write-Fehler
  545. target_satellite.sockets.audio_out.write.side_effect = ConnectionError("Verbindung verloren")
  546. event_data = MagicMock()
  547. event_data.satellite_id = "sat-source-001"
  548. event_data.audio_data = b"\x00" * 4096
  549. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  550. assert source_satellite.is_streaming is False
  551. @pytest.mark.asyncio
  552. async def test_audio_also_written_to_recording(self, source_satellite, target_satellite, temp_dir):
  553. """Testet paralleles Schreiben auf Disk während Streaming."""
  554. source_satellite.send_command = AsyncMock(return_value=True)
  555. await source_satellite.start_stream(target_satellite)
  556. wav_path = os.path.join(str(temp_dir), "parallel.wav")
  557. await source_satellite.start_record_stream(wav_path)
  558. # Mock writeframes um Aufruf zu prüfen
  559. mock_wav = MagicMock()
  560. source_satellite._stream_recording_wav = mock_wav
  561. audio_chunk = b"\x01\x02" * 2048
  562. event_data = MagicMock()
  563. event_data.satellite_id = "sat-source-001"
  564. event_data.audio_data = audio_chunk
  565. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  566. # Audio wurde zum Ziel UND zur WAV-Datei geschrieben
  567. target_satellite.sockets.audio_out.write.assert_called_once_with(audio_chunk)
  568. mock_wav.writeframes.assert_called_once_with(audio_chunk)
  569. assert source_satellite._stream_recording_bytes == len(audio_chunk)
  570. @pytest.mark.asyncio
  571. async def test_dict_event_data_supported(self, source_satellite, target_satellite):
  572. """Testet, dass dict-basierte Event-Daten unterstützt werden."""
  573. source_satellite.send_command = AsyncMock(return_value=True)
  574. await source_satellite.start_stream(target_satellite)
  575. audio_chunk = b"\x03\x04" * 1024
  576. event_data = {
  577. "satellite_id": "sat-source-001",
  578. "audio_data": audio_chunk,
  579. }
  580. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  581. target_satellite.sockets.audio_out.write.assert_called_once_with(audio_chunk)
  582. @pytest.mark.asyncio
  583. async def test_bytes_forwarded_accumulates(self, source_satellite, target_satellite):
  584. """Testet, dass bytes_forwarded korrekt aufsummiert wird."""
  585. source_satellite.send_command = AsyncMock(return_value=True)
  586. await source_satellite.start_stream(target_satellite)
  587. chunk1 = b"\x00" * 4096
  588. chunk2 = b"\x01" * 2048
  589. for chunk in (chunk1, chunk2):
  590. event_data = MagicMock()
  591. event_data.satellite_id = "sat-source-001"
  592. event_data.audio_data = chunk
  593. await source_satellite._on_stream_audio_received("audio_input_received", event_data)
  594. assert source_satellite._stream_bytes_forwarded == 4096 + 2048
  595. # =============================================================================
  596. # Tests für Wakeword-Stop Handler
  597. # =============================================================================
  598. class TestStreamRecordStoppedHandler:
  599. """Tests für _on_stream_record_stopped_event Handler."""
  600. @pytest.mark.asyncio
  601. async def test_wakeword_stops_stream(self, source_satellite, target_satellite, mock_event_manager):
  602. """Testet, dass Wakeword-Stop den Stream beendet."""
  603. source_satellite.send_command = AsyncMock(return_value=True)
  604. await source_satellite.start_stream(target_satellite)
  605. event_data = MagicMock()
  606. event_data.satellite_id = "sat-source-001"
  607. event_data.reason = "wakeword"
  608. await source_satellite._on_stream_record_stopped_event("satellite_record_stopped", event_data)
  609. assert source_satellite.is_streaming is False
  610. @pytest.mark.asyncio
  611. async def test_handler_ignores_other_satellites(self, source_satellite, target_satellite):
  612. """Testet, dass Handler Events anderer Satellites ignoriert."""
  613. source_satellite.send_command = AsyncMock(return_value=True)
  614. await source_satellite.start_stream(target_satellite)
  615. event_data = MagicMock()
  616. event_data.satellite_id = "sat-other-999"
  617. event_data.reason = "wakeword"
  618. await source_satellite._on_stream_record_stopped_event("satellite_record_stopped", event_data)
  619. # Stream sollte noch laufen
  620. assert source_satellite.is_streaming is True
  621. @pytest.mark.asyncio
  622. async def test_handler_ignores_when_not_streaming(self, source_satellite):
  623. """Testet, dass Handler ignoriert wenn kein Stream aktiv."""
  624. event_data = MagicMock()
  625. event_data.satellite_id = "sat-source-001"
  626. event_data.reason = "wakeword"
  627. # Kein Crash
  628. await source_satellite._on_stream_record_stopped_event("satellite_record_stopped", event_data)
  629. @pytest.mark.asyncio
  630. async def test_handler_with_dict_event(self, source_satellite, target_satellite, mock_event_manager):
  631. """Testet Handler mit dict-basiertem Event."""
  632. source_satellite.send_command = AsyncMock(return_value=True)
  633. await source_satellite.start_stream(target_satellite)
  634. event_data = {
  635. "satellite_id": "sat-source-001",
  636. "reason": "wakeword",
  637. }
  638. await source_satellite._on_stream_record_stopped_event("satellite_record_stopped", event_data)
  639. assert source_satellite.is_streaming is False
  640. # =============================================================================
  641. # Tests für Timeout-Handler
  642. # =============================================================================
  643. class TestStreamTimeout:
  644. """Tests für _on_stream_timeout Handler."""
  645. @pytest.mark.asyncio
  646. async def test_timeout_stops_stream(self, source_satellite, target_satellite, mock_event_manager):
  647. """Testet, dass Timeout den Stream beendet."""
  648. source_satellite.send_command = AsyncMock(return_value=True)
  649. await source_satellite.start_stream(target_satellite)
  650. await source_satellite._on_stream_timeout()
  651. assert source_satellite.is_streaming is False
  652. @pytest.mark.asyncio
  653. async def test_timeout_reason_in_event(self, source_satellite, target_satellite, mock_event_manager):
  654. """Testet, dass der Timeout-Grund im Event steht."""
  655. source_satellite.send_command = AsyncMock(return_value=True)
  656. await source_satellite.start_stream(target_satellite)
  657. mock_event_manager.trigger.reset_mock()
  658. await source_satellite._on_stream_timeout()
  659. event_data = mock_event_manager.trigger.call_args[0][1]
  660. assert event_data.reason == "timeout"
  661. @pytest.mark.asyncio
  662. async def test_timeout_ignored_when_not_streaming(self, source_satellite):
  663. """Testet, dass Timeout ignoriert wird wenn kein Stream aktiv."""
  664. # Kein Crash
  665. await source_satellite._on_stream_timeout()
  666. assert source_satellite.is_streaming is False
  667. # =============================================================================
  668. # Tests für Disconnect-Guard
  669. # =============================================================================
  670. class TestDisconnectGuard:
  671. """Tests für Streaming-Bereinigung bei Disconnect."""
  672. @pytest.mark.asyncio
  673. async def test_disconnect_stops_stream(self, source_satellite, target_satellite, mock_event_manager):
  674. """Testet, dass disconnect() aktiven Stream stoppt."""
  675. source_satellite.send_command = AsyncMock(return_value=True)
  676. await source_satellite.start_stream(target_satellite)
  677. await source_satellite.disconnect("test")
  678. assert source_satellite.is_streaming is False
  679. assert source_satellite.state == ConnectionState.DISCONNECTED
  680. # =============================================================================
  681. # Tests für Mutual Exclusion (Recording blockiert bei Stream)
  682. # =============================================================================
  683. class TestMutualExclusion:
  684. """Tests für gegenseitigen Ausschluss von Recording und Streaming."""
  685. @pytest.mark.asyncio
  686. async def test_recording_blocked_during_stream(self, source_satellite, target_satellite):
  687. """Testet, dass start_recording bei aktivem Stream fehlschlägt."""
  688. source_satellite.send_command = AsyncMock(return_value=True)
  689. await source_satellite.start_stream(target_satellite)
  690. result = await source_satellite.start_recording(timeout=60.0)
  691. assert result is False
  692. assert source_satellite.is_streaming is True
  693. @pytest.mark.asyncio
  694. async def test_stream_stops_recording(self, source_satellite, target_satellite, mock_event_manager):
  695. """Testet, dass start_stream laufende Aufnahme stoppt."""
  696. source_satellite.send_command = AsyncMock(return_value=True)
  697. # Recording starten
  698. await source_satellite.start_recording(timeout=60.0)
  699. assert source_satellite.is_recording is True
  700. # Stream starten — Recording sollte gestoppt werden
  701. result = await source_satellite.start_stream(target_satellite)
  702. assert result is True
  703. assert source_satellite.is_streaming is True
  704. assert source_satellite.is_recording is False
  705. # =============================================================================
  706. # Integration Tests mit echtem EventManager
  707. # =============================================================================
  708. class TestStreamingIntegration:
  709. """Integrationstests mit echtem EventManager."""
  710. @pytest.mark.asyncio
  711. async def test_full_stream_flow(self):
  712. """Testet den vollständigen Stream-Flow mit echtem EventManager."""
  713. from trixy_core.events.eventmanager import EventManager
  714. app = MagicMock()
  715. event_manager = EventManager(app)
  716. app.events = event_manager
  717. # Zwei Satellites erstellen
  718. source = Satellite(
  719. satellite_id="int-source",
  720. alias="Quelle",
  721. application=app,
  722. )
  723. source.set_connected()
  724. source.send_command = AsyncMock(return_value=True)
  725. target = Satellite(
  726. satellite_id="int-target",
  727. alias="Ziel",
  728. application=app,
  729. )
  730. target.set_connected()
  731. mock_writer = MagicMock()
  732. mock_writer.write = MagicMock()
  733. mock_writer.drain = AsyncMock()
  734. target.sockets.audio_out = mock_writer
  735. # Events sammeln
  736. stream_events = []
  737. async def on_stream_started(event_name, data):
  738. stream_events.append(("started", data))
  739. async def on_stream_stopped(event_name, data):
  740. stream_events.append(("stopped", data))
  741. event_manager.register("satellite_stream_started", on_stream_started)
  742. event_manager.register("satellite_stream_stopped", on_stream_stopped)
  743. # Stream starten
  744. result = await source.start_stream(target)
  745. assert result is True
  746. assert len(stream_events) == 1
  747. assert stream_events[0][0] == "started"
  748. # Stream stoppen
  749. await source.stop_stream(reason="test_complete")
  750. assert len(stream_events) == 2
  751. assert stream_events[1][0] == "stopped"
  752. assert stream_events[1][1].reason == "test_complete"
  753. @pytest.mark.asyncio
  754. async def test_full_stream_with_recording_flow(self, temp_dir):
  755. """Testet Stream + Recording Flow mit echtem EventManager."""
  756. from trixy_core.events.eventmanager import EventManager
  757. app = MagicMock()
  758. event_manager = EventManager(app)
  759. app.events = event_manager
  760. source = Satellite(
  761. satellite_id="rec-source",
  762. alias="Quelle",
  763. application=app,
  764. )
  765. source.set_connected()
  766. source.send_command = AsyncMock(return_value=True)
  767. target = Satellite(
  768. satellite_id="rec-target",
  769. alias="Ziel",
  770. application=app,
  771. )
  772. target.set_connected()
  773. mock_writer = MagicMock()
  774. mock_writer.write = MagicMock()
  775. mock_writer.drain = AsyncMock()
  776. target.sockets.audio_out = mock_writer
  777. # Events sammeln
  778. record_events = []
  779. async def on_record_started(event_name, data):
  780. record_events.append(("started", data))
  781. async def on_record_stopped(event_name, data):
  782. record_events.append(("stopped", data))
  783. event_manager.register("satellite_stream_record_started", on_record_started)
  784. event_manager.register("satellite_stream_record_stopped", on_record_stopped)
  785. # Stream starten
  786. await source.start_stream(target)
  787. # Recording starten
  788. wav_path = os.path.join(str(temp_dir), "integration.wav")
  789. await source.start_record_stream(wav_path)
  790. assert len(record_events) == 1
  791. assert record_events[0][0] == "started"
  792. # Recording stoppen (Stream läuft weiter)
  793. await source.stop_record_stream()
  794. assert len(record_events) == 2
  795. assert record_events[1][0] == "stopped"
  796. assert source.is_streaming is True
  797. # Stream stoppen
  798. await source.stop_stream()
  799. assert source.is_streaming is False
  800. @pytest.mark.asyncio
  801. async def test_audio_forwarding_integration(self):
  802. """Testet die Audio-Weiterleitung mit echtem EventManager."""
  803. from trixy_core.events.eventmanager import EventManager
  804. app = MagicMock()
  805. event_manager = EventManager(app)
  806. app.events = event_manager
  807. source = Satellite(
  808. satellite_id="fwd-source",
  809. alias="Quelle",
  810. application=app,
  811. )
  812. source.set_connected()
  813. source.send_command = AsyncMock(return_value=True)
  814. target = Satellite(
  815. satellite_id="fwd-target",
  816. alias="Ziel",
  817. application=app,
  818. )
  819. target.set_connected()
  820. mock_writer = MagicMock()
  821. mock_writer.write = MagicMock()
  822. mock_writer.drain = AsyncMock()
  823. target.sockets.audio_out = mock_writer
  824. # Stream starten
  825. await source.start_stream(target)
  826. # Audio-Event simulieren via EventManager (mit EventData-Objekt)
  827. from trixy_core.events.event_data import RawAudioInputReceived
  828. audio_chunk = b"\xAB\xCD" * 2048
  829. await event_manager.trigger("audio_input_received", RawAudioInputReceived(
  830. satellite_id="fwd-source",
  831. audio_data=audio_chunk,
  832. ))
  833. # Audio sollte zum Ziel weitergeleitet worden sein
  834. mock_writer.write.assert_called_once_with(audio_chunk)
  835. await source.stop_stream()