-Support for Sink Codecs

Signed-off-by: omkarmmore95 <[email protected]>

data-prepper-api/src/test/java/org/opensearch/dataprepper/model/codec/OutputCodecTest.java

@@ -0,0 +1,69 @@
+package org.opensearch.dataprepper.model.codec;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.opensearch.dataprepper.model.event.DefaultEventMetadata;
+import org.opensearch.dataprepper.model.event.Event;
+import org.opensearch.dataprepper.model.event.EventMetadata;
+import org.opensearch.dataprepper.model.event.EventType;
+import org.opensearch.dataprepper.model.event.JacksonEvent;
+import org.opensearch.dataprepper.model.log.JacksonLog;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.Arrays;
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.Set;
+import java.util.UUID;
+
+import static org.junit.Assert.assertNotEquals;
+
+public class OutputCodecTest {
+
+    @BeforeEach
+    public void setUp() {
+    }
+
+    @Test
+    public void testWriteMetrics() throws JsonProcessingException {
+        OutputCodec outputCodec = new OutputCodec() {
+            @Override
+            public void start(OutputStream outputStream) throws IOException {
+            }
+
+            @Override
+            public void writeEvent(Event event, OutputStream outputStream, String tagsTargetKey) throws IOException {
+            }
+
+            @Override
+            public void complete(OutputStream outputStream) throws IOException {
+            }
+
+            @Override
+            public String getExtension() {
+                return null;
+            }
+        };
+
+        final Set<String> testTags = Set.of("tag1");
+        final EventMetadata defaultEventMetadata = DefaultEventMetadata.builder().
+                withEventType(EventType.LOG.toString()).
+                withTags(testTags).build();
+        Map<String, Object> json = generateJson();
+        final JacksonEvent event = JacksonLog.builder().withData(json).withEventMetadata(defaultEventMetadata).build();
+        Event tagsToEvent = outputCodec.addTagsToEvent(event, "Tag");
+        assertNotEquals(event.toJsonString(), tagsToEvent.toJsonString());
+    }
+
+    private static Map<String, Object> generateJson() {
+        final Map<String, Object> jsonObject = new LinkedHashMap<>();
+        for (int i = 0; i < 2; i++) {
+            jsonObject.put(UUID.randomUUID().toString(), UUID.randomUUID().toString());
+        }
+        jsonObject.put(UUID.randomUUID().toString(), Arrays.asList(UUID.randomUUID().toString(),
+                UUID.randomUUID().toString(), UUID.randomUUID().toString()));
+        return jsonObject;
+    }
+}