Increased number of indexed docs to match segment count

Signed-off-by: expani <[email protected]>
expani · Dec 24, 2024 · b5ae870 · b5ae870
1 parent a74e65e
commit b5ae870
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/server/src/test/java/org/opensearch/index/engine/InternalEngineTests.java b/server/src/test/java/org/opensearch/index/engine/InternalEngineTests.java
@@ -724,12 +724,19 @@ public void testMergeSegmentsOnCommitDefault() throws Exception {
             segments = engine.segments(true);
             assertThat(segments.size(), equalTo(1));
 
-            ParsedDocument doc2 = testParsedDocument("2", null, testDocumentWithTextField(), B_2, null);
-            engine.index(indexForDoc(doc2));
-            engine.refresh("test");
-            ParsedDocument doc3 = testParsedDocument("3", null, testDocumentWithTextField(), B_3, null);
-            engine.index(indexForDoc(doc3));
-            engine.refresh("test");
+            // Fails for [2,3] in Lucene 10 due to changes in TieredMergePolicy https://github.com/apache/lucene/pull/266
+            // causing 3 documents to be merged within one segment.
+            for (int i = 2; i <= 4; i++) {
+                ParsedDocument document = testParsedDocument(
+                    String.valueOf(i),
+                    null,
+                    testDocumentWithTextField(),
+                    new BytesArray(new byte[] { (byte) i }),
+                    null
+                );
+                engine.index(indexForDoc(document));
+                engine.refresh("test");
+            }
 
             segments = engine.segments(true);
             assertThat(segments.size(), equalTo(2));