@@ -54,12 +54,12 @@ def callback(url, **kwargs):
5454 return callback , mock
5555
5656
57- class TestAsyncify (unittest .TestCase ):
57+ class TestAsyncify (unittest .IsolatedAsyncioTestCase ):
5858 @pytest .mark .asyncio
5959 @aioresponses ()
6060 async def test_get (self , mocked ):
6161 callback , mock = get_callback ()
62- await mocked .get (clients , callback = callback )
62+ mocked .get (clients , callback = callback )
6363 c = asyncify (Clients )(domain = "example.com" , token = "jwt" )
6464 self .assertEqual (await c .all_async (), payload )
6565 mock .assert_called_with (
@@ -74,7 +74,7 @@ async def test_get(self, mocked):
7474 @aioresponses ()
7575 async def test_post (self , mocked ):
7676 callback , mock = get_callback ()
77- await mocked .post (clients , callback = callback )
77+ mocked .post (clients , callback = callback )
7878 c = asyncify (Clients )(domain = "example.com" , token = "jwt" )
7979 data = {"client" : 1 }
8080 self .assertEqual (await c .create_async (data ), payload )
@@ -90,7 +90,7 @@ async def test_post(self, mocked):
9090 @aioresponses ()
9191 async def test_post_auth (self , mocked ):
9292 callback , mock = get_callback ()
93- await mocked .post (token , callback = callback )
93+ mocked .post (token , callback = callback )
9494 c = asyncify (GetToken )("example.com" , "cid" , client_secret = "clsec" )
9595 self .assertEqual (
9696 await c .login_async (username = "usrnm" , password = "pswd" ), payload
@@ -116,7 +116,7 @@ async def test_post_auth(self, mocked):
116116 @aioresponses ()
117117 async def test_user_info (self , mocked ):
118118 callback , mock = get_callback ()
119- await mocked .get (user_info , callback = callback )
119+ mocked .get (user_info , callback = callback )
120120 c = asyncify (Users )(domain = "example.com" )
121121 self .assertEqual (
122122 await c .userinfo_async (access_token = "access-token-example" ), payload
@@ -133,7 +133,7 @@ async def test_user_info(self, mocked):
133133 @aioresponses ()
134134 async def test_file_post (self , mocked ):
135135 callback , mock = get_callback ()
136- await mocked .post (users_imports , callback = callback )
136+ mocked .post (users_imports , callback = callback )
137137 j = asyncify (Jobs )(domain = "example.com" , token = "jwt" )
138138 users = TemporaryFile ()
139139 self .assertEqual (await j .import_users_async ("connection-1" , users ), payload )
@@ -158,7 +158,7 @@ async def test_file_post(self, mocked):
158158 @aioresponses ()
159159 async def test_patch (self , mocked ):
160160 callback , mock = get_callback ()
161- await mocked .patch (clients , callback = callback )
161+ mocked .patch (clients , callback = callback )
162162 c = asyncify (Clients )(domain = "example.com" , token = "jwt" )
163163 data = {"client" : 1 }
164164 self .assertEqual (await c .update_async ("client-1" , data ), payload )
@@ -174,7 +174,7 @@ async def test_patch(self, mocked):
174174 @aioresponses ()
175175 async def test_put (self , mocked ):
176176 callback , mock = get_callback ()
177- await mocked .put (factors , callback = callback )
177+ mocked .put (factors , callback = callback )
178178 g = asyncify (Guardian )(domain = "example.com" , token = "jwt" )
179179 data = {"factor" : 1 }
180180 self .assertEqual (await g .update_factor_async ("factor-1" , data ), payload )
@@ -190,7 +190,7 @@ async def test_put(self, mocked):
190190 @aioresponses ()
191191 async def test_delete (self , mocked ):
192192 callback , mock = get_callback ()
193- await mocked .delete (clients , callback = callback )
193+ mocked .delete (clients , callback = callback )
194194 c = asyncify (Clients )(domain = "example.com" , token = "jwt" )
195195 self .assertEqual (await c .delete_async ("client-1" ), payload )
196196 mock .assert_called_with (
@@ -206,7 +206,7 @@ async def test_delete(self, mocked):
206206 @aioresponses ()
207207 async def test_shared_session (self , mocked ):
208208 callback , mock = get_callback ()
209- await mocked .get (clients , callback = callback )
209+ mocked .get (clients , callback = callback )
210210 async with asyncify (Clients )(domain = "example.com" , token = "jwt" ) as c :
211211 self .assertEqual (await c .all_async (), payload )
212212 mock .assert_called_with (
@@ -221,10 +221,10 @@ async def test_shared_session(self, mocked):
221221 @aioresponses ()
222222 async def test_rate_limit (self , mocked ):
223223 callback , mock = get_callback (status = 429 )
224- await mocked .get (clients , callback = callback )
225- await mocked .get (clients , callback = callback )
226- await mocked .get (clients , callback = callback )
227- await mocked .get (clients , payload = payload )
224+ mocked .get (clients , callback = callback )
225+ mocked .get (clients , callback = callback )
226+ mocked .get (clients , callback = callback )
227+ mocked .get (clients , payload = payload )
228228 c = asyncify (Clients )(domain = "example.com" , token = "jwt" )
229229 rest_client = c ._async_client .client
230230 rest_client ._skip_sleep = True
@@ -237,21 +237,21 @@ async def test_rate_limit(self, mocked):
237237 @aioresponses ()
238238 async def test_rate_limit_post (self , mocked ):
239239 callback , mock = get_callback (status = 429 )
240- await mocked .post (clients , callback = callback )
241- await mocked .post (clients , callback = callback )
242- await mocked .post (clients , callback = callback )
243- await mocked .post (clients , payload = payload )
240+ mocked .post (clients , callback = callback )
241+ mocked .post (clients , callback = callback )
242+ mocked .post (clients , callback = callback )
243+ mocked .post (clients , payload = payload )
244244 c = asyncify (Clients )(domain = "example.com" , token = "jwt" )
245245 rest_client = c ._async_client .client
246246 rest_client ._skip_sleep = True
247- self .assertEqual (await c .all_async ( ), payload )
247+ self .assertEqual (await c .create_async ({} ), payload )
248248 self .assertEqual (3 , mock .call_count )
249249
250250 @pytest .mark .asyncio
251251 @aioresponses ()
252252 async def test_timeout (self , mocked ):
253253 callback , mock = get_callback ()
254- await mocked .get (clients , callback = callback )
254+ mocked .get (clients , callback = callback )
255255 c = asyncify (Clients )(domain = "example.com" , token = "jwt" , timeout = (8.8 , 9.9 ))
256256 self .assertEqual (await c .all_async (), payload )
257257 mock .assert_called_with (
0 commit comments