From 52999149640659344155cbfe6ded6634c3511e9e Mon Sep 17 00:00:00 2001
From: "devin-ai-integration[bot]"
 <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 10 Nov 2026 18:31:57 +0000
Subject: [PATCH] test(http2): drain client traffic after sending GOAWAY
 (#50972)

Co-authored-by: mateo <277851410+mateo-berri@users.noreply.github.com>
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 tests/integration/sdk/test_http2_wire.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/integration/sdk/test_http2_wire.py b/tests/integration/sdk/test_http2_wire.py
index a263472a45f..30a0635f3b8 100644
--- a/tests/integration/sdk/test_http2_wire.py
+++ b/tests/integration/sdk/test_http2_wire.py
@@ -531,6 +531,9 @@ def _serve_h2(context: ssl.SSLContext, raw: socket.socket, drain: bool) -> None:
         tls.sendall(connection.data_to_send())
         while data := tls.recv(65535):
             if _reply_to_ended_streams(connection, tls, data, drain) and drain:
+                tls.shutdown(socket.SHUT_WR)
+                while tls.recv(65535):
+                    pass
                 return