1 files changed, 70 insertions, 0 deletions
diff --git a/tests/test_asyncio/test_retry.py b/tests/test_asyncio/test_retry.py
new file mode 100644
index 0000000..e83e001
--- /dev/null
+++ b/tests/test_asyncio/test_retry.py
@@ -0,0 +1,70 @@
+import pytest
+
+from redis.asyncio.connection import Connection, UnixDomainSocketConnection
+from redis.asyncio.retry import Retry
+from redis.backoff import AbstractBackoff, NoBackoff
+from redis.exceptions import ConnectionError
+
+
+class BackoffMock(AbstractBackoff):
+    def __init__(self):
+        self.reset_calls = 0
+        self.calls = 0
+
+    def reset(self):
+        self.reset_calls += 1
+
+    def compute(self, failures):
+        self.calls += 1
+        return 0
+
+
+@pytest.mark.onlynoncluster
+class TestConnectionConstructorWithRetry:
+    "Test that the Connection constructors properly handles Retry objects"
+
+    @pytest.mark.parametrize("retry_on_timeout", [False, True])
+    @pytest.mark.parametrize("Class", [Connection, UnixDomainSocketConnection])
+    def test_retry_on_timeout_boolean(self, Class, retry_on_timeout):
+        c = Class(retry_on_timeout=retry_on_timeout)
+        assert c.retry_on_timeout == retry_on_timeout
+        assert isinstance(c.retry, Retry)
+        assert c.retry._retries == (1 if retry_on_timeout else 0)
+
+    @pytest.mark.parametrize("retries", range(10))
+    @pytest.mark.parametrize("Class", [Connection, UnixDomainSocketConnection])
+    def test_retry_on_timeout_retry(self, Class, retries: int):
+        retry_on_timeout = retries > 0
+        c = Class(retry_on_timeout=retry_on_timeout, retry=Retry(NoBackoff(), retries))
+        assert c.retry_on_timeout == retry_on_timeout
+        assert isinstance(c.retry, Retry)
+        assert c.retry._retries == retries
+
+
+@pytest.mark.onlynoncluster
+class TestRetry:
+    "Test that Retry calls backoff and retries the expected number of times"
+
+    def setup_method(self, test_method):
+        self.actual_attempts = 0
+        self.actual_failures = 0
+
+    async def _do(self):
+        self.actual_attempts += 1
+        raise ConnectionError()
+
+    async def _fail(self, error):
+        self.actual_failures += 1
+
+    @pytest.mark.parametrize("retries", range(10))
+    @pytest.mark.asyncio
+    async def test_retry(self, retries: int):
+        backoff = BackoffMock()
+        retry = Retry(backoff, retries)
+        with pytest.raises(ConnectionError):
+            await retry.call_with_retry(self._do, self._fail)
+
+        assert self.actual_attempts == 1 + retries
+        assert self.actual_failures == 1 + retries
+        assert backoff.reset_calls == 1
+        assert backoff.calls == retries