@@ -63,8 +63,12 @@ async def test_cursor_initialized(
6363 python_query_data : List [List [ColType ]],
6464) -> None :
6565 """Connection initialized its cursors properly."""
66- httpx_mock .add_callback (auth_callback , url = auth_url )
67- httpx_mock .add_callback (query_callback , url = query_url )
66+ httpx_mock .add_callback (auth_callback , url = auth_url , is_reusable = True )
67+ httpx_mock .add_callback (
68+ query_callback ,
69+ url = query_url ,
70+ is_reusable = True ,
71+ )
6872
6973 for url in (engine_url , f"https://{ engine_url } " ):
7074 async with (
@@ -112,7 +116,11 @@ async def test_connect_access_token(
112116 python_query_data : List [List [ColType ]],
113117 access_token : str ,
114118):
115- httpx_mock .add_callback (check_token_callback , url = query_url )
119+ httpx_mock .add_callback (
120+ check_token_callback ,
121+ url = query_url ,
122+ is_reusable = True ,
123+ )
116124 async with (
117125 await connect (
118126 engine_url = engine_url ,
@@ -162,10 +170,22 @@ async def test_connect_engine_name(
162170 ):
163171 pass
164172
165- httpx_mock .add_callback (auth_callback , url = auth_url )
166- httpx_mock .add_callback (query_callback , url = query_url )
167- httpx_mock .add_callback (account_id_callback , url = account_id_url )
168- httpx_mock .add_callback (get_engine_url_by_id_callback , url = get_engine_url_by_id_url )
173+ httpx_mock .add_callback (auth_callback , url = auth_url , is_reusable = True )
174+ httpx_mock .add_callback (
175+ query_callback ,
176+ url = query_url ,
177+ is_reusable = True ,
178+ )
179+ httpx_mock .add_callback (
180+ account_id_callback ,
181+ url = account_id_url ,
182+ is_reusable = True ,
183+ )
184+ httpx_mock .add_callback (
185+ get_engine_url_by_id_callback ,
186+ url = get_engine_url_by_id_url ,
187+ is_reusable = True ,
188+ )
169189
170190 # Mock engine id lookup error
171191 httpx_mock .add_response (
@@ -225,9 +245,17 @@ async def test_connect_default_engine(
225245 engine_by_db_url : str ,
226246 python_query_data : List [List [ColType ]],
227247):
228- httpx_mock .add_callback (auth_callback , url = auth_url )
229- httpx_mock .add_callback (query_callback , url = query_url )
230- httpx_mock .add_callback (account_id_callback , url = account_id_url )
248+ httpx_mock .add_callback (auth_callback , url = auth_url , is_reusable = True )
249+ httpx_mock .add_callback (
250+ query_callback ,
251+ url = query_url ,
252+ is_reusable = True ,
253+ )
254+ httpx_mock .add_callback (
255+ account_id_callback ,
256+ url = account_id_url ,
257+ is_reusable = True ,
258+ )
231259 engine_by_db_url = f"{ engine_by_db_url } ?database_name={ db_name } "
232260
233261 httpx_mock .add_response (
@@ -276,9 +304,17 @@ async def test_connection_token_caching(
276304 account_id_callback : Callable ,
277305 account_id_url : str ,
278306) -> None :
279- httpx_mock .add_callback (check_credentials_callback , url = auth_url )
280- httpx_mock .add_callback (query_callback , url = query_url )
281- httpx_mock .add_callback (account_id_callback , url = account_id_url )
307+ httpx_mock .add_callback (check_credentials_callback , url = auth_url , is_reusable = True )
308+ httpx_mock .add_callback (
309+ query_callback ,
310+ url = query_url ,
311+ is_reusable = True ,
312+ )
313+ httpx_mock .add_callback (
314+ account_id_callback ,
315+ url = account_id_url ,
316+ is_reusable = True ,
317+ )
282318
283319 with Patcher ():
284320 async with await connect (
@@ -336,9 +372,17 @@ async def test_connect_with_auth(
336372 account_id_callback : Callable ,
337373 account_id_url : str ,
338374) -> None :
339- httpx_mock .add_callback (check_credentials_callback , url = auth_url )
340- httpx_mock .add_callback (query_callback , url = query_url )
341- httpx_mock .add_callback (account_id_callback , url = account_id_url )
375+ httpx_mock .add_callback (check_credentials_callback , url = auth_url , is_reusable = True )
376+ httpx_mock .add_callback (
377+ query_callback ,
378+ url = query_url ,
379+ is_reusable = True ,
380+ )
381+ httpx_mock .add_callback (
382+ account_id_callback ,
383+ url = account_id_url ,
384+ is_reusable = True ,
385+ )
342386
343387 for auth in (
344388 UsernamePassword (
@@ -370,8 +414,12 @@ async def test_connect_account_name(
370414 account_id_url : Pattern ,
371415 account_id_callback : Callable ,
372416):
373- httpx_mock .add_callback (check_credentials_callback , url = auth_url )
374- httpx_mock .add_callback (account_id_callback , url = account_id_url )
417+ httpx_mock .add_callback (check_credentials_callback , url = auth_url , is_reusable = True )
418+ httpx_mock .add_callback (
419+ account_id_callback ,
420+ url = account_id_url ,
421+ is_reusable = True ,
422+ )
375423
376424 with raises (AccountNotFoundError ):
377425 async with await connect (
@@ -479,6 +527,10 @@ async def async_flow() -> None:
479527 await cursor .fetchmany (1 )
480528 await cursor .fetchall ()
481529
482- httpx_mock .add_callback (auth_callback , url = auth_url )
483- httpx_mock .add_callback (query_callback , url = query_url )
530+ httpx_mock .add_callback (auth_callback , url = auth_url , is_reusable = True )
531+ httpx_mock .add_callback (
532+ query_callback ,
533+ url = query_url ,
534+ is_reusable = True ,
535+ )
484536 run (async_flow ())
0 commit comments