DRILL-7100: Fixed IllegalArgumentException when reading Parquet data

1 files changed, 6 insertions, 6 deletions

diff --git a/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java b/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java
index 896675a3c..4cf150372 100644
--- a/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java
+++ b/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java
@@ -93,9 +93,9 @@ public class TestBatchSizingMemoryUtil extends PhysicalOpUnitTestBase {
 
     for (int columnIdx = 0; columnIdx < 3; columnIdx++) {
       final ColumnMemoryUsageInfo columnInfo = columnMemoryInfo[columnIdx];
-      final int remainingBitsCapacity = getRemainingBitsCapacity(columnInfo);
-      final int remainingOffsetsCapacity = getRemainingOffsetsCapacity(columnInfo);
-      final int remainingDataCapacity = getRemainingDataCapacity(columnInfo);
+      final long remainingBitsCapacity = getRemainingBitsCapacity(columnInfo);
+      final long remainingOffsetsCapacity = getRemainingOffsetsCapacity(columnInfo);
+      final long remainingDataCapacity = getRemainingDataCapacity(columnInfo);
 
       // Test current VV is within quota (since we are not adding new entries)
       Assert.assertTrue(BatchSizingMemoryUtil.canAddNewData(columnInfo, 0, 0, 0));
@@ -152,15 +152,15 @@ public class TestBatchSizingMemoryUtil extends PhysicalOpUnitTestBase {
     return result;
   }
 
-  private static int getRemainingBitsCapacity(ColumnMemoryUsageInfo columnInfo) {
+  private static long getRemainingBitsCapacity(ColumnMemoryUsageInfo columnInfo) {
     return columnInfo.vectorMemoryUsage.bitsBytesCapacity - columnInfo.vectorMemoryUsage.bitsBytesUsed;
   }
 
-  private static int getRemainingOffsetsCapacity(ColumnMemoryUsageInfo columnInfo) {
+  private static long getRemainingOffsetsCapacity(ColumnMemoryUsageInfo columnInfo) {
     return columnInfo.vectorMemoryUsage.offsetsByteCapacity - columnInfo.vectorMemoryUsage.offsetsBytesUsed;
   }
 
-  private static int getRemainingDataCapacity(ColumnMemoryUsageInfo columnInfo) {
+  private static long getRemainingDataCapacity(ColumnMemoryUsageInfo columnInfo) {
     return columnInfo.vectorMemoryUsage.dataByteCapacity - columnInfo.vectorMemoryUsage.dataBytesUsed;
   }