fix tests

adamjmcgrath · adamjmcgrath · commit 1151a40833ec · 2022-04-04T11:45:48.000+01:00
diff --git a/auth0/v3/management/__init__.py b/auth0/v3/management/__init__.py
@@ -1,5 +1,4 @@
 from .actions import Actions
-from .asyncify import asyncify
 from .attack_protection import AttackProtection
 from .auth0 import Auth0
 from .blacklists import Blacklists
@@ -57,5 +56,4 @@
     "UserBlocks",
     "UsersByEmail",
     "Users",
-    "asyncify",
 )
diff --git a/auth0/v3/test_async/test_asyncify.py b/auth0/v3/test_async/test_asyncify.py
@@ -12,7 +12,8 @@
 from callee import Attrs
 from mock import ANY, MagicMock
 
-from auth0.v3.management import Clients, Guardian, Jobs, asyncify
+from auth0.v3.management import Clients, Guardian, Jobs
+from auth0.v3.management.asyncify import asyncify
 
 clients = re.compile(r"^https://example\.com/api/v2/clients.*")
 factors = re.compile(r"^https://example\.com/api/v2/guardian/factors.*")
@@ -60,6 +61,7 @@ async def test_get(self, mocked):
             allow_redirects=True,
             params={"include_fields": "true"},
             headers=headers,
+            timeout=ANY,
         )
 
     @aioresponses()
@@ -74,6 +76,7 @@ async def test_post(self, mocked):
             allow_redirects=True,
             json=data,
             headers=headers,
+            timeout=ANY,
         )
 
     @aioresponses()
@@ -96,6 +99,7 @@ async def test_file_post(self, mocked):
                 "users": users,
             },
             headers=file_port_headers,
+            timeout=ANY,
         )
         users.close()
 
@@ -111,6 +115,7 @@ async def test_patch(self, mocked):
             allow_redirects=True,
             json=data,
             headers=headers,
+            timeout=ANY,
         )
 
     @aioresponses()
@@ -125,6 +130,7 @@ async def test_put(self, mocked):
             allow_redirects=True,
             json=data,
             headers=headers,
+            timeout=ANY,
         )
 
     @aioresponses()
@@ -139,6 +145,7 @@ async def test_delete(self, mocked):
             params={},
             json=None,
             headers=headers,
+            timeout=ANY,
         )
 
     @aioresponses()
@@ -152,6 +159,7 @@ async def test_shared_session(self, mocked):
             allow_redirects=True,
             params={"include_fields": "true"},
             headers=headers,
+            timeout=ANY,
         )
 
     @aioresponses()
