pipeline name missing from integration tests

Signed-off-by: Maxwell Brown <[email protected]>
Galactus22625 · Dec 19, 2024 · fcbbaf6 · fcbbaf6
1 parent 8450dfe
commit fcbbaf6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...grationTest/java/org/opensearch/dataprepper/plugins/sink/opensearch/OpenSearchSinkIT.java b/...grationTest/java/org/opensearch/dataprepper/plugins/sink/opensearch/OpenSearchSinkIT.java
@@ -151,6 +151,7 @@ public class OpenSearchSinkIT {
     private PluginConfigObservable pluginConfigObservable;
 
     public OpenSearchSink createObjectUnderTest(OpenSearchSinkConfig openSearchSinkConfig, boolean doInitialize) {
+        when(pipelineDescription.getPipelineName()).thenReturn(PIPELINE_NAME);
         OpenSearchSink sink = new OpenSearchSink(
                 pluginSetting, null, expressionEvaluator, awsCredentialsSupplier, pipelineDescription, pluginConfigObservable, openSearchSinkConfig);
         if (doInitialize) {
@@ -163,6 +164,7 @@ public OpenSearchSink createObjectUnderTestWithSinkContext(OpenSearchSinkConfig
         sinkContext = mock(SinkContext.class);
         testTagsTargetKey = RandomStringUtils.randomAlphabetic(5);
         when(sinkContext.getTagsTargetKey()).thenReturn(testTagsTargetKey);
+        when(pipelineDescription.getPipelineName()).thenReturn(PIPELINE_NAME);
         OpenSearchSink sink = new OpenSearchSink(
                 pluginSetting, null, expressionEvaluator, awsCredentialsSupplier, pipelineDescription, pluginConfigObservable, openSearchSinkConfig);
         if (doInitialize) {