Check build

Author: nik9000

server/src/test/java/org/elasticsearch/index/mapper/blockloader/KeywordFieldBlockLoaderTests.java

@@ -9,6 +9,10 @@
 
 package org.elasticsearch.index.mapper.blockloader;
 
+import com.carrotsearch.randomizedtesting.annotations.Repeat;
+
+import com.carrotsearch.randomizedtesting.annotations.Seed;
+
 import org.apache.lucene.util.BytesRef;
 import org.elasticsearch.datageneration.FieldType;
 import org.elasticsearch.index.mapper.BlockLoaderTestCase;

test/framework/src/main/java/org/elasticsearch/index/mapper/TestBlock.java

@@ -47,13 +47,32 @@ public BooleansBuilder appendBoolean(boolean value) {
                         add(value);
                         return this;
                     }
+
+                    @Override
+                    public TestBlock build() {
+                        TestBlock result = super.build();
+                        List<?> r = (List<?>) result.values.get(0);
+                        assertThat(r, hasSize(expectedCount));
+                        return result;
+                    }
                 }
                 return new BooleansBuilder();
             }
 
             @Override
             public BlockLoader.BytesRefBuilder bytesRefsFromDocValues(int expectedCount) {
-                return bytesRefs(expectedCount);
+                class BytesRefsFromDocValuesBuilder extends TestBlock.Builder implements BlockLoader.BytesRefBuilder {
+                    private BytesRefsFromDocValuesBuilder() {
+                        super(1);
+                    }
+
+                    @Override
+                    public BytesRefsFromDocValuesBuilder appendBytesRef(BytesRef value) {
+                        add(BytesRef.deepCopyOf(value));
+                        return this;
+                    }
+                }
+                return new BytesRefsFromDocValuesBuilder();
             }
 
             @Override