[Flaky-test] Fix ForwardIndexHandlerTest.testAddOtherIndexForForwardI…

…ndexDisabledColumn()
apache · Dec 2, 2024 · 890b077 · 890b077
1 parent 0ccc647
commit 890b077
Showing 1 changed file with 3 additions and 4 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
@@ -1949,12 +1949,11 @@ public void testAddOtherIndexForForwardIndexDisabledColumn()
         metadata.getMaxNumberOfMultiValues(), metadata.getTotalNumberOfEntries(), metadata.isAutoGenerated(),
         metadata.getMinValue(), metadata.getMaxValue(), true);
 
-    // Validate that expected metadata properties don't match. totalNumberOfEntries will definitely not match since
-    // duplicates will be removed, but maxNumberOfMultiValues may still match if the row with max multi-values didn't
-    // have any duplicates.
+    // Validate that all the duplicates are removed
     SegmentMetadataImpl segmentMetadata = new SegmentMetadataImpl(INDEX_DIR);
     ColumnMetadata columnMetadata = segmentMetadata.getColumnMetadataFor(column);
-    assertNotEquals(metadata.getTotalNumberOfEntries(), columnMetadata.getTotalNumberOfEntries());
+    assertEquals(columnMetadata.getMaxNumberOfMultiValues(), 1);
+    assertEquals(columnMetadata.getTotalNumberOfEntries(), metadata.getTotalDocs());
   }
 
   @Test