fix flaky UT to avoid IllegalArgumentException: bound must be positive (

#9799)
apache · Nov 15, 2022 · 1ac7fb3 · 1ac7fb3
1 parent e307106
commit 1ac7fb3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...est/java/org/apache/pinot/segment/local/segment/index/loader/ForwardIndexHandlerTest.java b/...est/java/org/apache/pinot/segment/local/segment/index/loader/ForwardIndexHandlerTest.java
@@ -297,7 +297,7 @@ private List<GenericRow> createTestData() {
     byte[][] tempBytesRows = new byte[rowLength][];
     BigDecimal[] tempBigDecimalRows = new BigDecimal[rowLength];
 
-    int maxNumberOfMVEntries = random.nextInt(500);
+    int maxNumberOfMVEntries = random.nextInt(500) + 1;
     String[][] tempMVStringRows = new String[rowLength][maxNumberOfMVEntries];
     Integer[][] tempMVIntRows = new Integer[rowLength][maxNumberOfMVEntries];
     Long[][] tempMVLongRows = new Long[rowLength][maxNumberOfMVEntries];
@@ -314,7 +314,7 @@ private List<GenericRow> createTestData() {
         tempBigDecimalRows[i] = BASE_BIG_DECIMAL.add(BigDecimal.valueOf(1001));
 
         // Avoid creating empty arrays.
-        int numMVElements = random.nextInt(maxNumberOfMVEntries - 1) + 1;
+        int numMVElements = random.nextInt(maxNumberOfMVEntries) + 1;
         for (int j = 0; j < numMVElements; j++) {
           tempMVIntRows[i][j] = 1001;
           tempMVLongRows[i][j] = 1001L;
@@ -330,7 +330,7 @@ private List<GenericRow> createTestData() {
         tempBigDecimalRows[i] = BASE_BIG_DECIMAL.add(BigDecimal.valueOf(i));
 
         // Avoid creating empty arrays.
-        int numMVElements = random.nextInt(maxNumberOfMVEntries - 1) + 1;
+        int numMVElements = random.nextInt(maxNumberOfMVEntries) + 1;
         for (int j = 0; j < numMVElements; j++) {
           tempMVIntRows[i][j] = j;
           tempMVLongRows[i][j] = (long) j;