S3-Source Bug Fix: Handle null object size for non S3-Create Events (#5449)

Signed-off-by: Jeremy Michael <[email protected]>

Author: jmsusanto

data-prepper-plugins/s3-source/src/main/java/org/opensearch/dataprepper/plugins/source/s3/parser/ParsedMessage.java

@@ -37,7 +37,8 @@ public ParsedMessage(final Message message, final boolean failedParsing) {
          // S3EventNotification contains only one S3EventNotificationRecord
         this.bucketName = notificationRecords.get(0).getS3().getBucket().getName();
         this.objectKey = notificationRecords.get(0).getS3().getObject().getUrlDecodedKey();
-        this.objectSize = notificationRecords.get(0).getS3().getObject().getSizeAsLong();
+        Long size = notificationRecords.get(0).getS3().getObject().getSizeAsLong();
+        this.objectSize = (size != null) ? size : 0L;
         this.eventName = notificationRecords.get(0).getEventName();
         this.eventTime = notificationRecords.get(0).getEventTime();
         this.failedParsing = false;

data-prepper-plugins/s3-source/src/test/java/org/opensearch/dataprepper/plugins/source/s3/parser/ParsedMessageTest.java

@@ -134,6 +134,13 @@ void toString_with_messageId() {
             assertThat(actualString, containsString(messageId));
             assertThat(actualString, containsString(testDecodedObjectKey));
         }
+
+        @Test
+        void test_parsed_message_with_null_object_size_defaults_to_zero() {
+            when(s3ObjectEntity.getSizeAsLong()).thenReturn(null);
+            final ParsedMessage parsedMessage = createObjectUnderTest();
+            assertThat(parsedMessage.getObjectSize(), equalTo(0L));
+        }
     }
 
     @Nested