From d724dfa3f997d51754d75724e884e626756dc4de Mon Sep 17 00:00:00 2001 From: InfinityPacer Date: Sun, 23 Aug 2026 03:07:40 +0800 Subject: [PATCH] test(agent): verify snapshot survives stream close --- tests/test_web_agent_stream.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_web_agent_stream.py b/tests/test_web_agent_stream.py index 01c0bd447..e13dc0ca9 100644 --- a/tests/test_web_agent_stream.py +++ b/tests/test_web_agent_stream.py @@ -1355,9 +1355,10 @@ def test_web_agent_traditional_stream_keeps_alive_and_saves_after_done(): await asyncio.sleep(0.001) assert snapshot_started.is_set() assert not snapshot_finished.is_set() + await iterator.aclose() + assert not snapshot_finished.is_set() snapshot_release.set() await asyncio.to_thread(snapshot_finished.wait, 1) - await iterator.aclose() return "".join(received) try: @@ -1432,9 +1433,10 @@ def test_web_agent_stream_sends_done_before_snapshot_persistence_finishes(): assert snapshot_started.is_set() assert not snapshot_finished.is_set() + await iterator.aclose() + assert not snapshot_finished.is_set() snapshot_release.set() await asyncio.to_thread(snapshot_finished.wait, 1) - await iterator.aclose() return "".join(received) try: