How to use

resolveArgument

method

in

org.springframework.security.oauth2.client.web.method.annotation.OAuth2AuthorizedClientArgumentResolver

@Test
public void resolveArgumentWhenRegistrationIdEmptyAndNotOAuth2AuthenticationThenThrowIllegalArgumentException() {
  MethodParameter methodParameter = this.getMethodParameter("registrationIdEmpty", OAuth2AuthorizedClient.class);
  assertThatThrownBy(() -> this.argumentResolver.resolveArgument(methodParameter, null, null, null))
      .isInstanceOf(IllegalArgumentException.class)
      .hasMessage("Unable to resolve the Client Registration Identifier. It must be provided via @RegisteredOAuth2AuthorizedClient(\"client1\") or @RegisteredOAuth2AuthorizedClient(registrationId = \"client1\").");
}

@Test
public void resolveArgumentWhenAuthorizedClientNotFoundForAuthorizationCodeClientThenThrowClientAuthorizationRequiredException() {
  when(this.authorizedClientRepository.loadAuthorizedClient(anyString(), any(), any(HttpServletRequest.class)))
      .thenReturn(null);
  MethodParameter methodParameter = this.getMethodParameter("paramTypeAuthorizedClient", OAuth2AuthorizedClient.class);
  assertThatThrownBy(() -> this.argumentResolver.resolveArgument(methodParameter, null, new ServletWebRequest(this.request), null))
      .isInstanceOf(ClientAuthorizationRequiredException.class);
}

@Test
public void resolveArgumentWhenRegistrationIdEmptyAndOAuth2AuthenticationThenResolves() throws Exception {
  OAuth2AuthenticationToken authentication = mock(OAuth2AuthenticationToken.class);
  when(authentication.getAuthorizedClientRegistrationId()).thenReturn("client1");
  SecurityContext securityContext = SecurityContextHolder.createEmptyContext();
  securityContext.setAuthentication(authentication);
  SecurityContextHolder.setContext(securityContext);
  MethodParameter methodParameter = this.getMethodParameter("registrationIdEmpty", OAuth2AuthorizedClient.class);
  assertThat(this.argumentResolver.resolveArgument(
      methodParameter, null, new ServletWebRequest(this.request), null)).isSameAs(this.authorizedClient1);
}

@SuppressWarnings("unchecked")
@Test
public void resolveArgumentWhenAuthorizedClientNotFoundForClientCredentialsClientThenResolvesFromTokenResponseClient() throws Exception {
  OAuth2AccessTokenResponseClient<OAuth2ClientCredentialsGrantRequest> clientCredentialsTokenResponseClient =
      mock(OAuth2AccessTokenResponseClient.class);
  this.argumentResolver.setClientCredentialsTokenResponseClient(clientCredentialsTokenResponseClient);
  OAuth2AccessTokenResponse accessTokenResponse = OAuth2AccessTokenResponse
      .withToken("access-token-1234")
      .tokenType(OAuth2AccessToken.TokenType.BEARER)
      .expiresIn(3600)
      .build();
  when(clientCredentialsTokenResponseClient.getTokenResponse(any())).thenReturn(accessTokenResponse);
  when(this.authorizedClientRepository.loadAuthorizedClient(anyString(), any(), any(HttpServletRequest.class)))
      .thenReturn(null);
  MethodParameter methodParameter = this.getMethodParameter("clientCredentialsClient", OAuth2AuthorizedClient.class);
  OAuth2AuthorizedClient authorizedClient = (OAuth2AuthorizedClient) this.argumentResolver.resolveArgument(
      methodParameter, null, new ServletWebRequest(this.request), null);
  assertThat(authorizedClient).isNotNull();
  assertThat(authorizedClient.getClientRegistration()).isSameAs(this.registration2);
  assertThat(authorizedClient.getPrincipalName()).isEqualTo(this.principalName);
  assertThat(authorizedClient.getAccessToken()).isSameAs(accessTokenResponse.getAccessToken());
  verify(this.authorizedClientRepository).saveAuthorizedClient(
      eq(authorizedClient), eq(this.authentication), any(HttpServletRequest.class), eq(null));
}

@Test
public void resolveArgumentWhenAuthorizedClientFoundThenResolves() throws Exception {
  MethodParameter methodParameter = this.getMethodParameter("paramTypeAuthorizedClient", OAuth2AuthorizedClient.class);
  assertThat(this.argumentResolver.resolveArgument(
      methodParameter, null, new ServletWebRequest(this.request), null)).isSameAs(this.authorizedClient1);
}

@Test
public void resolveArgumentWhenRegistrationIdInvalidThenDoesNotResolve() throws Exception {
  MethodParameter methodParameter = this.getMethodParameter("registrationIdInvalid", OAuth2AuthorizedClient.class);
  assertThat(this.argumentResolver.resolveArgument(
      methodParameter, null, new ServletWebRequest(this.request), null)).isNull();
}