GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-09-05 23:47:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(agent): verify snapshot survives stream close

Browse Source
This commit is contained in:
InfinityPacer
2026-08-23 11:43:07 +08:00
parent 9c4bead782
commit d724dfa3f9
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_web_agent_stream.py
+4 -2
View File
@@ -1355,9 +1355,10 @@ def test_web_agent_traditional_stream_keeps_alive_and_saves_after_done():
await asyncio.sleep(0.001) await asyncio.sleep(0.001)
assert snapshot_started.is_set() assert snapshot_started.is_set()
assert not snapshot_finished.is_set() assert not snapshot_finished.is_set()
await iterator.aclose()
assert not snapshot_finished.is_set()
snapshot_release.set() snapshot_release.set()
await asyncio.to_thread(snapshot_finished.wait, 1) await asyncio.to_thread(snapshot_finished.wait, 1)
await iterator.aclose()
return "".join(received) return "".join(received)
try: try:
@@ -1432,9 +1433,10 @@ def test_web_agent_stream_sends_done_before_snapshot_persistence_finishes():
assert snapshot_started.is_set() assert snapshot_started.is_set()
assert not snapshot_finished.is_set() assert not snapshot_finished.is_set()
await iterator.aclose()
assert not snapshot_finished.is_set()
snapshot_release.set() snapshot_release.set()
await asyncio.to_thread(snapshot_finished.wait, 1) await asyncio.to_thread(snapshot_finished.wait, 1)
await iterator.aclose()
return "".join(received) return "".join(received)
try: try: