Fix DM presence indicator never updating live (#63)

The sidebar shows every DM's online/offline dot at once, but the only
existing signal for a presence change (member_updated) is broadcast to
a room's own channel, which Presence only delivers to a connection that
currently has that specific room joined -- never true for a DM sitting
unopened in the sidebar. Add a dedicated per-user broadcast
(dm_presence_update) sent to each of a user's DM partners on their own
per-user channel whenever their global online/offline state changes, so
the sidebar dot updates without needing that DM to be the open room.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
2026-08-27 20:41:57 -06:00
co-authored by Claude Sonnet 5
parent d42bf114dd
commit 4cc3823adf
7 changed files with 142 additions and 5 deletions
+9 -5
View File
@@ -8,13 +8,17 @@ def _unique(prefix: str) -> str:
return f"{prefix}-{uuid.uuid4().hex[:8]}"
_NOISE_TYPES = {"member_updated", "dm_presence_update"}
def _recv(ws) -> dict:
"""Reads the next frame, transparently discarding member_updated
presence-change broadcasts -- another connection in the same room going
online/offline is real, expected noise these tests aren't about."""
"""Reads the next frame, transparently discarding presence-change
broadcasts (member_updated, and #63's dm_presence_update) -- another
connection sharing a room or a DM going online/offline is real,
expected noise these tests aren't about."""
while True:
msg = ws.receive_json()
if msg.get("type") != "member_updated":
if msg.get("type") not in _NOISE_TYPES:
return msg
@@ -201,7 +205,7 @@ def test_new_message_notifies_recipient_who_hid_the_dm_via_websocket(ws_client_f
alice_ws.send_json({"type": "join", "room_id": room["id"]})
assert alice_ws.receive_json()["type"] == "joined"
received = bob_ws.receive_json()
received = _recv(bob_ws)
assert received == {"type": "room_added", "room_id": room["id"]}
+65
View File
@@ -275,3 +275,68 @@ def test_new_message_unhides_dm_for_both_participants(ws_client):
"/api/auth/login", json={"username_or_email": bob["username"], "password": "password123"}
)
assert any(r["id"] == dm["id"] for r in ws_client.get("/api/rooms/mine").json())
def test_dm_partner_presence_update_delivered_without_room_open(ws_client_factory):
# #63: alice never joins the DM's own room channel anywhere in this
# test -- exactly the normal state for a DM sitting in the sidebar that
# isn't the currently open room. member_updated's room-channel broadcast
# would never reach her in that state (Presence gates it on having that
# specific room joined); this signal has to arrive on her own per-user
# channel instead, same as room_added.
#
# Only the connect ("online") side is exercised here, not disconnect --
# see test_presence.py's module docstring for why the offline half
# isn't reliably testable via a `with websocket_connect(...)` block
# closing (TestClient cancels the server task rather than delivering a
# real disconnect, which can interrupt a `finally` block's own awaits
# in tests only, never in production).
instance1 = ws_client_factory()
instance2 = ws_client_factory()
alice = _register_ws(instance1, _unique("alice"))
bob = _register_ws(instance2, _unique("bob"))
instance1.post("/api/rooms/dm", json={"other_user_id": bob["id"]})
with instance1.websocket_connect("/ws/chat") as alice_ws:
# Sync barrier: alice's own websocket_connect() returning only
# proves the handshake completed, not that chat.py's connection
# setup (register_user, in particular -- required before bob's
# connect can reach her per-user channel at all) has finished.
# Any reply -- even an error -- proves the connection has reached
# its main frame loop, which setup always completes before.
alice_ws.send_json({"type": "__sync_barrier__"})
assert alice_ws.receive_json()["type"] == "error"
with instance2.websocket_connect("/ws/chat"):
online_update = alice_ws.receive_json()
assert online_update == {
"type": "dm_presence_update",
"user_id": bob["id"],
"status": "online",
}
def test_dm_presence_update_not_sent_to_non_partner(ws_client_factory):
instance1 = ws_client_factory()
instance2 = ws_client_factory()
instance3 = ws_client_factory()
alice = _register_ws(instance1, _unique("alice"))
bob = _register_ws(instance2, _unique("bob"))
_register_ws(instance3, _unique("outsider"))
instance1.post("/api/rooms/dm", json={"other_user_id": bob["id"]})
with instance3.websocket_connect("/ws/chat") as outsider_ws:
with instance2.websocket_connect("/ws/chat"):
pass
# Nothing should ever arrive for an outsider who shares no DM with
# bob. Prove the socket stayed quiet the same way
# test_desktop_notifications.py's non-member test does: a harmless
# self-targeted join, whose prompt "joined" ack proves nothing else
# was already queued ahead of it.
room = instance3.post("/api/rooms", json={"name": _unique("outsiders-room")}).json()
outsider_ws.send_json({"type": "join", "room_id": room["id"]})
joined = outsider_ws.receive_json()
assert joined == {"type": "joined", "room_id": room["id"]}