diff --git a/spring-cloud-contract-wiremock/src/main/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippet.java b/spring-cloud-contract-wiremock/src/main/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippet.java index c0a4036841..f9fee052f8 100644 --- a/spring-cloud-contract-wiremock/src/main/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippet.java +++ b/spring-cloud-contract-wiremock/src/main/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippet.java @@ -73,9 +73,10 @@ public class ContractDslSnippet extends TemplatedSnippet { model.put("response_status", response.getStatus().value()); model.put("response_body_present", response.getContent().length > 0); model.put("response_body", response.getContentAsString()); - HttpHeaders headers = response.getHeaders(); + Map headers = response.getHeaders().toSingleValueMap(); + filterHeaders(headers); model.put("response_headers_present", !headers.isEmpty()); - model.put("response_headers", headers.toSingleValueMap().entrySet()); + model.put("response_headers", headers.entrySet()); @SuppressWarnings("unchecked") Set jsonPaths = (Set) operation.getAttributes() .get("contract.jsonPaths"); model.put("response_json_paths_present", jsonPaths != null && !jsonPaths.isEmpty()); diff --git a/spring-cloud-contract-wiremock/src/test/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippetTests.java b/spring-cloud-contract-wiremock/src/test/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippetTests.java index 0c54f79533..18fcad3383 100644 --- a/spring-cloud-contract-wiremock/src/test/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippetTests.java +++ b/spring-cloud-contract-wiremock/src/test/java/org/springframework/cloud/contract/wiremock/restdocs/ContractDslSnippetTests.java @@ -84,6 +84,8 @@ public class ContractDslSnippetTests { then(parsedContract.getRequest().getHeaders().getEntries()).isNotNull(); then(headerNames(parsedContract.getRequest().getHeaders().getEntries())).doesNotContain (HttpHeaders.HOST, HttpHeaders.CONTENT_LENGTH); + then(headerNames(parsedContract.getResponse().getHeaders().getEntries())).doesNotContain + (HttpHeaders.HOST, HttpHeaders.CONTENT_LENGTH); then(parsedContract.getRequest().getMethod().getClientValue()).isNotNull(); then(parsedContract.getRequest().getUrl().getClientValue()).isNotNull(); then(parsedContract.getRequest().getUrl().getClientValue().toString()).startsWith("/");