fix: do not echo back ws messages

This commit is contained in:
Vlad Stan 2026-07-17 11:07:06 +03:00
parent 486426e453
commit 65e43a25b3
2 changed files with 4 additions and 2 deletions

View file

@ -158,6 +158,8 @@ class WasmExtensionWebsocketHub:
data: str,
) -> None:
for active_conn in self.get_connections(conn.extension_id, conn.item_id):
if active_conn.websocket == conn.websocket:
continue
await active_conn.websocket.send_text(data)

View file

@ -90,7 +90,7 @@ async def test_wasm_extension_websocket_hub_rate_limits_per_channel():
@pytest.mark.anyio
async def test_wasm_extension_websocket_hub_rebroadcasts_client_messages():
async def test_wasm_extension_websocket_hub_rebroadcasts_client_messages_to_peers():
hub = WasmExtensionWebsocketHub()
sender = FakeWebSocket(received=['{"type":"input","paddle":0.5}'])
peer = FakeWebSocket()
@ -100,5 +100,5 @@ async def test_wasm_extension_websocket_hub_rebroadcasts_client_messages():
await hub.listen(conn)
assert sender.sent == ['{"type":"input","paddle":0.5}']
assert sender.sent == []
assert peer.sent == ['{"type":"input","paddle":0.5}']