Polish #7116

spring-projects · Aug 22, 2019 · f0515a0 · f0515a0
1 parent 2ddab8b
commit f0515a0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...ork/security/oauth2/client/endpoint/WebClientReactiveRefreshTokenTokenResponseClient.java b/...ork/security/oauth2/client/endpoint/WebClientReactiveRefreshTokenTokenResponseClient.java
@@ -18,6 +18,7 @@
 import org.springframework.core.io.buffer.DataBuffer;
 import org.springframework.core.io.buffer.DataBufferUtils;
 import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.security.oauth2.client.registration.ClientRegistration;
 import org.springframework.security.oauth2.core.AuthorizationGrantType;
@@ -66,7 +67,8 @@ public Mono<OAuth2AccessTokenResponse> getTokenResponse(OAuth2RefreshTokenGrantR
 					.body(tokenRequestBody(refreshTokenGrantRequest))
 					.exchange()
 					.flatMap(response -> {
-						if (!response.statusCode().is2xxSuccessful()) {
+						HttpStatus status = HttpStatus.resolve(response.rawStatusCode());
+						if (status == null || !status.is2xxSuccessful()) {
 							OAuth2Error oauth2Error = new OAuth2Error(INVALID_TOKEN_RESPONSE_ERROR_CODE,
 									"An error occurred while attempting to retrieve the OAuth 2.0 Access Token Response: " +
 											"HTTP Status Code " + response.rawStatusCode(), null);