How to use

OAuth2AuthorizedClientArgumentResolver

in

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

@Override
public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
  if (this.clientRegistrationRepository != null && this.authorizedClientRepository != null) {
    OAuth2AuthorizedClientArgumentResolver authorizedClientArgumentResolver =
        new OAuth2AuthorizedClientArgumentResolver(
            this.clientRegistrationRepository, this.authorizedClientRepository);
    argumentResolvers.add(authorizedClientArgumentResolver);
  }
}

              @Nullable WebDataBinderFactory binderFactory) throws Exception {
String clientRegistrationId = this.resolveClientRegistrationId(parameter);
if (StringUtils.isEmpty(clientRegistrationId)) {
  throw new IllegalArgumentException("Unable to resolve the Client Registration Identifier. " +
  authorizedClient = this.authorizeClientCredentialsClient(clientRegistration, servletRequest, servletResponse);

@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);
}

@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 supportsParameterWhenParameterTypeOAuth2AuthorizedClientWithoutAnnotationThenFalse() {
  MethodParameter methodParameter = this.getMethodParameter("paramTypeAuthorizedClientWithoutAnnotation", OAuth2AuthorizedClient.class);
  assertThat(this.argumentResolver.supportsParameter(methodParameter)).isFalse();
}

@Test
public void setClientCredentialsTokenResponseClientWhenClientIsNullThenThrowIllegalArgumentException() {
  assertThatThrownBy(() -> this.argumentResolver.setClientCredentialsTokenResponseClient(null))
      .isInstanceOf(IllegalArgumentException.class);
}

@Test
public void supportsParameterWhenParameterTypeUnsupportedThenFalse() {
  MethodParameter methodParameter = this.getMethodParameter("paramTypeUnsupported", String.class);
  assertThat(this.argumentResolver.supportsParameter(methodParameter)).isFalse();
}

@Test
public void constructorWhenClientRegistrationRepositoryIsNullThenThrowIllegalArgumentException() {
  assertThatThrownBy(() -> new OAuth2AuthorizedClientArgumentResolver(null, this.authorizedClientRepository))
      .isInstanceOf(IllegalArgumentException.class);
}

              @Nullable WebDataBinderFactory binderFactory) throws Exception {
String clientRegistrationId = this.resolveClientRegistrationId(parameter);
if (StringUtils.isEmpty(clientRegistrationId)) {
  throw new IllegalArgumentException("Unable to resolve the Client Registration Identifier. " +
  authorizedClient = this.authorizeClientCredentialsClient(clientRegistration, servletRequest, servletResponse);

@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 supportsParameterWhenParameterTypeOAuth2AuthorizedClientThenTrue() {
  MethodParameter methodParameter = this.getMethodParameter("paramTypeAuthorizedClient", OAuth2AuthorizedClient.class);
  assertThat(this.argumentResolver.supportsParameter(methodParameter)).isTrue();
}

@Test
public void constructorWhenOAuth2AuthorizedClientRepositoryIsNullThenThrowIllegalArgumentException() {
  assertThatThrownBy(() -> new OAuth2AuthorizedClientArgumentResolver(this.clientRegistrationRepository, null))
      .isInstanceOf(IllegalArgumentException.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);
}

@Test
public void supportsParameterWhenParameterTypeUnsupportedWithoutAnnotationThenFalse() {
  MethodParameter methodParameter = this.getMethodParameter("paramTypeUnsupportedWithoutAnnotation", String.class);
  assertThat(this.argumentResolver.supportsParameter(methodParameter)).isFalse();
}

@Override
public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
  if (this.clientRegistrationRepository != null && this.authorizedClientRepository != null) {
    OAuth2AuthorizedClientArgumentResolver authorizedClientArgumentResolver =
        new OAuth2AuthorizedClientArgumentResolver(
            this.clientRegistrationRepository, this.authorizedClientRepository);
    argumentResolvers.add(authorizedClientArgumentResolver);
  }
}

@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);
}

this.clientRegistrationRepository = new InMemoryClientRegistrationRepository(this.registration1, this.registration2);
this.authorizedClientRepository = mock(OAuth2AuthorizedClientRepository.class);
this.argumentResolver = new OAuth2AuthorizedClientArgumentResolver(
    this.clientRegistrationRepository, this.authorizedClientRepository);
this.authorizedClient1 = new OAuth2AuthorizedClient(this.registration1, this.principalName, mock(OAuth2AccessToken.class));

@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();
}

@Override
public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
  if (this.clientRegistrationRepository != null && this.authorizedClientRepository != null) {
    OAuth2AuthorizedClientArgumentResolver authorizedClientArgumentResolver =
        new OAuth2AuthorizedClientArgumentResolver(
            this.clientRegistrationRepository, this.authorizedClientRepository);
    argumentResolvers.add(authorizedClientArgumentResolver);
  }
}