Fixed s3 sink config validation (#5063)

* Fixed s3 sink config validation Signed-off-by: Krishna Kondaka <[email protected]> * Added test cases Signed-off-by: Krishna Kondaka <[email protected]> --------- Signed-off-by: Krishna Kondaka <[email protected]> Co-authored-by: Krishna Kondaka <[email protected]> (cherry picked from commit 47fdbda)
opensearch-project · Oct 15, 2024 · f9c289c · f9c289c
1 parent a18bc41
commit f9c289c
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 6 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,9 +50,9 @@ public class S3SinkConfig {
     private ObjectMetadataConfig objectMetadataConfig;
 
     @AssertTrue(message = "Only one of object_metadata and predefined_object_metadata can be used.")
-    private boolean isValidMetadataConfig() {
-       return (objectMetadataConfig != null && predefinedObjectMetadata == null) ||
-              (objectMetadataConfig == null && predefinedObjectMetadata != null);
+    boolean isValidMetadataConfig() {
+        // One of them or both should be null
+        return (objectMetadataConfig == null || predefinedObjectMetadata == null);
     }
 
     @AssertTrue(message = "You may not use both bucket and bucket_selector together in one S3 sink.")
@@ -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());
     }
-}
+}