File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed
Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments