diff options
Diffstat (limited to 'packages/meshbay-node/tests/test_reply_correlation.py')
| -rw-r--r-- | packages/meshbay-node/tests/test_reply_correlation.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/packages/meshbay-node/tests/test_reply_correlation.py b/packages/meshbay-node/tests/test_reply_correlation.py index bfb336e..a97ec1e 100644 --- a/packages/meshbay-node/tests/test_reply_correlation.py +++ b/packages/meshbay-node/tests/test_reply_correlation.py @@ -28,6 +28,7 @@ import base64 import msgpack import pytest from meshbay_common.protocol import MNP +from meshbay_node.transport.webrtc.channel import _REPLY_TO from meshbay_node.transport.webrtc_server import WebRTCPeerSession pytestmark = pytest.mark.asyncio @@ -163,3 +164,29 @@ async def test_work_handed_to_a_task_still_answers_under_the_right_id(): by_id = {m["req_id"]: m["detail"] for m in s._channel.sent} assert by_id == {11: "first", 12: "second"}, ( "a late reply answered under whichever request arrived most recently") + + +async def test_a_new_chat_epoch_answers_no_request_not_even_the_askers(): + """Revoking a device or removing a member opens a chat epoch, announced to + everyone connected — the asker included. Stamped with the asker's request, + that notice arrived as the answer to `device_revoke` before the real one.""" + asker, other = _session("asker"), _session("other") + asker._ctx = {"groups": {"g": {"_peers": {"a": asker, "b": other}}}} + asker._audit = lambda *a, **k: None + + async def run_op(fn, *args): + return {"epoch": 2} + asker._run_op = run_op + + token = _REPLY_TO.set((asker, 9)) # inside the handler of request 9 + try: + await asker._new_chat_epoch("g", "device_revoke") + asker._send({"type": MNP.DEVICE_ADD_ACK}) + finally: + _REPLY_TO.reset(token) + + notice, reply = asker._channel.sent + assert notice["type"] == MNP.CHAT_EPOCH_ACK and "req_id" not in notice + assert reply["type"] == MNP.DEVICE_ADD_ACK and reply["req_id"] == 9 + (seen,) = other._channel.sent + assert "req_id" not in seen |