assert chunk count

Author: original-brownbear

server/src/test/java/org/elasticsearch/action/admin/indices/mapping/get/GetMappingsResponseTests.java

@@ -18,6 +18,9 @@
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Objects;
+import java.util.function.Function;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
 
 public class GetMappingsResponseTests extends AbstractWireSerializingTestCase<GetMappingsResponse> {
 
@@ -63,6 +66,22 @@ protected GetMappingsResponse createTestInstance() {
         return resp;
     }
 
+    public void testChunkedXContentUsesChunkPerIndex() {
+        final int indexCount = randomIntBetween(1, 10);
+        final var response = new GetMappingsResponse(
+            IntStream.range(0, indexCount)
+                .mapToObj(i -> "index-" + i)
+                .collect(Collectors.toUnmodifiableMap(Function.identity(), k -> createMappingsForIndex()))
+        );
+        final var chunks = response.toXContentChunked();
+        int chunkCount = 0;
+        while (chunks.hasNext()) {
+            chunks.next();
+            chunkCount++;
+        }
+        assertEquals(indexCount, chunkCount);
+    }
+
     // Not meant to be exhaustive
     private static Map<String, Object> randomFieldMapping() {
         Map<String, Object> mappings = new HashMap<>();