9
9
10
10
from finch import Finch , AsyncFinch
11
11
from tests .utils import assert_matches_type
12
- from finch ._client import Finch , AsyncFinch
13
12
from finch .pagination import SyncSinglePage , AsyncSinglePage
14
13
from finch .types .hris .benefits import (
15
14
IndividualBenefit ,
19
18
)
20
19
21
20
base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
22
- access_token = "My Access Token"
23
21
24
22
25
23
class TestIndividuals :
26
- strict_client = Finch (base_url = base_url , access_token = access_token , _strict_response_validation = True )
27
- loose_client = Finch (base_url = base_url , access_token = access_token , _strict_response_validation = False )
28
- parametrize = pytest .mark .parametrize ("client" , [strict_client , loose_client ], ids = ["strict" , "loose" ])
24
+ parametrize = pytest .mark .parametrize ("client" , [False , True ], indirect = True , ids = ["loose" , "strict" ])
29
25
30
26
@parametrize
31
27
def test_method_enroll_many (self , client : Finch ) -> None :
@@ -201,21 +197,19 @@ def test_path_params_unenroll_many(self, client: Finch) -> None:
201
197
202
198
203
199
class TestAsyncIndividuals :
204
- strict_client = AsyncFinch (base_url = base_url , access_token = access_token , _strict_response_validation = True )
205
- loose_client = AsyncFinch (base_url = base_url , access_token = access_token , _strict_response_validation = False )
206
- parametrize = pytest .mark .parametrize ("client" , [strict_client , loose_client ], ids = ["strict" , "loose" ])
200
+ parametrize = pytest .mark .parametrize ("async_client" , [False , True ], indirect = True , ids = ["loose" , "strict" ])
207
201
208
202
@parametrize
209
- async def test_method_enroll_many (self , client : AsyncFinch ) -> None :
210
- individual = await client .hris .benefits .individuals .enroll_many (
203
+ async def test_method_enroll_many (self , async_client : AsyncFinch ) -> None :
204
+ individual = await async_client .hris .benefits .individuals .enroll_many (
211
205
"string" ,
212
206
individuals = [{}],
213
207
)
214
208
assert_matches_type (AsyncSinglePage [EnrolledIndividual ], individual , path = ["response" ])
215
209
216
210
@parametrize
217
- async def test_raw_response_enroll_many (self , client : AsyncFinch ) -> None :
218
- response = await client .hris .benefits .individuals .with_raw_response .enroll_many (
211
+ async def test_raw_response_enroll_many (self , async_client : AsyncFinch ) -> None :
212
+ response = await async_client .hris .benefits .individuals .with_raw_response .enroll_many (
219
213
"string" ,
220
214
individuals = [{}],
221
215
)
@@ -226,8 +220,8 @@ async def test_raw_response_enroll_many(self, client: AsyncFinch) -> None:
226
220
assert_matches_type (AsyncSinglePage [EnrolledIndividual ], individual , path = ["response" ])
227
221
228
222
@parametrize
229
- async def test_streaming_response_enroll_many (self , client : AsyncFinch ) -> None :
230
- async with client .hris .benefits .individuals .with_streaming_response .enroll_many (
223
+ async def test_streaming_response_enroll_many (self , async_client : AsyncFinch ) -> None :
224
+ async with async_client .hris .benefits .individuals .with_streaming_response .enroll_many (
231
225
"string" ,
232
226
individuals = [{}],
233
227
) as response :
@@ -240,23 +234,23 @@ async def test_streaming_response_enroll_many(self, client: AsyncFinch) -> None:
240
234
assert cast (Any , response .is_closed ) is True
241
235
242
236
@parametrize
243
- async def test_path_params_enroll_many (self , client : AsyncFinch ) -> None :
237
+ async def test_path_params_enroll_many (self , async_client : AsyncFinch ) -> None :
244
238
with pytest .raises (ValueError , match = r"Expected a non-empty value for `benefit_id` but received ''" ):
245
- await client .hris .benefits .individuals .with_raw_response .enroll_many (
239
+ await async_client .hris .benefits .individuals .with_raw_response .enroll_many (
246
240
"" ,
247
241
individuals = [{}],
248
242
)
249
243
250
244
@parametrize
251
- async def test_method_enrolled_ids (self , client : AsyncFinch ) -> None :
252
- individual = await client .hris .benefits .individuals .enrolled_ids (
245
+ async def test_method_enrolled_ids (self , async_client : AsyncFinch ) -> None :
246
+ individual = await async_client .hris .benefits .individuals .enrolled_ids (
253
247
"string" ,
254
248
)
255
249
assert_matches_type (IndividualEnrolledIDsResponse , individual , path = ["response" ])
256
250
257
251
@parametrize
258
- async def test_raw_response_enrolled_ids (self , client : AsyncFinch ) -> None :
259
- response = await client .hris .benefits .individuals .with_raw_response .enrolled_ids (
252
+ async def test_raw_response_enrolled_ids (self , async_client : AsyncFinch ) -> None :
253
+ response = await async_client .hris .benefits .individuals .with_raw_response .enrolled_ids (
260
254
"string" ,
261
255
)
262
256
@@ -266,8 +260,8 @@ async def test_raw_response_enrolled_ids(self, client: AsyncFinch) -> None:
266
260
assert_matches_type (IndividualEnrolledIDsResponse , individual , path = ["response" ])
267
261
268
262
@parametrize
269
- async def test_streaming_response_enrolled_ids (self , client : AsyncFinch ) -> None :
270
- async with client .hris .benefits .individuals .with_streaming_response .enrolled_ids (
263
+ async def test_streaming_response_enrolled_ids (self , async_client : AsyncFinch ) -> None :
264
+ async with async_client .hris .benefits .individuals .with_streaming_response .enrolled_ids (
271
265
"string" ,
272
266
) as response :
273
267
assert not response .is_closed
@@ -279,30 +273,30 @@ async def test_streaming_response_enrolled_ids(self, client: AsyncFinch) -> None
279
273
assert cast (Any , response .is_closed ) is True
280
274
281
275
@parametrize
282
- async def test_path_params_enrolled_ids (self , client : AsyncFinch ) -> None :
276
+ async def test_path_params_enrolled_ids (self , async_client : AsyncFinch ) -> None :
283
277
with pytest .raises (ValueError , match = r"Expected a non-empty value for `benefit_id` but received ''" ):
284
- await client .hris .benefits .individuals .with_raw_response .enrolled_ids (
278
+ await async_client .hris .benefits .individuals .with_raw_response .enrolled_ids (
285
279
"" ,
286
280
)
287
281
288
282
@parametrize
289
- async def test_method_retrieve_many_benefits (self , client : AsyncFinch ) -> None :
290
- individual = await client .hris .benefits .individuals .retrieve_many_benefits (
283
+ async def test_method_retrieve_many_benefits (self , async_client : AsyncFinch ) -> None :
284
+ individual = await async_client .hris .benefits .individuals .retrieve_many_benefits (
291
285
"string" ,
292
286
)
293
287
assert_matches_type (AsyncSinglePage [IndividualBenefit ], individual , path = ["response" ])
294
288
295
289
@parametrize
296
- async def test_method_retrieve_many_benefits_with_all_params (self , client : AsyncFinch ) -> None :
297
- individual = await client .hris .benefits .individuals .retrieve_many_benefits (
290
+ async def test_method_retrieve_many_benefits_with_all_params (self , async_client : AsyncFinch ) -> None :
291
+ individual = await async_client .hris .benefits .individuals .retrieve_many_benefits (
298
292
"string" ,
299
293
individual_ids = "d675d2b7-6d7b-41a8-b2d3-001eb3fb88f6,d02a6346-1f08-4312-a064-49ff3cafaa7a" ,
300
294
)
301
295
assert_matches_type (AsyncSinglePage [IndividualBenefit ], individual , path = ["response" ])
302
296
303
297
@parametrize
304
- async def test_raw_response_retrieve_many_benefits (self , client : AsyncFinch ) -> None :
305
- response = await client .hris .benefits .individuals .with_raw_response .retrieve_many_benefits (
298
+ async def test_raw_response_retrieve_many_benefits (self , async_client : AsyncFinch ) -> None :
299
+ response = await async_client .hris .benefits .individuals .with_raw_response .retrieve_many_benefits (
306
300
"string" ,
307
301
)
308
302
@@ -312,8 +306,8 @@ async def test_raw_response_retrieve_many_benefits(self, client: AsyncFinch) ->
312
306
assert_matches_type (AsyncSinglePage [IndividualBenefit ], individual , path = ["response" ])
313
307
314
308
@parametrize
315
- async def test_streaming_response_retrieve_many_benefits (self , client : AsyncFinch ) -> None :
316
- async with client .hris .benefits .individuals .with_streaming_response .retrieve_many_benefits (
309
+ async def test_streaming_response_retrieve_many_benefits (self , async_client : AsyncFinch ) -> None :
310
+ async with async_client .hris .benefits .individuals .with_streaming_response .retrieve_many_benefits (
317
311
"string" ,
318
312
) as response :
319
313
assert not response .is_closed
@@ -325,30 +319,30 @@ async def test_streaming_response_retrieve_many_benefits(self, client: AsyncFinc
325
319
assert cast (Any , response .is_closed ) is True
326
320
327
321
@parametrize
328
- async def test_path_params_retrieve_many_benefits (self , client : AsyncFinch ) -> None :
322
+ async def test_path_params_retrieve_many_benefits (self , async_client : AsyncFinch ) -> None :
329
323
with pytest .raises (ValueError , match = r"Expected a non-empty value for `benefit_id` but received ''" ):
330
- await client .hris .benefits .individuals .with_raw_response .retrieve_many_benefits (
324
+ await async_client .hris .benefits .individuals .with_raw_response .retrieve_many_benefits (
331
325
"" ,
332
326
)
333
327
334
328
@parametrize
335
- async def test_method_unenroll_many (self , client : AsyncFinch ) -> None :
336
- individual = await client .hris .benefits .individuals .unenroll_many (
329
+ async def test_method_unenroll_many (self , async_client : AsyncFinch ) -> None :
330
+ individual = await async_client .hris .benefits .individuals .unenroll_many (
337
331
"string" ,
338
332
)
339
333
assert_matches_type (AsyncSinglePage [UnenrolledIndividual ], individual , path = ["response" ])
340
334
341
335
@parametrize
342
- async def test_method_unenroll_many_with_all_params (self , client : AsyncFinch ) -> None :
343
- individual = await client .hris .benefits .individuals .unenroll_many (
336
+ async def test_method_unenroll_many_with_all_params (self , async_client : AsyncFinch ) -> None :
337
+ individual = await async_client .hris .benefits .individuals .unenroll_many (
344
338
"string" ,
345
339
individual_ids = ["string" , "string" , "string" ],
346
340
)
347
341
assert_matches_type (AsyncSinglePage [UnenrolledIndividual ], individual , path = ["response" ])
348
342
349
343
@parametrize
350
- async def test_raw_response_unenroll_many (self , client : AsyncFinch ) -> None :
351
- response = await client .hris .benefits .individuals .with_raw_response .unenroll_many (
344
+ async def test_raw_response_unenroll_many (self , async_client : AsyncFinch ) -> None :
345
+ response = await async_client .hris .benefits .individuals .with_raw_response .unenroll_many (
352
346
"string" ,
353
347
)
354
348
@@ -358,8 +352,8 @@ async def test_raw_response_unenroll_many(self, client: AsyncFinch) -> None:
358
352
assert_matches_type (AsyncSinglePage [UnenrolledIndividual ], individual , path = ["response" ])
359
353
360
354
@parametrize
361
- async def test_streaming_response_unenroll_many (self , client : AsyncFinch ) -> None :
362
- async with client .hris .benefits .individuals .with_streaming_response .unenroll_many (
355
+ async def test_streaming_response_unenroll_many (self , async_client : AsyncFinch ) -> None :
356
+ async with async_client .hris .benefits .individuals .with_streaming_response .unenroll_many (
363
357
"string" ,
364
358
) as response :
365
359
assert not response .is_closed
@@ -371,8 +365,8 @@ async def test_streaming_response_unenroll_many(self, client: AsyncFinch) -> Non
371
365
assert cast (Any , response .is_closed ) is True
372
366
373
367
@parametrize
374
- async def test_path_params_unenroll_many (self , client : AsyncFinch ) -> None :
368
+ async def test_path_params_unenroll_many (self , async_client : AsyncFinch ) -> None :
375
369
with pytest .raises (ValueError , match = r"Expected a non-empty value for `benefit_id` but received ''" ):
376
- await client .hris .benefits .individuals .with_raw_response .unenroll_many (
370
+ await async_client .hris .benefits .individuals .with_raw_response .unenroll_many (
377
371
"" ,
378
372
)
0 commit comments