fix flaky test

yupeng9 · yupeng9 · commit c7ee3749e317 · 2025-04-08T14:19:04.000-07:00
Signed-off-by: Yupeng Fu &lt;yupeng@uber.com&gt;
diff --git a/server/src/test/java/org/opensearch/index/engine/IngestionEngineTests.java b/server/src/test/java/org/opensearch/index/engine/IngestionEngineTests.java
@@ -131,8 +131,10 @@ public void testRecovery() throws IOException {
     }
 
     public void testPushAPIFailures() {
-        assertThrows(IngestionEngineException.class, () -> ingestionEngine.index(Mockito.any(Engine.Index.class)));
-        assertThrows(IngestionEngineException.class, () -> ingestionEngine.delete(Mockito.any(Engine.Delete.class)));
+        Engine.Index indexMock = Mockito.mock(Engine.Index.class);
+        assertThrows(IngestionEngineException.class, () -> ingestionEngine.index(indexMock));
+        Engine.Delete deleteMock = Mockito.mock(Engine.Delete.class);
+        assertThrows(IngestionEngineException.class, () -> ingestionEngine.delete(deleteMock));
     }
 
     public void testCreationFailure() throws IOException {

Original file line number	Diff line number	Diff line change
`@@ -131,8 +131,10 @@ public void testRecovery() throws IOException {`
`131`	`131`	`}`
`132`	`132`
`133`	`133`	`public void testPushAPIFailures() {`
`134`		`- assertThrows(IngestionEngineException.class, () -> ingestionEngine.index(Mockito.any(Engine.Index.class)));`
`135`		`- assertThrows(IngestionEngineException.class, () -> ingestionEngine.delete(Mockito.any(Engine.Delete.class)));`
	`134`	`+ Engine.Index indexMock = Mockito.mock(Engine.Index.class);`
	`135`	`+ assertThrows(IngestionEngineException.class, () -> ingestionEngine.index(indexMock));`
	`136`	`+ Engine.Delete deleteMock = Mockito.mock(Engine.Delete.class);`
	`137`	`+ assertThrows(IngestionEngineException.class, () -> ingestionEngine.delete(deleteMock));`
`136`	`138`	`}`
`137`	`139`
`138`	`140`	`public void testCreationFailure() throws IOException {`