1313from finch .pagination import SyncSinglePage , AsyncSinglePage
1414from finch .types .hris .benefits import (
1515 IndividualBenefit ,
16- IndividualEnrollManyResponse ,
16+ EnrolledIndividualBenifit ,
1717 IndividualEnrolledIDsResponse ,
18+ IndividualUnenrollManyResponse ,
1819)
1920
2021base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
@@ -28,7 +29,7 @@ def test_method_enroll_many(self, client: Finch) -> None:
2829 individual = client .hris .benefits .individuals .enroll_many (
2930 benefit_id = "benefit_id" ,
3031 )
31- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
32+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
3233
3334 @parametrize
3435 def test_method_enroll_many_with_all_params (self , client : Finch ) -> None :
@@ -54,7 +55,7 @@ def test_method_enroll_many_with_all_params(self, client: Finch) -> None:
5455 }
5556 ],
5657 )
57- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
58+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
5859
5960 @parametrize
6061 def test_raw_response_enroll_many (self , client : Finch ) -> None :
@@ -65,7 +66,7 @@ def test_raw_response_enroll_many(self, client: Finch) -> None:
6566 assert response .is_closed is True
6667 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
6768 individual = response .parse ()
68- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
69+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
6970
7071 @parametrize
7172 def test_streaming_response_enroll_many (self , client : Finch ) -> None :
@@ -76,7 +77,7 @@ def test_streaming_response_enroll_many(self, client: Finch) -> None:
7677 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
7778
7879 individual = response .parse ()
79- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
80+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
8081
8182 assert cast (Any , response .is_closed ) is True
8283
@@ -176,15 +177,15 @@ def test_method_unenroll_many(self, client: Finch) -> None:
176177 individual = client .hris .benefits .individuals .unenroll_many (
177178 benefit_id = "benefit_id" ,
178179 )
179- assert_matches_type (SyncSinglePage [ object ] , individual , path = ["response" ])
180+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
180181
181182 @parametrize
182183 def test_method_unenroll_many_with_all_params (self , client : Finch ) -> None :
183184 individual = client .hris .benefits .individuals .unenroll_many (
184185 benefit_id = "benefit_id" ,
185186 individual_ids = ["string" ],
186187 )
187- assert_matches_type (SyncSinglePage [ object ] , individual , path = ["response" ])
188+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
188189
189190 @parametrize
190191 def test_raw_response_unenroll_many (self , client : Finch ) -> None :
@@ -195,7 +196,7 @@ def test_raw_response_unenroll_many(self, client: Finch) -> None:
195196 assert response .is_closed is True
196197 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
197198 individual = response .parse ()
198- assert_matches_type (SyncSinglePage [ object ] , individual , path = ["response" ])
199+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
199200
200201 @parametrize
201202 def test_streaming_response_unenroll_many (self , client : Finch ) -> None :
@@ -206,7 +207,7 @@ def test_streaming_response_unenroll_many(self, client: Finch) -> None:
206207 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
207208
208209 individual = response .parse ()
209- assert_matches_type (SyncSinglePage [ object ] , individual , path = ["response" ])
210+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
210211
211212 assert cast (Any , response .is_closed ) is True
212213
@@ -226,7 +227,7 @@ async def test_method_enroll_many(self, async_client: AsyncFinch) -> None:
226227 individual = await async_client .hris .benefits .individuals .enroll_many (
227228 benefit_id = "benefit_id" ,
228229 )
229- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
230+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
230231
231232 @parametrize
232233 async def test_method_enroll_many_with_all_params (self , async_client : AsyncFinch ) -> None :
@@ -252,7 +253,7 @@ async def test_method_enroll_many_with_all_params(self, async_client: AsyncFinch
252253 }
253254 ],
254255 )
255- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
256+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
256257
257258 @parametrize
258259 async def test_raw_response_enroll_many (self , async_client : AsyncFinch ) -> None :
@@ -263,7 +264,7 @@ async def test_raw_response_enroll_many(self, async_client: AsyncFinch) -> None:
263264 assert response .is_closed is True
264265 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
265266 individual = response .parse ()
266- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
267+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
267268
268269 @parametrize
269270 async def test_streaming_response_enroll_many (self , async_client : AsyncFinch ) -> None :
@@ -274,7 +275,7 @@ async def test_streaming_response_enroll_many(self, async_client: AsyncFinch) ->
274275 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
275276
276277 individual = await response .parse ()
277- assert_matches_type (IndividualEnrollManyResponse , individual , path = ["response" ])
278+ assert_matches_type (EnrolledIndividualBenifit , individual , path = ["response" ])
278279
279280 assert cast (Any , response .is_closed ) is True
280281
@@ -374,15 +375,15 @@ async def test_method_unenroll_many(self, async_client: AsyncFinch) -> None:
374375 individual = await async_client .hris .benefits .individuals .unenroll_many (
375376 benefit_id = "benefit_id" ,
376377 )
377- assert_matches_type (AsyncSinglePage [ object ] , individual , path = ["response" ])
378+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
378379
379380 @parametrize
380381 async def test_method_unenroll_many_with_all_params (self , async_client : AsyncFinch ) -> None :
381382 individual = await async_client .hris .benefits .individuals .unenroll_many (
382383 benefit_id = "benefit_id" ,
383384 individual_ids = ["string" ],
384385 )
385- assert_matches_type (AsyncSinglePage [ object ] , individual , path = ["response" ])
386+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
386387
387388 @parametrize
388389 async def test_raw_response_unenroll_many (self , async_client : AsyncFinch ) -> None :
@@ -393,7 +394,7 @@ async def test_raw_response_unenroll_many(self, async_client: AsyncFinch) -> Non
393394 assert response .is_closed is True
394395 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
395396 individual = response .parse ()
396- assert_matches_type (AsyncSinglePage [ object ] , individual , path = ["response" ])
397+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
397398
398399 @parametrize
399400 async def test_streaming_response_unenroll_many (self , async_client : AsyncFinch ) -> None :
@@ -404,7 +405,7 @@ async def test_streaming_response_unenroll_many(self, async_client: AsyncFinch)
404405 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
405406
406407 individual = await response .parse ()
407- assert_matches_type (AsyncSinglePage [ object ] , individual , path = ["response" ])
408+ assert_matches_type (IndividualUnenrollManyResponse , individual , path = ["response" ])
408409
409410 assert cast (Any , response .is_closed ) is True
410411
0 commit comments