Support for non-standard HTTP status in reactive ClientHttpResponse

Issue: SPR-16748

Author: jhoeller

spring-test/src/main/java/org/springframework/mock/http/client/reactive/MockClientHttpResponse.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2016 the original author or authors.
+ * Copyright 2002-2018 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -40,6 +40,7 @@
 
 /**
  * Mock implementation of {@link ClientHttpResponse}.
+ *
  * @author Brian Clozel
  * @author Rossen Stoyanchev
  * @since 5.0
@@ -68,6 +69,11 @@ public HttpStatus getStatusCode() {
 		return this.status;
 	}
 
+	@Override
+	public int getRawStatusCode() {
+		return this.status.value();
+	}
+
 	@Override
 	public HttpHeaders getHeaders() {
 		String headerName = HttpHeaders.SET_COOKIE;
@@ -138,4 +144,4 @@ private Charset getCharset() {
 		return (charset != null ? charset : StandardCharsets.UTF_8);
 	}
 
-}
+}

spring-web/src/main/java/org/springframework/http/client/reactive/ClientHttpResponse.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2017 the original author or authors.
+ * Copyright 2002-2018 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -31,10 +31,23 @@
 public interface ClientHttpResponse extends ReactiveHttpInputMessage {
 
 	/**
-	 * Return the HTTP status as an {@link HttpStatus} enum value.
+	 * Return the HTTP status code of the response.
+	 * @return the HTTP status as an HttpStatus enum value
+	 * @throws IllegalArgumentException in case of an unknown HTTP status code
+	 * @see HttpStatus#valueOf(int)
 	 */
 	HttpStatus getStatusCode();
 
+	/**
+	 * Return the HTTP status code (potentially non-standard and not
+	 * resolvable through the {@link HttpStatus} enum) as an integer.
+	 * @return the HTTP status as an integer
+	 * @since 5.0.6
+	 * @see #getStatusCode()
+	 * @see HttpStatus#resolve(int)
+	 */
+	int getRawStatusCode();
+
 	/**
 	 * Return a read-only map of response cookies received from the server.
 	 */

spring-web/src/main/java/org/springframework/http/client/reactive/ClientHttpResponseDecorator.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2017 the original author or authors.
+ * Copyright 2002-2018 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -55,6 +55,11 @@ public HttpStatus getStatusCode() {
 		return this.delegate.getStatusCode();
 	}
 
