fix(tests): fix: tests which call HTTP endpoints directly with the example parameters

Author: stainless-app[bot]

tests/test_client.py

@@ -24,7 +24,6 @@
 from ydc_search_api import YdcSearchAPI, AsyncYdcSearchAPI, APIResponseValidationError
 from ydc_search_api._types import Omit
 from ydc_search_api._models import BaseModel, FinalRequestOptions
-from ydc_search_api._constants import RAW_RESPONSE_HEADER
 from ydc_search_api._exceptions import APIStatusError, APITimeoutError, YdcSearchAPIError, APIResponseValidationError
 from ydc_search_api._base_client import (
     DEFAULT_TIMEOUT,
@@ -723,22 +722,21 @@ def test_parse_retry_after_header(self, remaining_retries: int, retry_after: str
 
     @mock.patch("ydc_search_api._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout)
     @pytest.mark.respx(base_url=base_url)
-    def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> None:
+    def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter, client: YdcSearchAPI) -> None:
         respx_mock.get("/search").mock(side_effect=httpx.TimeoutException("Test timeout error"))
 
         with pytest.raises(APITimeoutError):
-            self.client.get("/search", cast_to=httpx.Response, options={"headers": {RAW_RESPONSE_HEADER: "stream"}})
+            client.search.with_streaming_response.query(query="query").__enter__()
 
         assert _get_open_connections(self.client) == 0
 
     @mock.patch("ydc_search_api._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout)
     @pytest.mark.respx(base_url=base_url)
-    def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> None:
+    def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter, client: YdcSearchAPI) -> None:
         respx_mock.get("/search").mock(return_value=httpx.Response(500))
 
         with pytest.raises(APIStatusError):
-            self.client.get("/search", cast_to=httpx.Response, options={"headers": {RAW_RESPONSE_HEADER: "stream"}})
-
+            client.search.with_streaming_response.query(query="query").__enter__()
         assert _get_open_connections(self.client) == 0
 
     @pytest.mark.parametrize("failures_before_success", [0, 2, 4])
@@ -1540,26 +1538,25 @@ async def test_parse_retry_after_header(self, remaining_retries: int, retry_afte
 
     @mock.patch("ydc_search_api._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout)
     @pytest.mark.respx(base_url=base_url)
-    async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> None:
+    async def test_retrying_timeout_errors_doesnt_leak(
+        self, respx_mock: MockRouter, async_client: AsyncYdcSearchAPI
+    ) -> None:
         respx_mock.get("/search").mock(side_effect=httpx.TimeoutException("Test timeout error"))
 
         with pytest.raises(APITimeoutError):
-            await self.client.get(
-                "/search", cast_to=httpx.Response, options={"headers": {RAW_RESPONSE_HEADER: "stream"}}
-            )
+            await async_client.search.with_streaming_response.query(query="query").__aenter__()
 
         assert _get_open_connections(self.client) == 0
 
     @mock.patch("ydc_search_api._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout)
     @pytest.mark.respx(base_url=base_url)
-    async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> None:
+    async def test_retrying_status_errors_doesnt_leak(
+        self, respx_mock: MockRouter, async_client: AsyncYdcSearchAPI
+    ) -> None:
         respx_mock.get("/search").mock(return_value=httpx.Response(500))
 
         with pytest.raises(APIStatusError):
-            await self.client.get(
-                "/search", cast_to=httpx.Response, options={"headers": {RAW_RESPONSE_HEADER: "stream"}}
-            )
-
+            await async_client.search.with_streaming_response.query(query="query").__aenter__()
         assert _get_open_connections(self.client) == 0
 
     @pytest.mark.parametrize("failures_before_success", [0, 2, 4])