feat(nodes): flag missing local models at queue and load time (#1057)

tests/routes/test_models_changed_broadcast.py

@@ -0,0 +1,40 @@
+"""Tests for the models_changed WebSocket broadcast helper."""
+
+import asyncio
+
+import pytest
+
+from py.routes.handlers import model_handlers
+
+
+@pytest.mark.asyncio
+async def test_broadcast_models_changed_payload(monkeypatch):
+    """The helper must broadcast the models_changed event to all clients."""
+    sent = []
+
+    async def fake_broadcast(data):
+        sent.append(data)
+
+    monkeypatch.setattr(
+        "py.services.websocket_manager.ws_manager.broadcast", fake_broadcast
+    )
+    # Give the create_task a chance to run.
+    for _ in range(10):
+        await asyncio.sleep(0)
+
+    model_handlers._broadcast_models_changed()
+    await asyncio.sleep(0)
+    assert sent == [{"type": "models_changed"}]
+
+
+@pytest.mark.asyncio
+async def test_broadcast_models_changed_survives_broadcast_failure(monkeypatch):
+    """A failing broadcast must not raise out of the mutation handler."""
+    async def fake_broadcast(data):
+        raise RuntimeError("socket closed")
+
+    monkeypatch.setattr(
+        "py.services.websocket_manager.ws_manager.broadcast", fake_broadcast
+    )
+    model_handlers._broadcast_models_changed()
+    await asyncio.sleep(0)