+	@Override
+	public int getRawStatusCode() {
+		return this.delegate.getRawStatusCode();
+	}
+
 	@Override
 	public HttpHeaders getHeaders() {
 		return this.delegate.getHeaders();

spring-web/src/main/java/org/springframework/http/client/reactive/ReactorClientHttpResponse.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2017 the original author or authors.
+ * Copyright 2002-2018 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -62,14 +62,18 @@ public Flux<DataBuffer> getBody() {
 	@Override
 	public HttpHeaders getHeaders() {
 		HttpHeaders headers = new HttpHeaders();
-		this.response.responseHeaders().entries()
-		             .forEach(e -> headers.add(e.getKey(), e.getValue()));
+		this.response.responseHeaders().entries().forEach(e -> headers.add(e.getKey(), e.getValue()));
 		return headers;
 	}
 
 	@Override
 	public HttpStatus getStatusCode() {
-		return HttpStatus.valueOf(this.response.status().code());
+		return HttpStatus.valueOf(getRawStatusCode());
+	}
+
+	@Override
+	public int getRawStatusCode() {
+		return this.response.status().code();
 	}
 
 	@Override
@@ -91,7 +95,7 @@ public MultiValueMap<String, ResponseCookie> getCookies() {
 	public String toString() {
 		return "ReactorClientHttpResponse{" +
 				"request=[" + this.response.method().name() + " " + this.response.uri() + "]," +
-				"status=" + getStatusCode() + '}';
+				"status=" + getRawStatusCode() + '}';
 	}
 
 }

spring-web/src/test/java/org/springframework/mock/http/client/reactive/test/MockClientHttpResponse.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2016 the original author or authors.
+ * Copyright 2002-2018 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -40,6 +40,7 @@
 
 /**
  * Mock implementation of {@link ClientHttpResponse}.
+ *
  * @author Brian Clozel
  * @author Rossen Stoyanchev
  * @since 5.0
@@ -68,6 +69,11 @@ public HttpStatus getStatusCode() {
 		return this.status;
 	}
 
+	@Override
+	public int getRawStatusCode() {
+		return this.status.value();
+	}
+
 	@Override
 	public HttpHeaders getHeaders() {
 		String headerName = HttpHeaders.SET_COOKIE;
@@ -138,4 +144,4 @@ private Charset getCharset() {
 		return (charset != null ? charset : StandardCharsets.UTF_8);
 	}
 
-}
+}

spring-webflux/src/main/java/org/springframework/web/reactive/function/client/ClientRequest.java

@@ -86,7 +86,6 @@ default Optional<Object> attribute(String name) {
 		}
 	}
 
-
 	/**
 	 * Return the attributes of this request.
 	 */
@@ -222,8 +221,7 @@ interface Builder {
 		 * @param <P> the type of the {@code Publisher}
 		 * @return the built request
 		 */
-		<S, P extends Publisher<S>> Builder body(P publisher,
-				ParameterizedTypeReference<S> typeReference);
+		<S, P extends Publisher<S>> Builder body(P publisher, ParameterizedTypeReference<S> typeReference);
 
 		/**
 		 * Set the attribute with the given name to the given value.
@@ -243,8 +241,7 @@ <S, P extends Publisher<S>> Builder body(P publisher,
 		Builder attributes(Consumer<Map<String, Object>> attributesConsumer);
 
 		/**
-		 * Builds the request.
-		 * @return the request
+		 * Build the request.
 		 */
 		ClientRequest build();
 	}

spring-webflux/src/main/java/org/springframework/web/reactive/function/client/ClientResponse.java

@@ -160,13 +160,13 @@ public interface ClientResponse {
 	 * @return the created builder
 	 */
 	static Builder from(ClientResponse other) {
-		Assert.notNull(other, "'other' must not be null");
+		Assert.notNull(other, "Other ClientResponse must not be null");
 		return new DefaultClientResponseBuilder(other);
 	}
 
 	/**
-	 * Create a response builder with the given status code and using default strategies for reading
-	 * the body.
+	 * Create a response builder with the given status code and using default strategies for
+	 * reading the body.
 	 * @param statusCode the status code
 	 * @return the created builder
 	 */
@@ -181,10 +181,7 @@ static Builder create(HttpStatus statusCode) {
 	 * @return the created builder
 	 */
 	static Builder create(HttpStatus statusCode, ExchangeStrategies strategies) {
-		Assert.notNull(statusCode, "'statusCode' must not be null");
-		Assert.notNull(strategies, "'strategies' must not be null");
-		return new DefaultClientResponseBuilder(strategies)
-				.statusCode(statusCode);
+		return new DefaultClientResponseBuilder(strategies).statusCode(statusCode);
 	}
 
 	/**
@@ -194,24 +191,20 @@ static Builder create(HttpStatus statusCode, ExchangeStrategies strategies) {
 	 * @return the created builder
 	 */
 	static Builder create(HttpStatus statusCode, List<HttpMessageReader<?>> messageReaders) {
-		Assert.notNull(statusCode, "'statusCode' must not be null");
-		Assert.notNull(messageReaders, "'messageReaders' must not be null");
-
 		return create(statusCode, new ExchangeStrategies() {
 			@Override
 			public List<HttpMessageReader<?>> messageReaders() {
 				return messageReaders;
 			}
-
 			@Override
 			public List<HttpMessageWriter<?>> messageWriters() {
 				// not used in the response
 				return Collections.emptyList();
 			}
 		});
-
 	}
 
+
 	/**
 	 * Represents the headers of the HTTP response.
 	 * @see ClientResponse#headers()
@@ -243,6 +236,7 @@ interface Headers {
 		HttpHeaders asHttpHeaders();
 	}
 
+
 	/**
 	 * Defines a builder for a response.
 	 */
@@ -295,7 +289,7 @@ interface Builder {
 		Builder cookies(Consumer<MultiValueMap<String, ResponseCookie>> cookiesConsumer);
 
 		/**
-		 * Sets the body of the response. Calling this methods will
+		 * Set the body of the response. Calling this methods will
 		 * {@linkplain org.springframework.core.io.buffer.DataBufferUtils#release(DataBuffer) release}
 		 * the existing body of the builder.
 		 * @param body the new body.
@@ -304,7 +298,7 @@ interface Builder {
 		Builder body(Flux<DataBuffer> body);
 
 		/**
-		 * Sets the body of the response to the UTF-8 encoded bytes of the given string.
+		 * Set the body of the response to the UTF-8 encoded bytes of the given string.
 		 * Calling this methods will
 		 * {@linkplain org.springframework.core.io.buffer.DataBufferUtils#release(DataBuffer) release}
 		 * the existing body of the builder.
@@ -314,9 +308,9 @@ interface Builder {
 		Builder body(String body);
 
 		/**
-		 * Builds the response.
-		 * @return the response
+		 * Build the response.
 		 */
 		ClientResponse build();
 	}
+
 }

spring-webflux/src/main/java/org/springframework/web/reactive/function/client/DefaultClientResponse.java

@@ -61,6 +61,7 @@ public DefaultClientResponse(ClientHttpResponse response, ExchangeStrategies str
 		this.headers = new DefaultHeaders();
 	}
 
+
 	@Override
 	public ExchangeStrategies strategies() {
 		return this.strategies;
@@ -88,12 +89,10 @@ public <T> T body(BodyExtractor<T, ? super ClientHttpResponse> extractor) {
 			public List<HttpMessageReader<?>> messageReaders() {
 				return strategies.messageReaders();
 			}
-
 			@Override
 			public Optional<ServerHttpResponse> serverResponse() {
 				return Optional.empty();
 			}
-
 			@Override
 			public Map<String, Object> hints() {
 				return Collections.emptyMap();
@@ -187,8 +186,7 @@ public <T> Mono<ResponseEntity<List<T>>> toEntityList(Class<T> responseType) {
 	}
 
 	@Override
-	public <T> Mono<ResponseEntity<List<T>>> toEntityList(
-			ParameterizedTypeReference<T> typeReference) {
+	public <T> Mono<ResponseEntity<List<T>>> toEntityList(ParameterizedTypeReference<T> typeReference) {
 		return toEntityListInternal(bodyToFlux(typeReference));
 	}
 
@@ -220,7 +218,7 @@ public Optional<MediaType> contentType() {
 		@Override
 		public List<String> header(String headerName) {
 			List<String> headerValues = delegate().get(headerName);
-			return headerValues != null ? headerValues : Collections.emptyList();
+			return (headerValues != null ? headerValues : Collections.emptyList());
 		}
 
 		@Override
@@ -229,12 +227,13 @@ public HttpHeaders asHttpHeaders() {
 		}
 
 		private OptionalLong toOptionalLong(long value) {
-			return value != -1 ? OptionalLong.of(value) : OptionalLong.empty();
+			return (value != -1 ? OptionalLong.of(value) : OptionalLong.empty());
 		}
-
 	}
 
+
 	@SuppressWarnings("serial")
-	private class ReadCancellationException extends RuntimeException {
+	private static class ReadCancellationException extends RuntimeException {
 	}
+
 }