fix: avoid null chunk when file size is a multiple of part size

src/main/java/com/box/sdkgen/internal/utils/UtilsManager.java

@@ -34,6 +34,7 @@
 import java.util.List;
 import java.util.Locale;
 import java.util.Map;
+import java.util.NoSuchElementException;
 import java.util.Objects;
 import java.util.Set;
 import java.util.UUID;
@@ -311,41 +312,54 @@ public static String hexToBase64(String hex) {
   public static Iterator<InputStream> iterateChunks(
       InputStream stream, long chunkSize, long fileSize) {
     return new Iterator<InputStream>() {
-      private boolean streamIsFinished = false;
+      private InputStream nextChunk;
+      private boolean isNextChunkPrepared = false;
 
-      @Override
-      public boolean hasNext() {
-        return !streamIsFinished;
-      }
-
-      @Override
-      public InputStream next() {
+      private void prepareNext() {
+        if (isNextChunkPrepared) {
+          return;
+        }
+        isNextChunkPrepared = true;
         try {
           byte[] buffer = new byte[(int) chunkSize];
           int bytesRead = 0;
 
           while (bytesRead < chunkSize) {
             int read = stream.read(buffer, bytesRead, (int) (chunkSize - bytesRead));
             if (read == -1) {
-              // End of stream
-              streamIsFinished = true;
               break;
             }
             bytesRead += read;
           }
 
           if (bytesRead == 0) {
-            // No more data to yield
-            streamIsFinished = true;
-            return null;
+            nextChunk = null;
+            return;
           }
 
-          // Return the chunk as a ByteArrayInputStream
-          return new ByteArrayInputStream(buffer, 0, bytesRead);
-        } catch (Exception e) {
+          nextChunk = new ByteArrayInputStream(buffer, 0, bytesRead);
+        } catch (IOException e) {
           throw new RuntimeException("Error reading from stream", e);
         }
       }
+
+      @Override
+      public boolean hasNext() {
+        prepareNext();
+        return nextChunk != null;
+      }
+
+      @Override
+      public InputStream next() {
+        prepareNext();
+        if (nextChunk == null) {
+          throw new NoSuchElementException();
+        }
+        InputStream result = nextChunk;
+        nextChunk = null;
+        isNextChunkPrepared = false;
+        return result;
+      }
     };
   }
 

src/test/java/com/box/sdkgen/internal/utils/UtilsManagerTest.java

@@ -0,0 +1,57 @@
+package com.box.sdkgen.internal.utils;
+
+import static com.box.sdkgen.internal.utils.UtilsManager.generateByteStream;
+import static com.box.sdkgen.internal.utils.UtilsManager.iterateChunks;
+import static com.box.sdkgen.internal.utils.UtilsManager.readByteStream;
+import static com.box.sdkgen.internal.utils.UtilsManager.reduceIterator;
+
+import java.io.InputStream;
+import java.util.Iterator;
+import org.junit.jupiter.api.Test;
+
+public class UtilsManagerTest {
+
+  @Test
+  public void testIterateChunksWhenFileSizeIsExactMultipleOfChunkSize() {
+    int chunkSize = 8;
+    int fileSize = 16;
+    drainAndAssertChunks(chunkSize, fileSize, fileSize, 2);
+  }
+
+  @Test
+  public void testIterateChunksWhenFileSizeHasRemainder() {
+    int chunkSize = 8;
+    int fileSize = 20;
+    drainAndAssertChunks(chunkSize, fileSize, fileSize, 3);
+  }
+
+  @Test
+  public void testIterateChunksReadsUntilEofWhenReportedFileSizeDoesNotMatch() {
+    int chunkSize = 8;
+    int streamSize = 16;
+    drainAndAssertChunks(chunkSize, streamSize, -1, 2);
+  }
+
+  private static void drainAndAssertChunks(
+      int chunkSize, int streamSize, long reportedFileSize, int expectedChunkCount) {
+    Iterator<InputStream> chunks =
+        iterateChunks(generateByteStream(streamSize), chunkSize, reportedFileSize);
+    ChunkDrainResult result =
+        reduceIterator(chunks, UtilsManagerTest::reduceChunks, new ChunkDrainResult());
+    assert result.chunkCount == expectedChunkCount;
+    assert result.totalBytes == streamSize;
+  }
+
+  private static ChunkDrainResult reduceChunks(ChunkDrainResult acc, InputStream chunk) {
+    assert chunk != null;
+    byte[] bytes = readByteStream(chunk);
+    acc.chunkCount += 1;
+    acc.totalBytes += bytes.length;
+    return acc;
+  }
+
+  private static final class ChunkDrainResult {
+    private int chunkCount;
+    private int totalBytes;
+  }
+}