Adopted unittest.mock.AsyncMock in async unit tests

Author: miguelgrinberg

tests/async/helpers.py

@@ -1,16 +1,4 @@
 import asyncio
-from unittest import mock
-
-
-def AsyncMock(*args, **kwargs):
- """Return a mock asynchronous function."""
- m = mock.MagicMock(*args, **kwargs)
-
- async def mock_coro(*args, **kwargs):
- return m(*args, **kwargs)
-
- mock_coro.mock = m
- return mock_coro
 
 
 def _run(coro):

tests/async/test_admin.py

@@ -10,7 +10,6 @@
 import socketio
 from socketio.exceptions import ConnectionError
 from tests.asyncio_web_server import SocketIOWebServer
-from .helpers import AsyncMock
 
 
 def with_instrumented_server(auth=False, **ikwargs):
@@ -211,7 +210,7 @@ def test_admin_connect_with_others(self):
  sid1 = client1.sid
 
  saved_check_for_upgrade = self.isvr._check_for_upgrade
- self.isvr._check_for_upgrade = AsyncMock()
+ self.isvr._check_for_upgrade = mock.AsyncMock()
  client2.connect('http://localhost:8900', namespace='/foo',
  transports=['polling'])
  sid2 = client2.sid