Sns Sink Test Case Changes

Signed-off-by: Uday Kumar Chintala <[email protected]>
opensearch-project · Aug 1, 2023 · c779a1f · c779a1f
1 parent 526b521
commit c779a1f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...ugins/sns-sink/src/test/java/org/opensearch/dataprepper/plugins/sink/sns/SnsSinkTest.java b/...ugins/sns-sink/src/test/java/org/opensearch/dataprepper/plugins/sink/sns/SnsSinkTest.java
@@ -42,7 +42,6 @@ class SnsSinkTest {
     private PluginFactory pluginFactory;
     private AwsCredentialsSupplier awsCredentialsSupplier;
     private SinkContext sinkContext;
-
     private PluginModel pluginModel;
 
     @BeforeEach
@@ -56,6 +55,8 @@ void setUp() {
         awsCredentialsSupplier = mock(AwsCredentialsSupplier.class);
         Map<String,Object> dlqMap = mock(HashMap.class);
 
+        when(snsSinkConfig.getMessageGroupId()).thenReturn("/message");
+        when(snsSinkConfig.getMessageDeduplicationId()).thenReturn("/message");
         when(snsSinkConfig.getDlq()).thenReturn(pluginModel);
         when(pluginModel.getPluginSettings()).thenReturn(dlqMap);
         when(snsSinkConfig.getAwsAuthenticationOptions()).thenReturn(awsAuthenticationOptions);