Fix mock signature to prevent resource warning for async method

redis · Aug 25, 2023 · f29a239 · f29a239
1 parent e83b922
commit f29a239
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_asyncio/test_connection.py b/tests/test_asyncio/test_connection.py
@@ -254,9 +254,8 @@ async def do_close():
  async def do_read():
  return await conn.read_response()
 
- reader = mock.AsyncMock()
- writer = mock.AsyncMock()
- writer.transport = mock.Mock()
+ reader = mock.Mock(spec=asyncio.StreamReader)
+ writer = mock.Mock(spec=asyncio.StreamWriter)
  writer.transport.get_extra_info.side_effect = None
 
  # for HiredisParser