13
13
from finch .pagination import SyncSinglePage , AsyncSinglePage
14
14
from finch .types .hris .benefits import (
15
15
IndividualBenefit ,
16
- EnrolledIndividualBenefit ,
17
16
IndividualEnrolledIDsResponse ,
18
17
IndividualUnenrollManyResponse ,
18
+ EnrolledIndividualBenefitResponse ,
19
19
)
20
20
21
21
base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
@@ -29,7 +29,7 @@ def test_method_enroll_many(self, client: Finch) -> None:
29
29
individual = client .hris .benefits .individuals .enroll_many (
30
30
benefit_id = "benefit_id" ,
31
31
)
32
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
32
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
33
33
34
34
@parametrize
35
35
def test_method_enroll_many_with_all_params (self , client : Finch ) -> None :
@@ -55,7 +55,7 @@ def test_method_enroll_many_with_all_params(self, client: Finch) -> None:
55
55
}
56
56
],
57
57
)
58
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
58
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
59
59
60
60
@parametrize
61
61
def test_raw_response_enroll_many (self , client : Finch ) -> None :
@@ -66,7 +66,7 @@ def test_raw_response_enroll_many(self, client: Finch) -> None:
66
66
assert response .is_closed is True
67
67
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
68
68
individual = response .parse ()
69
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
69
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
70
70
71
71
@parametrize
72
72
def test_streaming_response_enroll_many (self , client : Finch ) -> None :
@@ -77,7 +77,7 @@ def test_streaming_response_enroll_many(self, client: Finch) -> None:
77
77
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
78
78
79
79
individual = response .parse ()
80
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
80
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
81
81
82
82
assert cast (Any , response .is_closed ) is True
83
83
@@ -227,7 +227,7 @@ async def test_method_enroll_many(self, async_client: AsyncFinch) -> None:
227
227
individual = await async_client .hris .benefits .individuals .enroll_many (
228
228
benefit_id = "benefit_id" ,
229
229
)
230
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
230
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
231
231
232
232
@parametrize
233
233
async def test_method_enroll_many_with_all_params (self , async_client : AsyncFinch ) -> None :
@@ -253,7 +253,7 @@ async def test_method_enroll_many_with_all_params(self, async_client: AsyncFinch
253
253
}
254
254
],
255
255
)
256
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
256
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
257
257
258
258
@parametrize
259
259
async def test_raw_response_enroll_many (self , async_client : AsyncFinch ) -> None :
@@ -264,7 +264,7 @@ async def test_raw_response_enroll_many(self, async_client: AsyncFinch) -> None:
264
264
assert response .is_closed is True
265
265
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
266
266
individual = response .parse ()
267
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
267
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
268
268
269
269
@parametrize
270
270
async def test_streaming_response_enroll_many (self , async_client : AsyncFinch ) -> None :
@@ -275,7 +275,7 @@ async def test_streaming_response_enroll_many(self, async_client: AsyncFinch) ->
275
275
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
276
276
277
277
individual = await response .parse ()
278
- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
278
+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
279
279
280
280
assert cast (Any , response .is_closed ) is True
281
281
0 commit comments