Coverage

Author: gjmooney

tests/test_comm_awareness.py

@@ -3,6 +3,7 @@
 from unittest.mock import patch
 
 import pytest
+from anyio import sleep
 from pycrdt import Doc, YMessageType, create_awareness_message
 
 from ypywidgets.comm import CommProvider, CommWidget
@@ -67,3 +68,27 @@ def test_comm_widget_awareness_observe_and_unobserve(mock_create_comm):
     events.clear()
     widget.awareness.set_local_state({"ping": 2})
     assert events == []
+
+
+async def test_remote_manager_applies_awareness_messages(synced_widgets, context):
+    async with context:
+        local_widget = await synced_widgets.get_local_widget()
+        await synced_widgets.get_remote_widget()
+
+        local_widget.awareness.set_local_state({"role": "local"})
+        payload = local_widget.awareness.encode_awareness_update(
+            [local_widget.awareness.client_id]
+        )
+        frame = create_awareness_message(payload)
+
+        synced_widgets.comm.send_send_stream.send_nowait(
+            ("comm_msg", {}, None, [frame], None, None)
+        )
+        await sleep(0.01)
+
+        assert synced_widgets._remote_awareness is not None
+        remote_state = synced_widgets._remote_awareness.states.get(
+            local_widget.awareness.client_id
+        )
+        assert remote_state is not None
+        assert remote_state.get("role") == "local"