first commit

tests/livekit/test_bridge_gateway.py

@@ -0,0 +1,68 @@
+from __future__ import annotations
+
+import asyncio
+import json
+
+from app.livekit.adapters.bridge_gateway import BridgeGateway
+
+
+class _Participant:
+    def __init__(self) -> None:
+        self.calls = []
+
+    async def publish_data(self, payload, **kwargs) -> None:
+        self.calls.append((json.loads(payload), kwargs))
+
+
+class _Room:
+    name = "room-load-1"
+
+    def __init__(self) -> None:
+        self.local_participant = _Participant()
+
+
+def test_publish_debug_event_targets_originating_bridge() -> None:
+    async def _run() -> None:
+        room = _Room()
+        gateway = BridgeGateway(
+            room=room,
+            bridge_identity="bridge-load-1",
+            protocol="LOAD-1",
+            stress_test=True,
+        )
+
+        await gateway.publish_debug_event(
+            "stt.completed", duration_ms=123, text="texto reconhecido"
+        )
+
+        payload, kwargs = room.local_participant.calls[0]
+        assert payload["type"] == "debug_event"
+        assert payload["event"] == "stt.completed"
+        assert payload["stress_test"] is True
+        assert payload["data"] == {
+            "duration_ms": 123,
+            "text": "texto reconhecido",
+        }
+        assert kwargs == {
+            "reliable": True,
+            "destination_identities": ["bridge-load-1"],
+            "topic": "agent.debug",
+        }
+
+    asyncio.run(_run())
+
+
+def test_publish_debug_event_is_suppressed_outside_stress_test() -> None:
+    async def _run() -> None:
+        room = _Room()
+        gateway = BridgeGateway(
+            room=room,
+            bridge_identity="bridge-regular-1",
+            protocol="REGULAR-1",
+        )
+
+        await gateway.publish_debug_event("stt.completed", duration_ms=123)
+
+        assert room.local_participant.calls == []
+
+    asyncio.run(_run())