Add test for writeToStream with offset and size

Unidata · May 23, 2024 · a95b04e · a95b04e
1 parent e9c27a8
commit a95b04e
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/cdm/zarr/src/test/java/thredds/inventory/zarr/TestMFileZip.java b/cdm/zarr/src/test/java/thredds/inventory/zarr/TestMFileZip.java
@@ -55,6 +55,25 @@ public void shouldWriteZipToStream() throws IOException {
         assertThat(outputStream.size()).isEqualTo(mFile.getLength());
       }
     }
+
+    @Test
+    public void shouldWritePartialZipToStream() throws IOException {
+      try (ZipFile zipFile = createTemporaryZipFile("TestWritePartialZip", entrySize, numberOfEntries)) {
+        final MFileZip mFile = new MFileZip(zipFile.getName());
+        final int length = (int) mFile.getLength();
+
+        final int offset = 1;
+        final int maxBytes = 100;
+
+        final ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+
+        final int startPosition = Math.min(offset, length);
+        final int endPosition = Math.min(offset + maxBytes, length);
+
+        mFile.writeToStream(outputStream, offset, maxBytes);
+        assertThat(outputStream.size()).isEqualTo(Math.max(0, endPosition - startPosition));
+      }
+    }
   }
 
   public static class TestMFileZipNonParameterized {