fix unit test (opensearch-project#12817) (opensearch-project#12932)

opensearch-trigger-bot[bot] · github-actions[bot] · web-flow · commit 42f00ba55e8b · 2024-03-26T16:45:17.000-04:00
(cherry picked from commit 10fc755) Signed-off-by: kkewwei <kkewwei@163.com> Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
diff --git a/server/src/test/java/org/opensearch/cluster/service/TaskBatcherTests.java b/server/src/test/java/org/opensearch/cluster/service/TaskBatcherTests.java
@@ -218,7 +218,8 @@ public void testTasksAreExecutedInOrder() throws BrokenBarrierException, Interru
             executors[i] = new TaskExecutor();
         }
 
-        int tasksSubmittedPerThread = randomIntBetween(2, 1024);
+        // it will create at most 8192 threads, which will cause native memory oom. so we limit the number of created threads.
+        int tasksSubmittedPerThread = randomIntBetween(2, 128);
 
         CopyOnWriteArrayList<Tuple<String, Throwable>> failures = new CopyOnWriteArrayList<>();
         CountDownLatch updateLatch = new CountDownLatch(numberOfThreads * tasksSubmittedPerThread);
@@ -286,7 +287,7 @@ public void testNoTasksAreDroppedInParallelSubmission() throws BrokenBarrierExce
             executors[i] = new TaskExecutor();
         }
 
-        int tasksSubmittedPerThread = randomIntBetween(2, 1024);
+        int tasksSubmittedPerThread = randomIntBetween(2, 128);
 
         CopyOnWriteArrayList<Tuple<String, Throwable>> failures = new CopyOnWriteArrayList<>();
         CountDownLatch updateLatch = new CountDownLatch(numberOfThreads * tasksSubmittedPerThread);

Original file line number	Diff line number	Diff line change
`@@ -218,7 +218,8 @@ public void testTasksAreExecutedInOrder() throws BrokenBarrierException, Interru`
`218`	`218`	`executors[i] = new TaskExecutor();`
`219`	`219`	`}`
`220`	`220`
`221`		`- int tasksSubmittedPerThread = randomIntBetween(2, 1024);`
	`221`	`+ // it will create at most 8192 threads, which will cause native memory oom. so we limit the number of created threads.`
	`222`	`+ int tasksSubmittedPerThread = randomIntBetween(2, 128);`
`222`	`223`
`223`	`224`	`CopyOnWriteArrayList<Tuple<String, Throwable>> failures = new CopyOnWriteArrayList<>();`
`224`	`225`	`CountDownLatch updateLatch = new CountDownLatch(numberOfThreads * tasksSubmittedPerThread);`
`@@ -286,7 +287,7 @@ public void testNoTasksAreDroppedInParallelSubmission() throws BrokenBarrierExce`
`286`	`287`	`executors[i] = new TaskExecutor();`
`287`	`288`	`}`
`288`	`289`
`289`		`- int tasksSubmittedPerThread = randomIntBetween(2, 1024);`
	`290`	`+ int tasksSubmittedPerThread = randomIntBetween(2, 128);`
`290`	`291`
`291`	`292`	`CopyOnWriteArrayList<Tuple<String, Throwable>> failures = new CopyOnWriteArrayList<>();`
`292`	`293`	`CountDownLatch updateLatch = new CountDownLatch(numberOfThreads * tasksSubmittedPerThread);`