BearerTokenServerAuthenticationEntryPoint waits for subscriber

Fixes: gh-5742
This commit is contained in:
Rob Winch 2018-08-27 14:26:45 -05:00
parent 362c2ef1f2
commit 713e1e3356
2 changed files with 17 additions and 7 deletions

View File

@ -58,14 +58,16 @@ public final class BearerTokenServerAuthenticationEntryPoint implements
@Override
public Mono<Void> commence(ServerWebExchange exchange, AuthenticationException authException) {
HttpStatus status = getStatus(authException);
return Mono.defer(() -> {
HttpStatus status = getStatus(authException);
Map<String, String> parameters = createParameters(authException);
String wwwAuthenticate = computeWWWAuthenticateHeaderValue(parameters);
ServerHttpResponse response = exchange.getResponse();
response.getHeaders().set(HttpHeaders.WWW_AUTHENTICATE, wwwAuthenticate);
response.setStatusCode(status);
return response.setComplete();
Map<String, String> parameters = createParameters(authException);
String wwwAuthenticate = computeWWWAuthenticateHeaderValue(parameters);
ServerHttpResponse response = exchange.getResponse();
response.getHeaders().set(HttpHeaders.WWW_AUTHENTICATE, wwwAuthenticate);
response.setStatusCode(status);
return response.setComplete();
});
}
private Map<String, String> createParameters(AuthenticationException authException) {

View File

@ -91,6 +91,14 @@ public class BearerTokenServerAuthenticationEntryPointTests {
assertThat(getResponse().getStatusCode()).isEqualTo(HttpStatus.BAD_REQUEST);
}
@Test
public void commenceWhenNoSubscriberThenNothingHappens() {
this.entryPoint.commence(this.exchange, new BadCredentialsException(""));
assertThat(getResponse().getHeaders()).isEmpty();
assertThat(getResponse().getStatusCode()).isNull();
}
private MockServerHttpResponse getResponse() {
return this.exchange.getResponse();
}