fix: add cloudflare timeout to retryable status codes (#1468)

Author: andreitava-uip

packages/uipath-platform/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-platform"
-version = "0.1.2"
+version = "0.1.3"
 description = "HTTP client library for programmatic access to UiPath Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

packages/uipath-platform/src/uipath/platform/common/retry.py

@@ -11,7 +11,7 @@
 
 from ..errors import EnrichedException
 
-RETRYABLE_STATUS_CODES: frozenset[int] = frozenset({408, 429, 502, 503, 504})
+RETRYABLE_STATUS_CODES: frozenset[int] = frozenset({408, 429, 502, 503, 504, 524})
 NON_RETRYABLE_STATUS_CODES: frozenset[int] = frozenset({400, 401, 403, 404, 413, 422})
 
 

packages/uipath-platform/tests/services/test_retry.py

@@ -4,7 +4,6 @@
 from uipath.platform.common.retry import (
     _MAX_BACKOFF_DELAY,
     _MAX_RETRY_AFTER_DELAY,
-    RETRYABLE_STATUS_CODES,
     exponential_backoff_with_jitter,
     extract_retry_after_from_chain,
     is_retryable_platform_exception,
@@ -15,26 +14,6 @@
 from uipath.platform.errors import EnrichedException
 
 
-class TestRetryableStatusCodes:
-    def test_contains_expected_codes(self):
-        assert 408 in RETRYABLE_STATUS_CODES
-        assert 429 in RETRYABLE_STATUS_CODES
-        assert 502 in RETRYABLE_STATUS_CODES
-        assert 503 in RETRYABLE_STATUS_CODES
-        assert 504 in RETRYABLE_STATUS_CODES
-
-    def test_does_not_contain_non_retryable(self):
-        assert 400 not in RETRYABLE_STATUS_CODES
-        assert 401 not in RETRYABLE_STATUS_CODES
-        assert 403 not in RETRYABLE_STATUS_CODES
-        assert 404 not in RETRYABLE_STATUS_CODES
-        assert 500 not in RETRYABLE_STATUS_CODES
-        assert 501 not in RETRYABLE_STATUS_CODES
-
-    def test_is_frozenset(self):
-        assert isinstance(RETRYABLE_STATUS_CODES, frozenset)
-
-
 class TestParseRetryAfter:
     def test_valid_integer(self):
         assert parse_retry_after("120") == 120.0
@@ -170,16 +149,36 @@ def test_timeout_exception(self):
         err = httpx.TimeoutException("timed out")
         assert is_retryable_platform_exception(err) is True
 
+    def test_enriched_408(self):
+        http_err = _make_http_status_error(408)
+        err = EnrichedException(http_err)
+        assert is_retryable_platform_exception(err) is True
+
     def test_enriched_429(self):
         http_err = _make_http_status_error(429)
         err = EnrichedException(http_err)
         assert is_retryable_platform_exception(err) is True
 
+    def test_enriched_502(self):
+        http_err = _make_http_status_error(502)
+        err = EnrichedException(http_err)
+        assert is_retryable_platform_exception(err) is True
+
     def test_enriched_503(self):
         http_err = _make_http_status_error(503)
         err = EnrichedException(http_err)
         assert is_retryable_platform_exception(err) is True
 
+    def test_enriched_504(self):
+        http_err = _make_http_status_error(504)
+        err = EnrichedException(http_err)
+        assert is_retryable_platform_exception(err) is True
+
+    def test_enriched_524_cloudflare_timeout(self):
+        http_err = _make_http_status_error(524)
+        err = EnrichedException(http_err)
+        assert is_retryable_platform_exception(err) is True
+
     def test_enriched_400_not_retryable(self):
         http_err = _make_http_status_error(400)
         err = EnrichedException(http_err)