From fcbbaf66c1da442b31d57a974bf97a7be28f31d8 Mon Sep 17 00:00:00 2001 From: Maxwell Brown Date: Thu, 19 Dec 2024 11:01:23 -0800 Subject: [PATCH] pipeline name missing from integration tests Signed-off-by: Maxwell Brown --- .../dataprepper/plugins/sink/opensearch/OpenSearchSinkIT.java | 2 ++ 1 file changed, 2 insertions(+) diff --git a/data-prepper-plugins/opensearch/src/integrationTest/java/org/opensearch/dataprepper/plugins/sink/opensearch/OpenSearchSinkIT.java b/data-prepper-plugins/opensearch/src/integrationTest/java/org/opensearch/dataprepper/plugins/sink/opensearch/OpenSearchSinkIT.java index 095c649378..a6d52eaa74 100644 --- a/data-prepper-plugins/opensearch/src/integrationTest/java/org/opensearch/dataprepper/plugins/sink/opensearch/OpenSearchSinkIT.java +++ b/data-prepper-plugins/opensearch/src/integrationTest/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) {