From 74e349be9708cc09c6b0a807c755cc6f5cb20d5a Mon Sep 17 00:00:00 2001 From: Matthew Wolter Date: Sun, 12 Apr 2026 07:00:10 -0700 Subject: [PATCH] Fix test_r03 resolving mock to include expected_ack payload send_binary_req reads result.payload['expected_ack'] from MSG_SENT events. The resolving subscribe must provide that key for MSG_SENT event types to avoid KeyError. --- tests/unit/test_g7_standalone_bugs_and_cleanup.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_g7_standalone_bugs_and_cleanup.py b/tests/unit/test_g7_standalone_bugs_and_cleanup.py index bae056c..352c7e6 100644 --- a/tests/unit/test_g7_standalone_bugs_and_cleanup.py +++ b/tests/unit/test_g7_standalone_bugs_and_cleanup.py @@ -177,12 +177,14 @@ async def test_r03_placeholder_registered_before_send(): return_value=Event(EventType.MSG_SENT, {"expected_ack": b"\x01\x02\x03\x04"}) ) - # Resolve subscribed events immediately so send() doesn't block + # Resolve subscribed events immediately so send() doesn't block. + # Use MSG_SENT with expected_ack because send_binary_req reads that key. def resolving_subscribe(event_type, cb, attribute_filters=None): sub = MagicMock() sub.unsubscribe = MagicMock() + payload = {"expected_ack": b"\x01\x02\x03\x04"} if event_type == EventType.MSG_SENT else {} asyncio.get_event_loop().call_soon( - cb, Event(event_type, {}) + cb, Event(event_type, payload) ) return sub handler.dispatcher.subscribe = MagicMock(side_effect=resolving_subscribe)