lazy-wheel: allow redirects for HEAD request

Resolves: #9039 (cherry picked from commit 7bf0ee7)
python-poetry · Mar 2, 2024 · 33b7618 · 33b7618
1 parent 58995de
commit 33b7618
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/poetry/inspection/lazy_wheel.py b/src/poetry/inspection/lazy_wheel.py
@@ -410,7 +410,9 @@ def _content_length_from_head(self) -> int:
         :raises HTTPRangeRequestUnsupported: if the response fails to indicate support
                                              for "bytes" ranges."""
         self._request_count += 1
-        head = self._session.head(self._url, headers=self._uncached_headers())
+        head = self._session.head(
+            self._url, headers=self._uncached_headers(), allow_redirects=True
+        )
         head.raise_for_status()
         assert head.status_code == codes.ok
         accepted_range = head.headers.get("Accept-Ranges", None)

diff --git a/tests/inspection/test_lazy_wheel.py b/tests/inspection/test_lazy_wheel.py
@@ -317,6 +317,16 @@ def test_metadata_from_wheel_url_with_redirect(
     )
 
 
+def test_metadata_from_wheel_url_with_redirect_after_500(
+    assert_metadata_from_wheel_url: AssertMetadataFromWheelUrl,
+) -> None:
+    assert_metadata_from_wheel_url(
+        negative_offset_error=(codes.internal_server_error, b"Internal server error"),
+        expected_requests=10,
+        redirect=True,
+    )
+
+
 @pytest.mark.parametrize(
     ("negative_offset_failure", "expected_requests"),
     [