@@ -76,7 +76,7 @@ class CustomTokenExchangeApiRequestHandlerTest {
7676 " openid profile email"
7777 )
7878
79- whenever(mockApi.customTokenExchange(any(), any(), isNull())).thenReturn(mockRequest)
79+ whenever(mockApi.customTokenExchange(any(), any(), isNull(), isNull() )).thenReturn(mockRequest)
8080 whenever(mockRequest.validateClaims()).thenReturn(mockRequest)
8181
8282 doAnswer {
@@ -90,6 +90,7 @@ class CustomTokenExchangeApiRequestHandlerTest {
9090 verify(mockApi).customTokenExchange(
9191 " urn:acme:legacy-token" ,
9292 " external-token-123" ,
93+ null ,
9394 null
9495 )
9596 verify(mockRequest).validateClaims()
@@ -115,7 +116,7 @@ class CustomTokenExchangeApiRequestHandlerTest {
115116 on { getDescription() } doReturn " Token validation failed"
116117 }
117118
118- whenever(mockApi.customTokenExchange(any(), any(), isNull())).thenReturn(mockRequest)
119+ whenever(mockApi.customTokenExchange(any(), any(), isNull(), isNull() )).thenReturn(mockRequest)
119120 whenever(mockRequest.validateClaims()).thenReturn(mockRequest)
120121
121122 doAnswer {
@@ -152,7 +153,7 @@ class CustomTokenExchangeApiRequestHandlerTest {
152153 " openid"
153154 )
154155
155- whenever(mockApi.customTokenExchange(any(), any(), isNull())).thenReturn(mockRequest)
156+ whenever(mockApi.customTokenExchange(any(), any(), isNull(), isNull() )).thenReturn(mockRequest)
156157 whenever(mockRequest.setAudience(any())).thenReturn(mockRequest)
157158 whenever(mockRequest.validateClaims()).thenReturn(mockRequest)
158159
@@ -191,7 +192,7 @@ class CustomTokenExchangeApiRequestHandlerTest {
191192 " openid profile email read:data"
192193 )
193194
194- whenever(mockApi.customTokenExchange(any(), any(), isNull())).thenReturn(mockRequest)
195+ whenever(mockApi.customTokenExchange(any(), any(), isNull(), isNull() )).thenReturn(mockRequest)
195196 whenever(mockRequest.setScope(any())).thenReturn(mockRequest)
196197 whenever(mockRequest.validateClaims()).thenReturn(mockRequest)
197198
@@ -231,7 +232,7 @@ class CustomTokenExchangeApiRequestHandlerTest {
231232 " openid profile email"
232233 )
233234
234- whenever(mockApi.customTokenExchange(any(), any(), isNull())).thenReturn(mockRequest)
235+ whenever(mockApi.customTokenExchange(any(), any(), isNull(), isNull() )).thenReturn(mockRequest)
235236 whenever(mockRequest.setAudience(any())).thenReturn(mockRequest)
236237 whenever(mockRequest.setScope(any())).thenReturn(mockRequest)
237238 whenever(mockRequest.validateClaims()).thenReturn(mockRequest)
@@ -273,7 +274,7 @@ class CustomTokenExchangeApiRequestHandlerTest {
273274 " openid"
274275 )
275276
276- whenever(mockApi.customTokenExchange(any(), any(), eq(" org_abc123" ))).thenReturn(mockRequest)
277+ whenever(mockApi.customTokenExchange(any(), any(), eq(" org_abc123" ), isNull() )).thenReturn(mockRequest)
277278 whenever(mockRequest.validateClaims()).thenReturn(mockRequest)
278279
279280 doAnswer {
@@ -287,7 +288,8 @@ class CustomTokenExchangeApiRequestHandlerTest {
287288 verify(mockApi).customTokenExchange(
288289 " urn:acme:legacy-token" ,
289290 " external-token-org" ,
290- " org_abc123"
291+ " org_abc123" ,
292+ null
291293 )
292294 verify(mockResult).success(any())
293295 }
0 commit comments