fix sync do_request().stream

kevin1024 · Dec 15, 2023 · 356ff41 · 356ff41
1 parent cf76592
commit 356ff41
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/integration/test_httpx.py b/tests/integration/test_httpx.py
@@ -51,8 +51,14 @@ def __call__(self, *args, **kwargs):
             return self.client.request(*args, timeout=60, **kwargs)
 
     def stream(self, *args, **kwargs):
-        with self.client.stream(*args, **kwargs) as response:
-            return b"".join(response.iter_bytes())
+        if hasattr(self, "_client"):
+            with self.client.stream(*args, **kwargs) as response:
+                return b"".join(response.iter_bytes())
+
+        # Use one-time context and dispose of the client afterwards
+        with self:
+            with self.client.stream(*args, **kwargs) as response:
+                return b"".join(response.iter_bytes())
 
 
 class DoAsyncRequest(BaseDoRequest):