Fix NPE in HttpResponseSnippet when response has custom status

See gh-653

Author: Florian Ferreira

spring-restdocs-core/src/main/java/org/springframework/restdocs/http/HttpResponseSnippet.java

@@ -59,9 +59,15 @@ protected Map<String, Object> createModel(Operation operation) {
 		HttpStatus status = response.getStatus();
 		Map<String, Object> model = new HashMap<>();
 		model.put("responseBody", responseBody(response));
-		model.put("statusCode", status.value());
-		model.put("statusReason", status.getReasonPhrase());
 		model.put("headers", headers(response));
+		if (status != null) {
+			model.put("statusCode", status.value());
+			model.put("statusReason", status.getReasonPhrase());
+		}
+		else {
+			model.put("statusCode", response.getStatusCode());
+			model.put("statusReason", "Http custom status.");
+		}
 		return model;
 	}
 

spring-restdocs-core/src/test/java/org/springframework/restdocs/AbstractSnippetTests.java

@@ -89,6 +89,10 @@ public HttpResponseCondition httpResponse(HttpStatus responseStatus) {
 		return SnippetConditions.httpResponse(this.templateFormat, responseStatus);
 	}
 
+	public HttpResponseCondition httpResponse(Integer responseStatusCode, String responseStatusReason) {
+		return SnippetConditions.httpResponse(this.templateFormat, responseStatusCode, responseStatusReason);
+	}
+
 	protected FileSystemResource snippetResource(String name) {
 		return new FileSystemResource(
 				"src/test/resources/custom-snippet-templates/" + this.templateFormat.getId() + "/" + name + ".snippet");

spring-restdocs-core/src/test/java/org/springframework/restdocs/http/HttpResponseSnippetTests.java

@@ -97,4 +97,9 @@ public void responseWithCustomSnippetAttributes() throws IOException {
 		assertThat(this.generatedSnippets.httpResponse()).contains("Title for the response");
 	}
 
+	@Test
+	public void responseWithCustomStatus() throws IOException {
+		new HttpResponseSnippet().document(this.operationBuilder.response().status(215).build());
+		assertThat(this.generatedSnippets.httpResponse()).is(httpResponse(215,"Http custom status."));
+	}
 }

spring-restdocs-core/src/test/java/org/springframework/restdocs/test/SnippetConditions.java

@@ -71,6 +71,13 @@ public static HttpResponseCondition httpResponse(TemplateFormat format, HttpStat
 		return new HttpResponseCondition(status, new MarkdownCodeBlockCondition<>("http"), 2);
 	}
 
+	public static HttpResponseCondition httpResponse(TemplateFormat format, Integer responseStatusCode, String responseStatusReason) {
+		if ("adoc".equals(format.getFileExtension())) {
+			return new HttpResponseCondition(responseStatusCode, responseStatusReason, new AsciidoctorCodeBlockCondition<>("http", "nowrap"), 3);
+		}
+		return new HttpResponseCondition(responseStatusCode, responseStatusReason, new MarkdownCodeBlockCondition<>("http"), 2);
+	}
+
 	@SuppressWarnings({ "rawtypes" })
 	public static CodeBlockCondition<?> codeBlock(TemplateFormat format, String language) {
 		if ("adoc".equals(format.getFileExtension())) {
@@ -235,6 +242,12 @@ private HttpResponseCondition(HttpStatus status, CodeBlockCondition<?> delegate,
 			this.content("");
 		}
 
+		private HttpResponseCondition(Integer responseStatusCode, String responseStatusReason, CodeBlockCondition<?> delegate, int headerOffset) {
+			super(delegate, headerOffset);
+			this.content("HTTP/1.1 " + responseStatusCode + " " + responseStatusReason);
+			this.content("");
+		}
+
 	}
 
 	/**