Refactor code to trigger build

Signed-off-by: Liyun Xiu <[email protected]>
opensearch-project · Jun 14, 2024 · 5ed3400 · 5ed3400
1 parent 02dde2a
commit 5ed3400
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/server/src/internalClusterTest/java/org/opensearch/ingest/IngestClientIT.java b/server/src/internalClusterTest/java/org/opensearch/ingest/IngestClientIT.java
@@ -162,6 +162,14 @@ public void testSimulate() throws Exception {
     }
 
     public void testBulkWithIngestFailures() throws Exception {
+        runBulkTestWithRandomDocs(false);
+    }
+
+    public void testBulkWithIngestFailuresWithBatchSize() throws Exception {
+        runBulkTestWithRandomDocs(true);
+    }
+
+    private void runBulkTestWithRandomDocs(boolean shouldSetBatchSize) throws Exception {
         createIndex("index");
 
         BytesReference source = BytesReference.bytes(
@@ -180,7 +188,9 @@ public void testBulkWithIngestFailures() throws Exception {
 
         int numRequests = scaledRandomIntBetween(32, 128);
         BulkRequest bulkRequest = new BulkRequest();
-        bulkRequest.batchSize(numRequests);
+        if (shouldSetBatchSize) {
+            bulkRequest.batchSize(numRequests);
+        }
         for (int i = 0; i < numRequests; i++) {
             IndexRequest indexRequest = new IndexRequest("index").id(Integer.toString(i)).setPipeline("_id");
             indexRequest.source(Requests.INDEX_CONTENT_TYPE, "field", "value", "fail", i % 2 == 0);