Handle malformed status codes in retry classification

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

hyperbrowser/client/polling.py

@@ -114,6 +114,8 @@ def _is_retryable_exception(exc: Exception) -> bool:
     if isinstance(exc, _NonRetryablePollingError):
         return False
     if isinstance(exc, HyperbrowserError) and exc.status_code is not None:
+        if isinstance(exc.status_code, bool) or not isinstance(exc.status_code, int):
+            return True
         if (
             _CLIENT_ERROR_STATUS_MIN <= exc.status_code < _CLIENT_ERROR_STATUS_MAX
             and exc.status_code not in _RETRYABLE_CLIENT_ERROR_STATUS_CODES

tests/test_polling.py

@@ -169,6 +169,34 @@ async def get_status() -> str:
     asyncio.run(run())
 
 
+def test_poll_until_terminal_status_async_handles_non_integer_status_codes_as_retryable():
+    async def run() -> None:
+        attempts = {"count": 0}
+
+        async def get_status() -> str:
+            attempts["count"] += 1
+            if attempts["count"] < 3:
+                raise HyperbrowserError(
+                    "malformed status code",
+                    status_code="400",  # type: ignore[arg-type]
+                )
+            return "completed"
+
+        status = await poll_until_terminal_status_async(
+            operation_name="async poll malformed status code retries",
+            get_status=get_status,
+            is_terminal_status=lambda value: value == "completed",
+            poll_interval_seconds=0.0001,
+            max_wait_seconds=1.0,
+            max_status_failures=5,
+        )
+
+        assert status == "completed"
+        assert attempts["count"] == 3
+
+    asyncio.run(run())
+
+
 def test_poll_until_terminal_status_retries_server_errors():
     attempts = {"count": 0}
 
@@ -337,6 +365,29 @@ def operation() -> str:
     assert attempts["count"] == 3
 
 
+def test_retry_operation_handles_non_integer_status_codes_as_retryable():
+    attempts = {"count": 0}
+
+    def operation() -> str:
+        attempts["count"] += 1
+        if attempts["count"] < 3:
+            raise HyperbrowserError(
+                "malformed status code",
+                status_code="400",  # type: ignore[arg-type]
+            )
+        return "ok"
+
+    result = retry_operation(
+        operation_name="sync retry malformed status code",
+        operation=operation,
+        max_attempts=5,
+        retry_delay_seconds=0.0001,
+    )
+
+    assert result == "ok"
+    assert attempts["count"] == 3
+
+
 def test_retry_operation_rejects_awaitable_operation_result():
     async def async_operation() -> str:
         return "ok"