@@ -722,20 +722,20 @@ def test_parse_retry_after_header(self, remaining_retries: int, retry_after: str
722722 @mock .patch ("browser_use_sdk._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
723723 @pytest .mark .respx (base_url = base_url )
724724 def test_retrying_timeout_errors_doesnt_leak (self , respx_mock : MockRouter , client : BrowserUse ) -> None :
725- respx_mock .get ("/users/me " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
725+ respx_mock .post ("/tasks " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
726726
727727 with pytest .raises (APITimeoutError ):
728- client .users . me . with_streaming_response .retrieve ( ).__enter__ ()
728+ client .tasks . with_streaming_response .create ( task = "x" ).__enter__ ()
729729
730730 assert _get_open_connections (self .client ) == 0
731731
732732 @mock .patch ("browser_use_sdk._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
733733 @pytest .mark .respx (base_url = base_url )
734734 def test_retrying_status_errors_doesnt_leak (self , respx_mock : MockRouter , client : BrowserUse ) -> None :
735- respx_mock .get ("/users/me " ).mock (return_value = httpx .Response (500 ))
735+ respx_mock .post ("/tasks " ).mock (return_value = httpx .Response (500 ))
736736
737737 with pytest .raises (APIStatusError ):
738- client .users . me . with_streaming_response .retrieve ( ).__enter__ ()
738+ client .tasks . with_streaming_response .create ( task = "x" ).__enter__ ()
739739 assert _get_open_connections (self .client ) == 0
740740
741741 @pytest .mark .parametrize ("failures_before_success" , [0 , 2 , 4 ])
@@ -762,9 +762,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
762762 return httpx .Response (500 )
763763 return httpx .Response (200 )
764764
765- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
765+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
766766
767- response = client .users . me . with_raw_response .retrieve ( )
767+ response = client .tasks . with_raw_response .create ( task = "x" )
768768
769769 assert response .retries_taken == failures_before_success
770770 assert int (response .http_request .headers .get ("x-stainless-retry-count" )) == failures_before_success
@@ -786,9 +786,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
786786 return httpx .Response (500 )
787787 return httpx .Response (200 )
788788
789- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
789+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
790790
791- response = client .users . me . with_raw_response .retrieve ( extra_headers = {"x-stainless-retry-count" : Omit ()})
791+ response = client .tasks . with_raw_response .create ( task = "x" , extra_headers = {"x-stainless-retry-count" : Omit ()})
792792
793793 assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
794794
@@ -809,9 +809,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
809809 return httpx .Response (500 )
810810 return httpx .Response (200 )
811811
812- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
812+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
813813
814- response = client .users . me . with_raw_response .retrieve ( extra_headers = {"x-stainless-retry-count" : "42" })
814+ response = client .tasks . with_raw_response .create ( task = "x" , extra_headers = {"x-stainless-retry-count" : "42" })
815815
816816 assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
817817
@@ -1539,10 +1539,10 @@ async def test_parse_retry_after_header(self, remaining_retries: int, retry_afte
15391539 async def test_retrying_timeout_errors_doesnt_leak (
15401540 self , respx_mock : MockRouter , async_client : AsyncBrowserUse
15411541 ) -> None :
1542- respx_mock .get ("/users/me " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
1542+ respx_mock .post ("/tasks " ).mock (side_effect = httpx .TimeoutException ("Test timeout error" ))
15431543
15441544 with pytest .raises (APITimeoutError ):
1545- await async_client .users . me . with_streaming_response .retrieve ( ).__aenter__ ()
1545+ await async_client .tasks . with_streaming_response .create ( task = "x" ).__aenter__ ()
15461546
15471547 assert _get_open_connections (self .client ) == 0
15481548
@@ -1551,10 +1551,10 @@ async def test_retrying_timeout_errors_doesnt_leak(
15511551 async def test_retrying_status_errors_doesnt_leak (
15521552 self , respx_mock : MockRouter , async_client : AsyncBrowserUse
15531553 ) -> None :
1554- respx_mock .get ("/users/me " ).mock (return_value = httpx .Response (500 ))
1554+ respx_mock .post ("/tasks " ).mock (return_value = httpx .Response (500 ))
15551555
15561556 with pytest .raises (APIStatusError ):
1557- await async_client .users . me . with_streaming_response .retrieve ( ).__aenter__ ()
1557+ await async_client .tasks . with_streaming_response .create ( task = "x" ).__aenter__ ()
15581558 assert _get_open_connections (self .client ) == 0
15591559
15601560 @pytest .mark .parametrize ("failures_before_success" , [0 , 2 , 4 ])
@@ -1582,9 +1582,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
15821582 return httpx .Response (500 )
15831583 return httpx .Response (200 )
15841584
1585- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
1585+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
15861586
1587- response = await client .users . me . with_raw_response .retrieve ( )
1587+ response = await client .tasks . with_raw_response .create ( task = "x" )
15881588
15891589 assert response .retries_taken == failures_before_success
15901590 assert int (response .http_request .headers .get ("x-stainless-retry-count" )) == failures_before_success
@@ -1607,9 +1607,11 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
16071607 return httpx .Response (500 )
16081608 return httpx .Response (200 )
16091609
1610- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
1610+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
16111611
1612- response = await client .users .me .with_raw_response .retrieve (extra_headers = {"x-stainless-retry-count" : Omit ()})
1612+ response = await client .tasks .with_raw_response .create (
1613+ task = "x" , extra_headers = {"x-stainless-retry-count" : Omit ()}
1614+ )
16131615
16141616 assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
16151617
@@ -1631,9 +1633,11 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
16311633 return httpx .Response (500 )
16321634 return httpx .Response (200 )
16331635
1634- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
1636+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
16351637
1636- response = await client .users .me .with_raw_response .retrieve (extra_headers = {"x-stainless-retry-count" : "42" })
1638+ response = await client .tasks .with_raw_response .create (
1639+ task = "x" , extra_headers = {"x-stainless-retry-count" : "42" }
1640+ )
16371641
16381642 assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
16391643
0 commit comments