Update response with times (#574)

This commit is contained in:
Glenn Jocher
2025-10-21 13:33:17 +02:00
committed by GitHub
parent d373a223dd
commit 925f4852ee
2 changed files with 10 additions and 7 deletions
+8 -7
View File
@@ -125,21 +125,22 @@ class Action:
def _request(self, method: str, url: str, headers=None, expected_status=None, hard=False, **kwargs):
"""Unified request handler with error checking."""
response = getattr(requests, method)(url, headers=headers or self.headers, **kwargs)
r = getattr(requests, method)(url, headers=headers or self.headers, **kwargs)
expected = expected_status or self._default_status[method]
success = response.status_code in expected
success = r.status_code in expected
if self.verbose:
print(f"{'✓' if success else '✗'} {method.upper()} {url} → {response.status_code}")
elapsed = r.elapsed.total_seconds()
print(f"{'✓' if success else '✗'} {method.upper()} {url} → {r.status_code} ({elapsed:.1f}s)")
if not success:
try:
print(f" ❌ Error: {response.json().get('message', 'Unknown error')}")
print(f" ❌ Error: {r.json().get('message', 'Unknown error')}")
except Exception:
print(f" ❌ Error: {response.text[:200]}")
print(f" ❌ Error: {r.text[:200]}")
if not success and hard:
response.raise_for_status()
return response
r.raise_for_status()
return r
def get(self, url, **kwargs):
"""Performs GET request with error handling."""
+2
View File
@@ -15,6 +15,7 @@ def test_check_pypi_version():
with patch("requests.get") as mock_get:
mock_response = MagicMock()
mock_response.status_code = 200
mock_response.elapsed.total_seconds.return_value = 0.5
mock_response.json.return_value = {"info": {"version": "0.9.0"}}
mock_get.return_value = mock_response
@@ -43,6 +44,7 @@ def test_action_request_methods():
with patch("requests.get") as mock_get:
mock_response = MagicMock()
mock_response.status_code = 200
mock_response.elapsed.total_seconds.return_value = 0.3
mock_get.return_value = mock_response
action = Action(token="test-token")