Merge pull request #14 from sacha-development-stuff/codex/fix-notification-assertion-error-in-test_streamablehttp_clie

Fix race in resumption test

Author: SoldierSacha

tests/shared/test_streamable_http.py

@@ -1156,6 +1156,12 @@ async def run_tool():
             assert result.content[0].type == "text"
             assert "Completed" in result.content[0].text
 
+            # Allow any pending notifications to be processed
+            for _ in range(50):
+                if captured_notifications:
+                    break
+                await anyio.sleep(0.1)
+
             # We should have received the remaining notifications
             assert len(captured_notifications) > 0