Skip to content

Commit 81145c9

Browse files
more gemini suggestions
1 parent 1da4886 commit 81145c9

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/server/events/test_event_queue.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -298,6 +298,7 @@ async def test_close_sets_flag_and_handles_internal_queue_new_python(
298298
queue.shutdown = MagicMock() # type: ignore[attr-defined]
299299
await event_queue.close()
300300
assert event_queue.is_closed() is True
301+
queue.shutdown.assert_called_once_with(False)
301302

302303

303304
@pytest.mark.asyncio
@@ -366,9 +367,11 @@ async def test_close_idempotent(event_queue: EventQueue) -> None:
366367
queue.shutdown = MagicMock() # type: ignore[attr-defined]
367368
await event_queue_new.close()
368369
assert event_queue_new.is_closed() is True
370+
queue.shutdown.assert_called_once()
369371

370372
await event_queue_new.close()
371373
assert event_queue_new.is_closed() is True
374+
queue.shutdown.assert_called_once() # Still only called once
372375

373376

374377
@pytest.mark.asyncio

0 commit comments

Comments
 (0)