#424: added "hal+json" as compatible content type for ArrayLimitingJsonContentModifier (#425)

Author: mustaphazorgati

spring-auto-restdocs-core/src/main/java/capital/scalable/restdocs/response/JsonContentModifier.java

@@ -19,9 +19,9 @@
  */
 package capital.scalable.restdocs.response;
 
-import static org.springframework.http.MediaType.APPLICATION_JSON;
-
 import java.io.IOException;
+import java.util.Arrays;
+import java.util.List;
 
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
@@ -30,16 +30,16 @@
 
 public abstract class JsonContentModifier implements ContentModifier {
 
-    private ObjectMapper objectMapper;
+    private static final List<MediaType> supportedMediaTypes = Arrays.asList(MediaType.APPLICATION_JSON, MediaType.valueOf("application/*+json"));
+    private final ObjectMapper objectMapper;
 
     public JsonContentModifier(ObjectMapper objectMapper) {
         this.objectMapper = objectMapper;
     }
 
     @Override
     public byte[] modifyContent(byte[] originalContent, MediaType contentType) {
-        if (originalContent.length > 0 && contentType != null
-                && contentType.includes(APPLICATION_JSON)) {
+        if (originalContent.length > 0 && supportedMediaTypes.stream().anyMatch(type -> type.isCompatibleWith(contentType))) {
             return modifyContent(originalContent);
         } else {
             return originalContent;

spring-auto-restdocs-core/src/test/java/capital/scalable/restdocs/response/ResponseModifyingPreprocessorsTest.java

@@ -73,6 +73,26 @@ public void shouldShortenJsonArrays() {
                 .expectedResult("[1,2,3]"));
     }
 
+    @Test
+    public void shouldShortenJsonArraysForHALMediaType() {
+        preprocessorTest(new TestData()
+                .content("[1,2,3,4]")
+                .contentType("application/hal+json")
+                .preprocessor(
+                        ResponseModifyingPreprocessors.limitJsonArrayLength(mapper))
+                .expectedResult("[1,2,3]"));
+    }
+
+    @Test
+    public void shouldShortenJsonArraysForAnyJsonSuffixMediaType() {
+        preprocessorTest(new TestData()
+                .content("[1,2,3,4]")
+                .contentType("application/whatever+json")
+                .preprocessor(
+                        ResponseModifyingPreprocessors.limitJsonArrayLength(mapper))
+                .expectedResult("[1,2,3]"));
+    }
+
     @Test
     public void shouldShortenNestedJsonArrays() {
         preprocessorTest(new TestData()