Use correct exception type in unit tests

Signed-off-by: Chase Engelbrecht <[email protected]>
engechas · Oct 23, 2023 · 6db4177 · 6db4177
1 parent c28d48f
commit 6db4177
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/...va/org/opensearch/dataprepper/plugins/sink/opensearch/index/DynamicIndexManagerTests.java b/...va/org/opensearch/dataprepper/plugins/sink/opensearch/index/DynamicIndexManagerTests.java
@@ -74,6 +74,9 @@ public class DynamicIndexManagerTests {
     @Mock
     private TemplateStrategy templateStrategy;
 
+    @Mock
+    private OpenSearchException openSearchException;
+
     static final String EVENT_TYPE = "event";
 
     @BeforeEach
@@ -195,8 +198,8 @@ public void getIndexName_DoesRetryOnOpenSearchExceptions_UntilSuccess() throws I
         innerIndexManager = mock(IndexManager.class);
         when(mockIndexManagerFactory.getIndexManager(
                 IndexType.CUSTOM, openSearchClient, restHighLevelClient, openSearchSinkConfiguration, templateStrategy, expectedIndexAlias)).thenReturn(innerIndexManager);
-        doThrow(new OpenSearchException(new RuntimeException()))
-                .doThrow(new OpenSearchException(new RuntimeException()))
+        doThrow(openSearchException)
+                .doThrow(openSearchException)
                 .doNothing()
                 .when(innerIndexManager).setupIndex();
         when(innerIndexManager.getIndexName(expectedIndexAlias)).thenReturn(expectedIndexAlias);
@@ -216,8 +219,8 @@ public void getIndexName_DoesRetryOnOpenSearchExceptions_UntilFailure() throws I
         innerIndexManager = mock(IndexManager.class);
         when(mockIndexManagerFactory.getIndexManager(
                 IndexType.CUSTOM, openSearchClient, restHighLevelClient, openSearchSinkConfiguration, templateStrategy, expectedIndexAlias)).thenReturn(innerIndexManager);
-        doThrow(new OpenSearchException(new RuntimeException()))
-                .doThrow(new OpenSearchException(new RuntimeException()))
+        doThrow(openSearchException)
+                .doThrow(openSearchException)
                 .doThrow(new RuntimeException())
                 .when(innerIndexManager).setupIndex();
         when(innerIndexManager.getIndexName(expectedIndexAlias)).thenReturn(expectedIndexAlias);