Added test cases

Signed-off-by: Krishna Kondaka <[email protected]>
opensearch-project · Oct 14, 2024 · 4a6f69e · 4a6f69e
1 parent d20ca87
commit 4a6f69e
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 4 deletions.
diff --git a/...lugins/s3-sink/src/main/java/org/opensearch/dataprepper/plugins/sink/s3/S3SinkConfig.java b/...lugins/s3-sink/src/main/java/org/opensearch/dataprepper/plugins/sink/s3/S3SinkConfig.java
@@ -50,7 +50,7 @@ public class S3SinkConfig {
     private ObjectMetadataConfig objectMetadataConfig;
 
     @AssertTrue(message = "Only one of object_metadata and predefined_object_metadata can be used.")
-    private boolean isValidMetadataConfig() {
+    boolean isValidMetadataConfig() {
         // One of them or both should be null
         return (objectMetadataConfig == null || predefinedObjectMetadata == null);
     }
@@ -151,8 +151,8 @@ public ObjectKeyOptions getObjectKeyOptions() {
         return objectKeyOptions;
     }
 
-    public ObjectMetadataConfig getObjectMetadataConfig() {
-        return objectMetadataConfig;
+    public Object getObjectMetadataConfig() {
+        return objectMetadataConfig != null ? objectMetadataConfig : predefinedObjectMetadata;
     }
 
     /**

diff --git a/...ns/s3-sink/src/test/java/org/opensearch/dataprepper/plugins/sink/s3/S3SinkConfigTest.java b/...ns/s3-sink/src/test/java/org/opensearch/dataprepper/plugins/sink/s3/S3SinkConfigTest.java
@@ -55,6 +55,39 @@ void get_bucket_name_with_s3_prefix_test() throws NoSuchFieldException, IllegalA
         assertThat(objectUnderTest.getBucketName(), equalTo(bucketName));
     }
 
+    @Test
+    void test_default_metadata_config() throws NoSuchFieldException, IllegalAccessException {
+        final String bucketName = UUID.randomUUID().toString();
+        final String bucketNameWithPrefix = S3_PREFIX + bucketName;
+        final S3SinkConfig objectUnderTest = new S3SinkConfig();
+        assertThat(objectUnderTest.isValidMetadataConfig(), equalTo(true));
+        assertThat(objectUnderTest.getObjectMetadataConfig(), equalTo(null));
+    }
+
+    @Test
+    void test_object_metadata_config() throws NoSuchFieldException, IllegalAccessException {
+        final String bucketName = UUID.randomUUID().toString();
+        final String bucketNameWithPrefix = S3_PREFIX + bucketName;
+        final S3SinkConfig objectUnderTest = new S3SinkConfig();
+
+        ObjectMetadataConfig objectMetadataConfig = new ObjectMetadataConfig();
+        ReflectivelySetField.setField(S3SinkConfig.class, objectUnderTest, "objectMetadataConfig", objectMetadataConfig);
+        assertThat(objectUnderTest.isValidMetadataConfig(), equalTo(true));
+        assertThat(objectUnderTest.getObjectMetadataConfig(), equalTo(objectMetadataConfig));
+    }
+
+    @Test
+    void test_predefined_metadata_config() throws NoSuchFieldException, IllegalAccessException {
+        final String bucketName = UUID.randomUUID().toString();
+        final String bucketNameWithPrefix = S3_PREFIX + bucketName;
+        final S3SinkConfig objectUnderTest = new S3SinkConfig();
+
+        PredefinedObjectMetadata objectMetadataConfig = new PredefinedObjectMetadata();
+        ReflectivelySetField.setField(S3SinkConfig.class, objectUnderTest, "predefinedObjectMetadata", objectMetadataConfig);
+        assertThat(objectUnderTest.isValidMetadataConfig(), equalTo(true));
+        assertThat(objectUnderTest.getObjectMetadataConfig(), equalTo(objectMetadataConfig));
+    }
+
     @Test
     void get_object_key_test() {
         assertThat("Object key is not an instance of ObjectKeyOptions",
@@ -81,4 +114,4 @@ void get_json_codec_test() {
     void get_client_option_test() {
         assertNull(new S3SinkConfig().getClientOptions());
     }
-}
+}