Add test that proves you can write a ByteReference using its iterator

iverase · Oct 11, 2023 · 1c203cc · 1c203cc
1 parent bd4d45a
commit 1c203cc
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/...ramework/src/main/java/org/elasticsearch/common/bytes/AbstractBytesReferenceTestCase.java b/...ramework/src/main/java/org/elasticsearch/common/bytes/AbstractBytesReferenceTestCase.java
@@ -666,4 +666,20 @@ public void testIndexOf() throws IOException {
         final byte missing = randomValueOtherThanMany(map::containsKey, ESTestCase::randomByte);
         assertEquals(-1, bytesReference.indexOf(missing, randomIntBetween(0, Math.max(0, size - 1))));
     }
+
+    public void testWriteWithIterator() throws IOException {
+        final int length = randomIntBetween(1014, 1024 * 1024);
+        final byte[] bytes = new byte[length];
+        random().nextBytes(bytes);
+        final BytesReference bytesReference = randomBytesReference(length);
+        final BytesRefIterator iterator = bytesReference.iterator();
+        BytesRef bytesRef;
+        int offset = 0;
+        while ((bytesRef = iterator.next()) != null) {
+            final int len = Math.min(bytesRef.length, length - offset);
+            System.arraycopy(bytes, offset, bytesRef.bytes, bytesRef.offset, len);
+            offset += len;
+        }
+        assertArrayEquals(bytes, BytesReference.toBytes(bytesReference));
+    }
 }