fix caplog tests

DriverX · Dec 12, 2023 · 29e16ac · 29e16ac
1 parent 2a8164c
commit 29e16ac
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tests/unit_tests/aioredis_cluster/test_pooler.py b/tests/unit_tests/aioredis_cluster/test_pooler.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from aioredis_cluster.abc import AbcPool
+from aioredis_cluster.abc import AbcConnection, AbcPool
 from aioredis_cluster.pooler import Pooler
 from aioredis_cluster.structs import Address
 
@@ -34,7 +34,11 @@ async def test_ensure_pool__identical_address(add_async_finalizer):
 
 
 async def test_ensure_pool__multiple(add_async_finalizer):
-    pools = [object(), object(), object()]
+    pools = [
+        mock.AsyncMock(AbcConnection),
+        mock.AsyncMock(AbcConnection),
+        mock.AsyncMock(AbcConnection),
+    ]
     mocked_create_pool = mock.AsyncMock(side_effect=pools)
 
     pooler = Pooler(mocked_create_pool)
@@ -92,7 +96,7 @@ async def create_pool_se(addr):
 
 
 async def test_ensure_pool__error(add_async_finalizer):
-    pools = [RuntimeError(), object()]
+    pools = [RuntimeError(), mock.AsyncMock(AbcConnection)]
     mocked_create_pool = mock.AsyncMock(side_effect=pools)
 
     pooler = Pooler(mocked_create_pool)