Skip to content

Commit 23c95da

Browse files
chore(internal): share client instances between all tests (#271)
1 parent f1d513b commit 23c95da

25 files changed

+352
-465
lines changed

tests/api_resources/hris/benefits/test_individuals.py

+38-44
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99

1010
from finch import Finch, AsyncFinch
1111
from tests.utils import assert_matches_type
12-
from finch._client import Finch, AsyncFinch
1312
from finch.pagination import SyncSinglePage, AsyncSinglePage
1413
from finch.types.hris.benefits import (
1514
IndividualBenefit,
@@ -19,13 +18,10 @@
1918
)
2019

2120
base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
22-
access_token = "My Access Token"
2321

2422

2523
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"])
2925

3026
@parametrize
3127
def test_method_enroll_many(self, client: Finch) -> None:
@@ -201,21 +197,19 @@ def test_path_params_unenroll_many(self, client: Finch) -> None:
201197

202198

203199
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"])
207201

208202
@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(
211205
"string",
212206
individuals=[{}],
213207
)
214208
assert_matches_type(AsyncSinglePage[EnrolledIndividual], individual, path=["response"])
215209

216210
@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(
219213
"string",
220214
individuals=[{}],
221215
)
@@ -226,8 +220,8 @@ async def test_raw_response_enroll_many(self, client: AsyncFinch) -> None:
226220
assert_matches_type(AsyncSinglePage[EnrolledIndividual], individual, path=["response"])
227221

228222
@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(
231225
"string",
232226
individuals=[{}],
233227
) as response:
@@ -240,23 +234,23 @@ async def test_streaming_response_enroll_many(self, client: AsyncFinch) -> None:
240234
assert cast(Any, response.is_closed) is True
241235

242236
@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:
244238
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(
246240
"",
247241
individuals=[{}],
248242
)
249243

250244
@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(
253247
"string",
254248
)
255249
assert_matches_type(IndividualEnrolledIDsResponse, individual, path=["response"])
256250

257251
@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(
260254
"string",
261255
)
262256

@@ -266,8 +260,8 @@ async def test_raw_response_enrolled_ids(self, client: AsyncFinch) -> None:
266260
assert_matches_type(IndividualEnrolledIDsResponse, individual, path=["response"])
267261

268262
@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(
271265
"string",
272266
) as response:
273267
assert not response.is_closed
@@ -279,30 +273,30 @@ async def test_streaming_response_enrolled_ids(self, client: AsyncFinch) -> None
279273
assert cast(Any, response.is_closed) is True
280274

281275
@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:
283277
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(
285279
"",
286280
)
287281

288282
@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(
291285
"string",
292286
)
293287
assert_matches_type(AsyncSinglePage[IndividualBenefit], individual, path=["response"])
294288

295289
@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(
298292
"string",
299293
individual_ids="d675d2b7-6d7b-41a8-b2d3-001eb3fb88f6,d02a6346-1f08-4312-a064-49ff3cafaa7a",
300294
)
301295
assert_matches_type(AsyncSinglePage[IndividualBenefit], individual, path=["response"])
302296

303297
@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(
306300
"string",
307301
)
308302

@@ -312,8 +306,8 @@ async def test_raw_response_retrieve_many_benefits(self, client: AsyncFinch) ->
312306
assert_matches_type(AsyncSinglePage[IndividualBenefit], individual, path=["response"])
313307

314308
@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(
317311
"string",
318312
) as response:
319313
assert not response.is_closed
@@ -325,30 +319,30 @@ async def test_streaming_response_retrieve_many_benefits(self, client: AsyncFinc
325319
assert cast(Any, response.is_closed) is True
326320

327321
@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:
329323
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(
331325
"",
332326
)
333327

334328
@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(
337331
"string",
338332
)
339333
assert_matches_type(AsyncSinglePage[UnenrolledIndividual], individual, path=["response"])
340334

341335
@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(
344338
"string",
345339
individual_ids=["string", "string", "string"],
346340
)
347341
assert_matches_type(AsyncSinglePage[UnenrolledIndividual], individual, path=["response"])
348342

349343
@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(
352346
"string",
353347
)
354348

@@ -358,8 +352,8 @@ async def test_raw_response_unenroll_many(self, client: AsyncFinch) -> None:
358352
assert_matches_type(AsyncSinglePage[UnenrolledIndividual], individual, path=["response"])
359353

360354
@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(
363357
"string",
364358
) as response:
365359
assert not response.is_closed
@@ -371,8 +365,8 @@ async def test_streaming_response_unenroll_many(self, client: AsyncFinch) -> Non
371365
assert cast(Any, response.is_closed) is True
372366

373367
@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:
375369
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(
377371
"",
378372
)

0 commit comments

Comments
 (0)