@@ -722,20 +722,20 @@ def test_parse_retry_after_header(self, remaining_retries: int, retry_after: str
722
722
@mock .patch ("browser_use_sdk._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
723
723
@pytest .mark .respx (base_url = base_url )
724
724
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" ))
726
726
727
727
with pytest .raises (APITimeoutError ):
728
- client .users . me . with_streaming_response .retrieve ( ).__enter__ ()
728
+ client .tasks . with_streaming_response .create ( task = "x" ).__enter__ ()
729
729
730
730
assert _get_open_connections (self .client ) == 0
731
731
732
732
@mock .patch ("browser_use_sdk._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
733
733
@pytest .mark .respx (base_url = base_url )
734
734
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 ))
736
736
737
737
with pytest .raises (APIStatusError ):
738
- client .users . me . with_streaming_response .retrieve ( ).__enter__ ()
738
+ client .tasks . with_streaming_response .create ( task = "x" ).__enter__ ()
739
739
assert _get_open_connections (self .client ) == 0
740
740
741
741
@pytest .mark .parametrize ("failures_before_success" , [0 , 2 , 4 ])
@@ -762,9 +762,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
762
762
return httpx .Response (500 )
763
763
return httpx .Response (200 )
764
764
765
- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
765
+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
766
766
767
- response = client .users . me . with_raw_response .retrieve ( )
767
+ response = client .tasks . with_raw_response .create ( task = "x" )
768
768
769
769
assert response .retries_taken == failures_before_success
770
770
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:
786
786
return httpx .Response (500 )
787
787
return httpx .Response (200 )
788
788
789
- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
789
+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
790
790
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 ()})
792
792
793
793
assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
794
794
@@ -809,9 +809,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
809
809
return httpx .Response (500 )
810
810
return httpx .Response (200 )
811
811
812
- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
812
+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
813
813
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" })
815
815
816
816
assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
817
817
@@ -1539,10 +1539,10 @@ async def test_parse_retry_after_header(self, remaining_retries: int, retry_afte
1539
1539
async def test_retrying_timeout_errors_doesnt_leak (
1540
1540
self , respx_mock : MockRouter , async_client : AsyncBrowserUse
1541
1541
) -> 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" ))
1543
1543
1544
1544
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__ ()
1546
1546
1547
1547
assert _get_open_connections (self .client ) == 0
1548
1548
@@ -1551,10 +1551,10 @@ async def test_retrying_timeout_errors_doesnt_leak(
1551
1551
async def test_retrying_status_errors_doesnt_leak (
1552
1552
self , respx_mock : MockRouter , async_client : AsyncBrowserUse
1553
1553
) -> None :
1554
- respx_mock .get ("/users/me " ).mock (return_value = httpx .Response (500 ))
1554
+ respx_mock .post ("/tasks " ).mock (return_value = httpx .Response (500 ))
1555
1555
1556
1556
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__ ()
1558
1558
assert _get_open_connections (self .client ) == 0
1559
1559
1560
1560
@pytest .mark .parametrize ("failures_before_success" , [0 , 2 , 4 ])
@@ -1582,9 +1582,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
1582
1582
return httpx .Response (500 )
1583
1583
return httpx .Response (200 )
1584
1584
1585
- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
1585
+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
1586
1586
1587
- response = await client .users . me . with_raw_response .retrieve ( )
1587
+ response = await client .tasks . with_raw_response .create ( task = "x" )
1588
1588
1589
1589
assert response .retries_taken == failures_before_success
1590
1590
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:
1607
1607
return httpx .Response (500 )
1608
1608
return httpx .Response (200 )
1609
1609
1610
- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
1610
+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
1611
1611
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
+ )
1613
1615
1614
1616
assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
1615
1617
@@ -1631,9 +1633,11 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
1631
1633
return httpx .Response (500 )
1632
1634
return httpx .Response (200 )
1633
1635
1634
- respx_mock .get ("/users/me " ).mock (side_effect = retry_handler )
1636
+ respx_mock .post ("/tasks " ).mock (side_effect = retry_handler )
1635
1637
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
+ )
1637
1641
1638
1642
assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
1639
1643
0 commit comments