Polish gh-15237
This commit is contained in:
@@ -286,9 +286,13 @@ public class OAuth2ClientConfigurerTests {
|
||||
}
|
||||
|
||||
@Test
|
||||
public void configureWhenCustomAuthorizationRequestResolverBeanPresentThenAuthorizationRequestIncludesCustomParameters()
|
||||
public void configureWhenCustomAuthorizationRequestResolverBeanPresentThenAuthorizationRequestResolverUsed()
|
||||
throws Exception {
|
||||
this.spring.register(OAuth2ClientBeanConfig.class).autowire();
|
||||
OAuth2AuthorizationRequestResolver defaultAuthorizationRequestResolver = authorizationRequestResolver;
|
||||
authorizationRequestResolver = mock(OAuth2AuthorizationRequestResolver.class);
|
||||
given(authorizationRequestResolver.resolve(any()))
|
||||
.willAnswer((invocation) -> defaultAuthorizationRequestResolver.resolve(invocation.getArgument(0)));
|
||||
this.spring.register(OAuth2ClientInLambdaConfig.class, AuthorizationRequestResolverConfig.class).autowire();
|
||||
// @formatter:off
|
||||
this.mockMvc.perform(get("/oauth2/authorization/registration-1"))
|
||||
.andExpect(status().is3xxRedirection())
|
||||
@@ -374,59 +378,14 @@ public class OAuth2ClientConfigurerTests {
|
||||
|
||||
}
|
||||
|
||||
@EnableWebSecurity
|
||||
@Configuration
|
||||
@EnableWebMvc
|
||||
static class OAuth2ClientBeanConfig {
|
||||
|
||||
@Bean
|
||||
SecurityFilterChain filterChain(HttpSecurity http) throws Exception {
|
||||
// @formatter:off
|
||||
http
|
||||
.authorizeRequests()
|
||||
.anyRequest().authenticated()
|
||||
.and()
|
||||
.requestCache()
|
||||
.requestCache(requestCache)
|
||||
.and()
|
||||
.oauth2Client()
|
||||
.authorizationCodeGrant()
|
||||
.authorizationRedirectStrategy(authorizationRedirectStrategy)
|
||||
.accessTokenResponseClient(accessTokenResponseClient);
|
||||
return http.build();
|
||||
// @formatter:on
|
||||
}
|
||||
|
||||
@Bean
|
||||
ClientRegistrationRepository clientRegistrationRepository() {
|
||||
return clientRegistrationRepository;
|
||||
}
|
||||
|
||||
@Bean
|
||||
OAuth2AuthorizedClientRepository authorizedClientRepository() {
|
||||
return authorizedClientRepository;
|
||||
}
|
||||
static class AuthorizationRequestResolverConfig {
|
||||
|
||||
@Bean
|
||||
OAuth2AuthorizationRequestResolver authorizationRequestResolver() {
|
||||
OAuth2AuthorizationRequestResolver defaultAuthorizationRequestResolver = authorizationRequestResolver;
|
||||
authorizationRequestResolver = mock(OAuth2AuthorizationRequestResolver.class);
|
||||
given(authorizationRequestResolver.resolve(any()))
|
||||
.willAnswer((invocation) -> defaultAuthorizationRequestResolver.resolve(invocation.getArgument(0)));
|
||||
return authorizationRequestResolver;
|
||||
}
|
||||
|
||||
@RestController
|
||||
class ResourceController {
|
||||
|
||||
@GetMapping("/resource1")
|
||||
String resource1(
|
||||
@RegisteredOAuth2AuthorizedClient("registration-1") OAuth2AuthorizedClient authorizedClient) {
|
||||
return "resource1";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -458,7 +458,7 @@ public class OAuth2LoginTests {
|
||||
OidcUser user = TestOidcUsers.create();
|
||||
ReactiveOAuth2UserService<OidcUserRequest, OidcUser> userService = config.userService;
|
||||
given(userService.loadUser(any())).willReturn(Mono.just(user));
|
||||
ServerOAuth2AuthorizationRequestResolver resolver = config.resolver;
|
||||
ServerOAuth2AuthorizationRequestResolver authorizationRequestResolver = config.authorizationRequestResolver;
|
||||
// @formatter:off
|
||||
webTestClient.get()
|
||||
.uri("/login/oauth2/code/google")
|
||||
@@ -468,7 +468,7 @@ public class OAuth2LoginTests {
|
||||
verify(config.jwtDecoderFactory).createDecoder(any());
|
||||
verify(tokenResponseClient).getTokenResponse(any());
|
||||
verify(securityContextRepository).save(any(), any());
|
||||
verify(resolver).resolve(any());
|
||||
verify(authorizationRequestResolver).resolve(any());
|
||||
}
|
||||
|
||||
// gh-5562
|
||||
@@ -840,7 +840,7 @@ public class OAuth2LoginTests {
|
||||
|
||||
ServerSecurityContextRepository securityContextRepository = mock(ServerSecurityContextRepository.class);
|
||||
|
||||
ServerOAuth2AuthorizationRequestResolver resolver = spy(
|
||||
ServerOAuth2AuthorizationRequestResolver authorizationRequestResolver = spy(
|
||||
new DefaultServerOAuth2AuthorizationRequestResolver(new InMemoryReactiveClientRegistrationRepository(
|
||||
TestClientRegistrations.clientRegistration().build())));
|
||||
|
||||
@@ -872,8 +872,8 @@ public class OAuth2LoginTests {
|
||||
}
|
||||
|
||||
@Bean
|
||||
ServerOAuth2AuthorizationRequestResolver resolver() {
|
||||
return this.resolver;
|
||||
ServerOAuth2AuthorizationRequestResolver authorizationRequestResolver() {
|
||||
return this.authorizationRequestResolver;
|
||||
}
|
||||
|
||||
@Bean
|
||||
|
||||
Reference in New Issue
Block a user