GH-997 Adding test to validate AWS Context set as header

Resolves #997

Author: olegz

spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/test/java/org/springframework/cloud/function/adapter/aws/FunctionInvokerTests.java

@@ -44,6 +44,7 @@
 import com.fasterxml.jackson.databind.ObjectMapper;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import org.mockito.Mockito;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 
@@ -784,7 +785,7 @@ public void testLBEventAsMessage() throws Exception {
 
 		InputStream targetStream = new ByteArrayInputStream(this.sampleLBEvent.getBytes());
 		ByteArrayOutputStream output = new ByteArrayOutputStream();
-		invoker.handleRequest(targetStream, output, null);
+		invoker.handleRequest(targetStream, output, Mockito.mock(Context.class));
 
 		Map result = mapper.readValue(output.toByteArray(), Map.class);
 		assertThat(result.get("body")).isEqualTo("\"Hello from ELB\"");
@@ -1004,26 +1005,6 @@ public void testApiGatewayInAndOutV2() throws Exception {
 		assertThat(headers.get("foo")).isEqualTo("bar");
 	}
 
-//	@SuppressWarnings("rawtypes")
-//	@Test
-//	public void testApiGatewayInAndOutWithException() throws Exception {
-//		System.setProperty("MAIN_CLASS", ApiGatewayConfiguration.class.getName());
-//		System.setProperty("spring.cloud.function.definition", "inputOutputApiEventException");
-//		FunctionInvoker invoker = new FunctionInvoker();
-//
-//		InputStream targetStream = new ByteArrayInputStream(this.apiGatewayEvent.getBytes());
-//		ByteArrayOutputStream output = new ByteArrayOutputStream();
-//		invoker.handleRequest(targetStream, output, null);
-//
-//		Map result = mapper.readValue(output.toByteArray(), Map.class);
-//		assertThat(result.get("body")).isEqualTo("Intentional");
-//
-//		Map headers = (Map) result.get("headers");
-//		assertThat(headers.get("foo")).isEqualTo("bar");
-//	}
-
-
-
 	@SuppressWarnings("rawtypes")
 	@Test
 	public void testApiGatewayEventAsMessage() throws Exception {
@@ -1347,9 +1328,10 @@ public Function<ApplicationLoadBalancerRequestEvent, ApplicationLoadBalancerResp
 
 		@Bean
 		public Function<Message<ApplicationLoadBalancerRequestEvent>, String> inputLBEventAsMessage(JsonMapper jsonMapper) {
-			return v -> {
-				System.out.println("Received: " + v);
-				return v.getPayload().getBody();
+			return message -> {
+				System.out.println("Received: " + message);
+				assertThat(message.getHeaders().get(AWSLambdaUtils.AWS_CONTEXT)).isNotNull();
+				return message.getPayload().getBody();
 			};
 		}