Adding unit test

Signed-off-by: Thanh Tran <[email protected]>
opensearch-project · Apr 13, 2024 · 24ae806 · 24ae806
1 parent 1ea7f86
commit 24ae806
Showing 1 changed file with 57 additions and 0 deletions.
diff --git a/...ommon/src/test/java/org/opensearch/dataprepper/plugins/otel/codec/OTelProtoCodecTest.java b/...ommon/src/test/java/org/opensearch/dataprepper/plugins/otel/codec/OTelProtoCodecTest.java
@@ -1318,4 +1318,61 @@ public void testBoundsKeyNotEqualsSign() {
         assertNotEquals(k1, k2);
     }
 
+    @Test
+    void testConvertKeysOfDataPointAttributesRemoveDuplicatedKey() {
+        final String keyName = "duplicate_key_name";
+        final String keyValue = "duplicate_value";
+        final KeyValue duplicateAttribute1 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final KeyValue duplicateAttribute2 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final NumberDataPoint actual = NumberDataPoint.newBuilder()
+                .addAllAttributes(Arrays.asList(duplicateAttribute1, duplicateAttribute2)).build();
+
+        Map<String, Object> map = OTelProtoCodec.convertKeysOfDataPointAttributes(actual);
+        assertThat(map.size(), is(equalTo(1)));
+    }
+
+    @Test
+    void testUnpackKeyValueListRemoveDuplicatedKey() {
+        final String keyName = "duplicate_key_name";
+        final String keyValue = "duplicate_value";
+        final KeyValue duplicateAttribute1 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final KeyValue duplicateAttribute2 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final List<KeyValue> actual = Arrays.asList(duplicateAttribute1, duplicateAttribute2);
+
+        Map<String, Object> map = OTelProtoCodec.unpackKeyValueList(actual);
+        assertThat(map.size(), is(equalTo(1)));
+    }
+
+    @Test
+    void testUnpackKeyValueListLogRemoveDuplicatedKey() {
+        final String keyName = "duplicate_key_name";
+        final String keyValue = "duplicate_value";
+        final KeyValue duplicateAttribute1 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final KeyValue duplicateAttribute2 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final List<KeyValue> actual = Arrays.asList(duplicateAttribute1, duplicateAttribute2);        
+
+        Map<String, Object> map = OTelProtoCodec.unpackKeyValueListLog(actual);
+        assertThat(map.size(), is(equalTo(1)));
+    }
+
+    @Test
+    void testUnpackExemplarValueListRemoveDuplicatedKey() {
+        final String keyName = "duplicate_key_name";
+        final String keyValue = "duplicate_value";
+        final KeyValue duplicateAttribute1 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final KeyValue duplicateAttribute2 = KeyValue.newBuilder().setKey(keyName).setValue(AnyValue.newBuilder()
+                .setStringValue(keyValue).build()).build();
+        final List<KeyValue> actual = Arrays.asList(duplicateAttribute1, duplicateAttribute2);        
+
+        Map<String, Object> map = OTelProtoCodec.unpackExemplarValueList(actual);
+        assertThat(map.size(), is(equalTo(1)));
+    }
+
 }