66 lines
2.5 KiB
Diff
66 lines
2.5 KiB
Diff
|
From e078b83c2dbd7fddc88d9a698479b91e78eae964 Mon Sep 17 00:00:00 2001
|
||
|
|
From: Illia Volochii <illia.volochii@gmail.com>
|
||
|
|
Date: Wed, 7 Jan 2026 18:07:30 +0200
|
||
|
|
Subject: [PATCH] Merge commit from fork
|
||
|
|
|
||
|
|
* Stop decoding response content during redirects needlessly
|
||
|
|
|
||
|
|
* Rename the new query parameter
|
||
|
|
|
||
|
|
* Add a changelog entry
|
||
|
|
---
|
||
|
|
src/urllib3/response.py | 6 +++++-
|
||
|
|
test/with_dummyserver/test_connectionpool.py | 19 +++++++++++++++++++
|
||
|
|
2 files changed, 24 insertions(+), 1 deletion(-)
|
||
|
|
|
||
|
|
diff --git a/src/urllib3/response.py b/src/urllib3/response.py
|
||
|
|
index 37936f93..fbbccdbd 100644
|
||
|
|
--- a/src/urllib3/response.py
|
||
|
|
+++ b/src/urllib3/response.py
|
||
|
|
@@ -610,7 +610,11 @@ class HTTPResponse(BaseHTTPResponse):
|
||
|
|
Unread data in the HTTPResponse connection blocks the connection from being released back to the pool.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
- self.read()
|
||
|
|
+ self.read(
|
||
|
|
+ # Do not spend resources decoding the content unless
|
||
|
|
+ # decoding has already been initiated.
|
||
|
|
+ decode_content=self._has_decoded_content,
|
||
|
|
+ )
|
||
|
|
except (HTTPError, OSError, BaseSSLError, HTTPException):
|
||
|
|
pass
|
||
|
|
|
||
|
|
diff --git a/test/with_dummyserver/test_connectionpool.py b/test/with_dummyserver/test_connectionpool.py
|
||
|
|
index c0f4d9d5..5e58c971 100644
|
||
|
|
--- a/test/with_dummyserver/test_connectionpool.py
|
||
|
|
+++ b/test/with_dummyserver/test_connectionpool.py
|
||
|
|
@@ -480,6 +480,25 @@ class TestConnectionPool(HTTPDummyServerTestCase):
|
||
|
|
assert r.status == 200
|
||
|
|
assert r.data == b"Dummy server!"
|
||
|
|
|
||
|
|
+ @mock.patch("urllib3.response.GzipDecoder.decompress")
|
||
|
|
+ def test_no_decoding_with_redirect_when_preload_disabled(
|
||
|
|
+ self, gzip_decompress: mock.MagicMock
|
||
|
|
+ ) -> None:
|
||
|
|
+ """
|
||
|
|
+ Test that urllib3 does not attempt to decode a gzipped redirect
|
||
|
|
+ response when `preload_content` is set to `False`.
|
||
|
|
+ """
|
||
|
|
+ with HTTPConnectionPool(self.host, self.port) as pool:
|
||
|
|
+ # Three requests are expected: two redirects and one final / 200 OK.
|
||
|
|
+ response = pool.request(
|
||
|
|
+ "GET",
|
||
|
|
+ "/redirect",
|
||
|
|
+ fields={"target": "/redirect?compressed=true", "compressed": "true"},
|
||
|
|
+ preload_content=False,
|
||
|
|
+ )
|
||
|
|
+ assert response.status == 200
|
||
|
|
+ gzip_decompress.assert_not_called()
|
||
|
|
+
|
||
|
|
def test_303_redirect_makes_request_lose_body(self) -> None:
|
||
|
|
with HTTPConnectionPool(self.host, self.port) as pool:
|
||
|
|
response = pool.request(
|
||
|
|
--
|
||
|
|
2.52.0
|
||
|
|
|