port retry-after fix into corehttp

sdk/core/corehttp/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Release History
 
+## 1.0.0b5 (2024-02-29)
+
+### Other Changes
+
+- Accept float for `retry_after` header.
+
 ## 1.0.0b4 (2024-02-23)
 
 ### Other Changes

sdk/core/corehttp/corehttp/_version.py

@@ -9,4 +9,4 @@
 # regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "1.0.0b4"
+VERSION = "1.0.0b5"

sdk/core/corehttp/corehttp/runtime/policies/_utils.py

@@ -61,7 +61,7 @@ def parse_retry_after(retry_after: str) -> float:
     """
     delay: float  # Using the Mypy recommendation to use float for "int or float"
     try:
-        delay = int(retry_after)
+        delay = float(retry_after)
     except ValueError:
         # Not an integer? Try HTTP date
         retry_date = _parse_http_date(retry_after)

sdk/core/corehttp/tests/async_tests/test_retry_policy_async.py

@@ -57,7 +57,7 @@ def test_retry_types():
 
 @pytest.mark.parametrize(
     "retry_after_input,http_response",
-    product(["0", "800", "1000", "1200"], ASYNC_HTTP_RESPONSES),
+    product(["0", "800", "1000", "1200", "0.9"], ASYNC_HTTP_RESPONSES),
 )
 def test_retry_after(retry_after_input, http_response):
     retry_policy = AsyncRetryPolicy()

sdk/core/corehttp/tests/test_retry_policy.py

@@ -54,7 +54,9 @@ def test_retry_types():
     assert backoff_time == 4
 
 
-@pytest.mark.parametrize("retry_after_input,http_response", product(["0", "800", "1000", "1200"], HTTP_RESPONSES))
+@pytest.mark.parametrize(
+    "retry_after_input,http_response", product(["0", "800", "1000", "1200", "0.9"], HTTP_RESPONSES)
+)
 def test_retry_after(retry_after_input, http_response):
     retry_policy = RetryPolicy()
     request = HttpRequest("GET", "http://localhost")

sdk/core/corehttp/tests/test_utils.py

@@ -8,6 +8,7 @@
 import pytest
 from corehttp.utils import case_insensitive_dict
 from corehttp.utils._utils import get_running_async_lock
+from corehttp.runtime.policies._utils import parse_retry_after
 
 
 @pytest.fixture()
@@ -124,3 +125,14 @@ async def test_get_running_async_module_asyncio():
 def test_get_running_async_module_sync():
     with pytest.raises(RuntimeError):
         get_running_async_lock()
+
+
+def test_parse_retry_after():
+    ret = parse_retry_after("100")
+    assert ret == 100
+    ret = parse_retry_after("Fri, 1 Oct 2100 00:00:00 GMT")
+    assert ret > 0
+    ret = parse_retry_after("0")
+    assert ret == 0
+    ret = parse_retry_after("0.9")
+    assert ret == 0.9