@@ -883,36 +883,20 @@ def test_parse_retry_after_header(
883883 @mock .patch ("context.dev._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
884884 @pytest .mark .respx (base_url = base_url )
885885 def test_retrying_timeout_errors_doesnt_leak (self , respx_mock : MockRouter , client : ContextDev ) -> None :
886- respx_mock .post ("/web/extract " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
886+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
887887
888888 with pytest .raises (APITimeoutError ):
889- client .web .with_streaming_response .extract (
890- schema = {
891- "type" : "bar" ,
892- "properties" : "bar" ,
893- "required" : "bar" ,
894- "additionalProperties" : "bar" ,
895- },
896- url = "https://example.com" ,
897- ).__enter__ ()
889+ client .brand .with_streaming_response .retrieve (domain = "domain" ).__enter__ ()
898890
899891 assert _get_open_connections (client ) == 0
900892
901893 @mock .patch ("context.dev._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
902894 @pytest .mark .respx (base_url = base_url )
903895 def test_retrying_status_errors_doesnt_leak (self , respx_mock : MockRouter , client : ContextDev ) -> None :
904- respx_mock .post ("/web/extract " ).mock (return_value = httpx .Response (500 ))
896+ respx_mock .post ("/brand/retrieve " ).mock (return_value = httpx .Response (500 ))
905897
906898 with pytest .raises (APIStatusError ):
907- client .web .with_streaming_response .extract (
908- schema = {
909- "type" : "bar" ,
910- "properties" : "bar" ,
911- "required" : "bar" ,
912- "additionalProperties" : "bar" ,
913- },
914- url = "https://example.com" ,
915- ).__enter__ ()
899+ client .brand .with_streaming_response .retrieve (domain = "domain" ).__enter__ ()
916900 assert _get_open_connections (client ) == 0
917901
918902 @pytest .mark .parametrize ("failures_before_success" , [0 , 2 , 4 ])
@@ -939,17 +923,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
939923 return httpx .Response (500 )
940924 return httpx .Response (200 )
941925
942- respx_mock .post ("/web/extract " ).mock (side_effect = retry_handler )
926+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = retry_handler )
943927
944- response = client .web .with_raw_response .extract (
945- schema = {
946- "type" : "bar" ,
947- "properties" : "bar" ,
948- "required" : "bar" ,
949- "additionalProperties" : "bar" ,
950- },
951- url = "https://example.com" ,
952- )
928+ response = client .brand .with_raw_response .retrieve (domain = "domain" )
953929
954930 assert response .retries_taken == failures_before_success
955931 assert int (response .http_request .headers .get ("x-stainless-retry-count" )) == failures_before_success
@@ -971,17 +947,10 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
971947 return httpx .Response (500 )
972948 return httpx .Response (200 )
973949
974- respx_mock .post ("/web/extract " ).mock (side_effect = retry_handler )
950+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = retry_handler )
975951
976- response = client .web .with_raw_response .extract (
977- schema = {
978- "type" : "bar" ,
979- "properties" : "bar" ,
980- "required" : "bar" ,
981- "additionalProperties" : "bar" ,
982- },
983- url = "https://example.com" ,
984- extra_headers = {"x-stainless-retry-count" : Omit ()},
952+ response = client .brand .with_raw_response .retrieve (
953+ domain = "domain" , extra_headers = {"x-stainless-retry-count" : Omit ()}
985954 )
986955
987956 assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
@@ -1003,17 +972,10 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
1003972 return httpx .Response (500 )
1004973 return httpx .Response (200 )
1005974
1006- respx_mock .post ("/web/extract " ).mock (side_effect = retry_handler )
975+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = retry_handler )
1007976
1008- response = client .web .with_raw_response .extract (
1009- schema = {
1010- "type" : "bar" ,
1011- "properties" : "bar" ,
1012- "required" : "bar" ,
1013- "additionalProperties" : "bar" ,
1014- },
1015- url = "https://example.com" ,
1016- extra_headers = {"x-stainless-retry-count" : "42" },
977+ response = client .brand .with_raw_response .retrieve (
978+ domain = "domain" , extra_headers = {"x-stainless-retry-count" : "42" }
1017979 )
1018980
1019981 assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
@@ -1859,18 +1821,10 @@ async def test_parse_retry_after_header(
18591821 async def test_retrying_timeout_errors_doesnt_leak (
18601822 self , respx_mock : MockRouter , async_client : AsyncContextDev
18611823 ) -> None :
1862- respx_mock .post ("/web/extract " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
1824+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
18631825
18641826 with pytest .raises (APITimeoutError ):
1865- await async_client .web .with_streaming_response .extract (
1866- schema = {
1867- "type" : "bar" ,
1868- "properties" : "bar" ,
1869- "required" : "bar" ,
1870- "additionalProperties" : "bar" ,
1871- },
1872- url = "https://example.com" ,
1873- ).__aenter__ ()
1827+ await async_client .brand .with_streaming_response .retrieve (domain = "domain" ).__aenter__ ()
18741828
18751829 assert _get_open_connections (async_client ) == 0
18761830
@@ -1879,18 +1833,10 @@ async def test_retrying_timeout_errors_doesnt_leak(
18791833 async def test_retrying_status_errors_doesnt_leak (
18801834 self , respx_mock : MockRouter , async_client : AsyncContextDev
18811835 ) -> None :
1882- respx_mock .post ("/web/extract " ).mock (return_value = httpx .Response (500 ))
1836+ respx_mock .post ("/brand/retrieve " ).mock (return_value = httpx .Response (500 ))
18831837
18841838 with pytest .raises (APIStatusError ):
1885- await async_client .web .with_streaming_response .extract (
1886- schema = {
1887- "type" : "bar" ,
1888- "properties" : "bar" ,
1889- "required" : "bar" ,
1890- "additionalProperties" : "bar" ,
1891- },
1892- url = "https://example.com" ,
1893- ).__aenter__ ()
1839+ await async_client .brand .with_streaming_response .retrieve (domain = "domain" ).__aenter__ ()
18941840 assert _get_open_connections (async_client ) == 0
18951841
18961842 @pytest .mark .parametrize ("failures_before_success" , [0 , 2 , 4 ])
@@ -1917,17 +1863,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
19171863 return httpx .Response (500 )
19181864 return httpx .Response (200 )
19191865
1920- respx_mock .post ("/web/extract " ).mock (side_effect = retry_handler )
1866+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = retry_handler )
19211867
1922- response = await client .web .with_raw_response .extract (
1923- schema = {
1924- "type" : "bar" ,
1925- "properties" : "bar" ,
1926- "required" : "bar" ,
1927- "additionalProperties" : "bar" ,
1928- },
1929- url = "https://example.com" ,
1930- )
1868+ response = await client .brand .with_raw_response .retrieve (domain = "domain" )
19311869
19321870 assert response .retries_taken == failures_before_success
19331871 assert int (response .http_request .headers .get ("x-stainless-retry-count" )) == failures_before_success
@@ -1949,17 +1887,10 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
19491887 return httpx .Response (500 )
19501888 return httpx .Response (200 )
19511889
1952- respx_mock .post ("/web/extract " ).mock (side_effect = retry_handler )
1890+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = retry_handler )
19531891
1954- response = await client .web .with_raw_response .extract (
1955- schema = {
1956- "type" : "bar" ,
1957- "properties" : "bar" ,
1958- "required" : "bar" ,
1959- "additionalProperties" : "bar" ,
1960- },
1961- url = "https://example.com" ,
1962- extra_headers = {"x-stainless-retry-count" : Omit ()},
1892+ response = await client .brand .with_raw_response .retrieve (
1893+ domain = "domain" , extra_headers = {"x-stainless-retry-count" : Omit ()}
19631894 )
19641895
19651896 assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
@@ -1981,17 +1912,10 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
19811912 return httpx .Response (500 )
19821913 return httpx .Response (200 )
19831914
1984- respx_mock .post ("/web/extract " ).mock (side_effect = retry_handler )
1915+ respx_mock .post ("/brand/retrieve " ).mock (side_effect = retry_handler )
19851916
1986- response = await client .web .with_raw_response .extract (
1987- schema = {
1988- "type" : "bar" ,
1989- "properties" : "bar" ,
1990- "required" : "bar" ,
1991- "additionalProperties" : "bar" ,
1992- },
1993- url = "https://example.com" ,
1994- extra_headers = {"x-stainless-retry-count" : "42" },
1917+ response = await client .brand .with_raw_response .retrieve (
1918+ domain = "domain" , extra_headers = {"x-stainless-retry-count" : "42" }
19951919 )
19961920
19971921 assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
0 commit comments