Skip to content

Commit 8d659c6

Browse files
committed
test: cover replay_events_after None-message branch (135->132)
1 parent 0124539 commit 8d659c6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/shared/test_streamable_http.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2312,12 +2312,13 @@ async def test_replay_events_after_replays_messages():
23122312
stream_id = "stream-1"
23132313
msg1: types.JSONRPCMessage = types.JSONRPCResponse(jsonrpc="2.0", id="1", result={"first": True})
23142314
msg2: types.JSONRPCMessage = types.JSONRPCResponse(jsonrpc="2.0", id="2", result={"second": True})
2315-
# Store: priming event (None), then two real messages
2315+
# Store: priming event (None), real message, another None (priming), then real message
23162316
eid0 = await store.store_event(stream_id, None)
23172317
eid1 = await store.store_event(stream_id, msg1)
2318+
_eid_none = await store.store_event(stream_id, None)
23182319
eid2 = await store.store_event(stream_id, msg2)
23192320

2320-
# Replay after priming event — should get both messages
2321+
# Replay after priming event — should get only real messages, skipping None
23212322
replayed: list[EventMessage] = []
23222323

23232324
async def callback(event_msg: EventMessage) -> None:

0 commit comments

Comments
 (0)