Fix size of the bulks used in BulkInferenceExecutorTests (#129640) (#129687)

afoucret · web-flow · commit 2cb2308a66f7 · 2025-06-19T17:44:09.000+10:00
diff --git a/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/inference/bulk/BulkInferenceExecutorTests.java b/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/inference/bulk/BulkInferenceExecutorTests.java
@@ -61,7 +61,7 @@ public void shutdownThreadPool() {
     }
 
     public void testSuccessfulExecution() throws Exception {
-        List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 10_000));
+        List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 1000));
         List<InferenceAction.Response> responses = randomInferenceResponseList(requests.size());
 
         InferenceRunner inferenceRunner = mockInferenceRunner(invocation -> {
@@ -93,7 +93,7 @@ public void testSuccessfulExecutionOnEmptyRequest() throws Exception {
     }
 
     public void testInferenceRunnerAlwaysFails() throws Exception {
-        List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 10_000));
+        List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 1000));
 
         InferenceRunner inferenceRunner = mock(invocation -> {
             runWithRandomDelay(() -> {
@@ -115,7 +115,7 @@ public void testInferenceRunnerAlwaysFails() throws Exception {
     }
 
     public void testInferenceRunnerSometimesFails() throws Exception {
-        List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 10_000));
+        List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 1000));
 
         InferenceRunner inferenceRunner = mockInferenceRunner(invocation -> {
             ActionListener<InferenceAction.Response> listener = invocation.getArgument(1);

Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@ public void shutdownThreadPool() {`
`61`	`61`	`}`
`62`	`62`
`63`	`63`	`public void testSuccessfulExecution() throws Exception {`
`64`		`- List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 10_000));`
	`64`	`+ List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 1000));`
`65`	`65`	`List<InferenceAction.Response> responses = randomInferenceResponseList(requests.size());`
`66`	`66`
`67`	`67`	`InferenceRunner inferenceRunner = mockInferenceRunner(invocation -> {`
`@@ -93,7 +93,7 @@ public void testSuccessfulExecutionOnEmptyRequest() throws Exception {`
`93`	`93`	`}`
`94`	`94`
`95`	`95`	`public void testInferenceRunnerAlwaysFails() throws Exception {`
`96`		`- List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 10_000));`
	`96`	`+ List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 1000));`
`97`	`97`
`98`	`98`	`InferenceRunner inferenceRunner = mock(invocation -> {`
`99`	`99`	`runWithRandomDelay(() -> {`
`@@ -115,7 +115,7 @@ public void testInferenceRunnerAlwaysFails() throws Exception {`
`115`	`115`	`}`
`116`	`116`
`117`	`117`	`public void testInferenceRunnerSometimesFails() throws Exception {`
`118`		`- List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 10_000));`
	`118`	`+ List<InferenceAction.Request> requests = randomInferenceRequestList(between(1, 1000));`
`119`	`119`
`120`	`120`	`InferenceRunner inferenceRunner = mockInferenceRunner(invocation -> {`
`121`	`121`	`ActionListener<InferenceAction.Response> listener = invocation.getArgument(1);`