fixs connection mocks

redis · Jan 7, 2024 · 583ac94 · 583ac94
1 parent be39f47
commit 583ac94
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -356,6 +356,7 @@ def _gen_cluster_mock_resp(r, response):
     connection = Mock(spec=Connection)
     connection.retry = Retry(NoBackoff(), 0)
     connection.read_response.return_value = response
+    connection._get_from_local_cache.return_value = None
     with mock.patch.object(r, "connection", connection):
         yield r
 

diff --git a/tests/test_asyncio/conftest.py b/tests/test_asyncio/conftest.py
@@ -141,6 +141,7 @@ def _gen_cluster_mock_resp(r, response):
     connection = mock.AsyncMock(spec=Connection)
     connection.retry = Retry(NoBackoff(), 0)
     connection.read_response.return_value = response
+    connection.read_response.return_value = response
     with mock.patch.object(r, "connection", connection):
         yield r