Add test for DynamodbEvent.DynamodbStreamRecord load

Author: mo7ty

aws-lambda-java-tests/src/main/java/com/amazonaws/services/lambda/runtime/tests/EventLoader.java

@@ -65,6 +65,10 @@ public static DynamodbEvent loadDynamoDbEvent(String filename) {
         return loadEvent(filename, DynamodbEvent.class);
     }
 
+    public static DynamodbEvent.DynamodbStreamRecord loadDynamodbStreamRecord(String filename) {
+        return loadEvent(filename, DynamodbEvent.DynamodbStreamRecord.class);
+    }
+
     public static KafkaEvent loadKafkaEvent(String filename) {
         return loadEvent(filename, KafkaEvent.class);
     }

aws-lambda-java-tests/src/test/java/com/amazonaws/services/lambda/runtime/tests/EventLoaderTest.java

@@ -174,7 +174,10 @@ public void testLoadDynamoEvent() {
                 .returns("4421584500000000017450439091", StreamRecord::getSequenceNumber)
                 .returns(26L, StreamRecord::getSizeBytes)
                 .returns("NEW_AND_OLD_IMAGES", StreamRecord::getStreamViewType)
-                .returns(Date.from(ofEpochSecond(1428537600)), StreamRecord::getApproximateCreationDateTime);
+                .returns(Date.from(ofEpochSecond(1428537600)), StreamRecord::getApproximateCreationDateTime)
+                .doesNotReturn(null, from(StreamRecord::getKeys))
+                .doesNotReturn(null, from(StreamRecord::getNewImage))
+                .returns(null, from(StreamRecord::getOldImage));
 
         assertThat(streamRecord.getKeys()).contains(entry("Id", new AttributeValue().withN("101")));
         assertThat(streamRecord.getNewImage()).containsAnyOf(
@@ -183,6 +186,23 @@ public void testLoadDynamoEvent() {
         );
     }
 
+    @Test
+    public void testLoadDynamodbStreamRecord() {
+        DynamodbEvent.DynamodbStreamRecord record = EventLoader.loadDynamodbStreamRecord("dynamodb_streamRecord.json");
+        assertThat(record).isNotNull();
+
+        StreamRecord streamRecord = record.getDynamodb();
+        assertThat(streamRecord)
+                .isNotNull()
+                .returns("4421584500000000017450439092", StreamRecord::getSequenceNumber)
+                .returns(59L, StreamRecord::getSizeBytes)
+                .returns("NEW_AND_OLD_IMAGES", StreamRecord::getStreamViewType)
+                .returns(Date.from(ofEpochSecond(1428537600)), StreamRecord::getApproximateCreationDateTime)
+                .doesNotReturn(null, from(StreamRecord::getKeys))
+                .doesNotReturn(null, from(StreamRecord::getNewImage))
+                .doesNotReturn(null, from(StreamRecord::getOldImage));
+    }
+
     @Test
     public void testLoadKinesisEvent() {
         KinesisEvent event = EventLoader.loadKinesisEvent("kinesis_event.json");

aws-lambda-java-tests/src/test/resources/dynamodb_streamRecord.json

@@ -0,0 +1,35 @@
+{
+    "eventID": "c81e728d9d4c2f636f067f89cc14862c",
+    "eventName": "MODIFY",
+    "eventVersion": "1.1",
+    "eventSource": "aws:dynamodb",
+    "awsRegion": "eu-central-1",
+    "dynamodb": {
+        "Keys": {
+            "Id": {
+                "N": "101"
+            }
+        },
+        "NewImage": {
+            "Message": {
+                "S": "This item has changed"
+            },
+            "Id": {
+                "N": "101"
+            }
+        },
+        "OldImage": {
+            "Message": {
+                "S": "New item!"
+            },
+            "Id": {
+                "N": "101"
+            }
+        },
+        "ApproximateCreationDateTime": 1428537600,
+        "SequenceNumber": "4421584500000000017450439092",
+        "SizeBytes": 59,
+        "StreamViewType": "NEW_AND_OLD_IMAGES"
+    },
+    "eventSourceARN": "arn:aws:dynamodb:eu-central-1:123456789012:table/ExampleTableWithStream/stream/2015-06-27T00:48:05.899"
+}