Skip to content

Commit f29a239

Browse files
committed
Fix mock signature to prevent resource warning for async method
1 parent e83b922 commit f29a239

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/test_asyncio/test_connection.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -254,9 +254,8 @@ async def do_close():
254254
async def do_read():
255255
return await conn.read_response()
256256

257-
reader = mock.AsyncMock()
258-
writer = mock.AsyncMock()
259-
writer.transport = mock.Mock()
257+
reader = mock.Mock(spec=asyncio.StreamReader)
258+
writer = mock.Mock(spec=asyncio.StreamWriter)
260259
writer.transport.get_extra_info.side_effect = None
261260

262261
# for HiredisParser

0 commit comments

Comments
 (0)