Add test for KafkaEventRecord headers

Peiyu Wu · Peiyu Wu · commit 561ed44a1afc · 2021-07-22T00:20:45.000-07:00
diff --git a/aws-lambda-java-tests/src/test/java/com/amazonaws/services/lambda/runtime/tests/EventLoaderTest.java b/aws-lambda-java-tests/src/test/java/com/amazonaws/services/lambda/runtime/tests/EventLoaderTest.java
@@ -82,6 +82,9 @@ public void testLoadKafkaEvent() {
 
         KafkaEvent.KafkaEventRecord record = event.getRecords().get("mytopic-01").get(0);
         assertThat(record.getValue()).decodedAsBase64().asString().isEqualTo("Hello from Kafka !!");
+         
+        String headerValue = new String(record.getHeaders().get(0).get("headerKey"));    
+        assertThat(headerValue).isEqualTo("headerValue");
     }
 
     @Test
diff --git a/aws-lambda-java-tests/src/test/resources/kafka_event.json b/aws-lambda-java-tests/src/test/resources/kafka_event.json
@@ -10,7 +10,24 @@
         "offset": 15,
         "timestamp": 1596480920837,
         "timestampType": "CREATE_TIME",
-        "value": "SGVsbG8gZnJvbSBLYWZrYSAhIQ=="
+        "value": "SGVsbG8gZnJvbSBLYWZrYSAhIQ==",
+        "headers": [
+          {
+            "headerKey": [
+              104,
+              101,
+              97,
+              100,
+              101,
+              114,
+              86,
+              97,
+              108,
+              117,
+              101
+            ]
+          }
+        ]
       }
     ]
   }

Original file line number	Diff line number	Diff line change
`@@ -82,6 +82,9 @@ public void testLoadKafkaEvent() {`
`82`	`82`
`83`	`83`	`KafkaEvent.KafkaEventRecord record = event.getRecords().get("mytopic-01").get(0);`
`84`	`84`	`assertThat(record.getValue()).decodedAsBase64().asString().isEqualTo("Hello from Kafka !!");`
	`85`	`+`
	`86`	`+ String headerValue = new String(record.getHeaders().get(0).get("headerKey"));`
	`87`	`+ assertThat(headerValue).isEqualTo("headerValue");`
`85`	`88`	`}`
`86`	`89`
`87`	`90`	`@Test`