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