@@ -735,13 +735,13 @@ def test_parse_retry_after_header(self, remaining_retries: int, retry_after: str
735
735
@mock .patch ("gitpod._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
736
736
@pytest .mark .respx (base_url = base_url )
737
737
def test_retrying_timeout_errors_doesnt_leak (self , respx_mock : MockRouter ) -> None :
738
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (
738
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (
739
739
side_effect = httpx .TimeoutException ("Test timeout error" )
740
740
)
741
741
742
742
with pytest .raises (APITimeoutError ):
743
743
self .client .post (
744
- "/gitpod.v1.RunnerService/CreateRunner " ,
744
+ "/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ,
745
745
body = cast (object , dict ()),
746
746
cast_to = httpx .Response ,
747
747
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
@@ -752,11 +752,11 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
752
752
@mock .patch ("gitpod._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
753
753
@pytest .mark .respx (base_url = base_url )
754
754
def test_retrying_status_errors_doesnt_leak (self , respx_mock : MockRouter ) -> None :
755
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (return_value = httpx .Response (500 ))
755
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (return_value = httpx .Response (500 ))
756
756
757
757
with pytest .raises (APIStatusError ):
758
758
self .client .post (
759
- "/gitpod.v1.RunnerService/CreateRunner " ,
759
+ "/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ,
760
760
body = cast (object , dict ()),
761
761
cast_to = httpx .Response ,
762
762
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
@@ -788,9 +788,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
788
788
return httpx .Response (500 )
789
789
return httpx .Response (200 )
790
790
791
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (side_effect = retry_handler )
791
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (side_effect = retry_handler )
792
792
793
- response = client .runners .with_raw_response .create ()
793
+ response = client .identity .with_raw_response .get_authenticated_identity ()
794
794
795
795
assert response .retries_taken == failures_before_success
796
796
assert int (response .http_request .headers .get ("x-stainless-retry-count" )) == failures_before_success
@@ -812,9 +812,11 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
812
812
return httpx .Response (500 )
813
813
return httpx .Response (200 )
814
814
815
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (side_effect = retry_handler )
815
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (side_effect = retry_handler )
816
816
817
- response = client .runners .with_raw_response .create (extra_headers = {"x-stainless-retry-count" : Omit ()})
817
+ response = client .identity .with_raw_response .get_authenticated_identity (
818
+ extra_headers = {"x-stainless-retry-count" : Omit ()}
819
+ )
818
820
819
821
assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
820
822
@@ -835,9 +837,11 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
835
837
return httpx .Response (500 )
836
838
return httpx .Response (200 )
837
839
838
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (side_effect = retry_handler )
840
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (side_effect = retry_handler )
839
841
840
- response = client .runners .with_raw_response .create (extra_headers = {"x-stainless-retry-count" : "42" })
842
+ response = client .identity .with_raw_response .get_authenticated_identity (
843
+ extra_headers = {"x-stainless-retry-count" : "42" }
844
+ )
841
845
842
846
assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
843
847
@@ -1531,13 +1535,13 @@ async def test_parse_retry_after_header(self, remaining_retries: int, retry_afte
1531
1535
@mock .patch ("gitpod._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
1532
1536
@pytest .mark .respx (base_url = base_url )
1533
1537
async def test_retrying_timeout_errors_doesnt_leak (self , respx_mock : MockRouter ) -> None :
1534
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (
1538
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (
1535
1539
side_effect = httpx .TimeoutException ("Test timeout error" )
1536
1540
)
1537
1541
1538
1542
with pytest .raises (APITimeoutError ):
1539
1543
await self .client .post (
1540
- "/gitpod.v1.RunnerService/CreateRunner " ,
1544
+ "/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ,
1541
1545
body = cast (object , dict ()),
1542
1546
cast_to = httpx .Response ,
1543
1547
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
@@ -1548,11 +1552,11 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
1548
1552
@mock .patch ("gitpod._base_client.BaseClient._calculate_retry_timeout" , _low_retry_timeout )
1549
1553
@pytest .mark .respx (base_url = base_url )
1550
1554
async def test_retrying_status_errors_doesnt_leak (self , respx_mock : MockRouter ) -> None :
1551
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (return_value = httpx .Response (500 ))
1555
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (return_value = httpx .Response (500 ))
1552
1556
1553
1557
with pytest .raises (APIStatusError ):
1554
1558
await self .client .post (
1555
- "/gitpod.v1.RunnerService/CreateRunner " ,
1559
+ "/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ,
1556
1560
body = cast (object , dict ()),
1557
1561
cast_to = httpx .Response ,
1558
1562
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
@@ -1585,9 +1589,9 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
1585
1589
return httpx .Response (500 )
1586
1590
return httpx .Response (200 )
1587
1591
1588
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (side_effect = retry_handler )
1592
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (side_effect = retry_handler )
1589
1593
1590
- response = await client .runners .with_raw_response .create ()
1594
+ response = await client .identity .with_raw_response .get_authenticated_identity ()
1591
1595
1592
1596
assert response .retries_taken == failures_before_success
1593
1597
assert int (response .http_request .headers .get ("x-stainless-retry-count" )) == failures_before_success
@@ -1610,9 +1614,11 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
1610
1614
return httpx .Response (500 )
1611
1615
return httpx .Response (200 )
1612
1616
1613
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (side_effect = retry_handler )
1617
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (side_effect = retry_handler )
1614
1618
1615
- response = await client .runners .with_raw_response .create (extra_headers = {"x-stainless-retry-count" : Omit ()})
1619
+ response = await client .identity .with_raw_response .get_authenticated_identity (
1620
+ extra_headers = {"x-stainless-retry-count" : Omit ()}
1621
+ )
1616
1622
1617
1623
assert len (response .http_request .headers .get_list ("x-stainless-retry-count" )) == 0
1618
1624
@@ -1634,9 +1640,11 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
1634
1640
return httpx .Response (500 )
1635
1641
return httpx .Response (200 )
1636
1642
1637
- respx_mock .post ("/gitpod.v1.RunnerService/CreateRunner " ).mock (side_effect = retry_handler )
1643
+ respx_mock .post ("/gitpod.v1.IdentityService/GetAuthenticatedIdentity " ).mock (side_effect = retry_handler )
1638
1644
1639
- response = await client .runners .with_raw_response .create (extra_headers = {"x-stainless-retry-count" : "42" })
1645
+ response = await client .identity .with_raw_response .get_authenticated_identity (
1646
+ extra_headers = {"x-stainless-retry-count" : "42" }
1647
+ )
1640
1648
1641
1649
assert response .http_request .headers .get ("x-stainless-retry-count" ) == "42"
1642
1650
0 commit comments