Added slow init tests for Spring and SpringBoot 1.x applications that use the asyn initialization wrapper (#210)

Author: sapessi

aws-serverless-java-container-spring/src/test/java/com/amazonaws/serverless/proxy/spring/SlowAppTest.java

@@ -0,0 +1,60 @@
+package com.amazonaws.serverless.proxy.spring;
+
+import com.amazonaws.serverless.exceptions.ContainerInitializationException;
+import com.amazonaws.serverless.proxy.internal.testutils.AwsProxyRequestBuilder;
+import com.amazonaws.serverless.proxy.internal.testutils.MockLambdaContext;
+import com.amazonaws.serverless.proxy.model.AwsProxyRequest;
+import com.amazonaws.serverless.proxy.model.AwsProxyResponse;
+import com.amazonaws.serverless.proxy.spring.springbootslowapp.SBLambdaHandler;
+import com.amazonaws.serverless.proxy.spring.springslowapp.LambdaHandler;
+import com.amazonaws.serverless.proxy.spring.springslowapp.MessageController;
+import com.amazonaws.serverless.proxy.spring.springslowapp.SlowAppConfig;
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.time.Instant;
+
+import static org.junit.Assert.*;
+
+public class SlowAppTest {
+
+    @Test
+    public void springSlowApp_continuesInBackgroundThread_returnsCorrect() {
+        LambdaHandler slowApp = null;
+        try {
+            slowApp = new LambdaHandler();
+        } catch (ContainerInitializationException e) {
+            e.printStackTrace();
+            fail("Exception during initialization");
+        }
+        System.out.println("Start time: " + slowApp.getConstructorTime());
+        assertTrue(slowApp.getConstructorTime() < 10_000);
+        AwsProxyRequest req = new AwsProxyRequestBuilder("/hello", "GET").build();
+        long startRequestTime = Instant.now().toEpochMilli();
+        AwsProxyResponse resp = slowApp.handleRequest(req, new MockLambdaContext());
+        long endRequestTime = Instant.now().toEpochMilli();
+        assertTrue(endRequestTime - startRequestTime > SlowAppConfig.SlowDownInit.INIT_SLEEP_TIME_MS - 10_000);
+        assertEquals(200, resp.getStatusCode());
+        Assert.assertEquals(MessageController.HELLO_MESSAGE, resp.getBody());
+    }
+
+    @Test
+    public void springBootSlowApp_continuesInBackgroundThread_returnsCorrect() {
+        SBLambdaHandler slowApp = null;
+        try {
+            slowApp = new SBLambdaHandler();
+        } catch (ContainerInitializationException e) {
+            e.printStackTrace();
+            fail("Exception during initialization");
+        }
+        System.out.println("Start time: " + slowApp.getConstructorTime());
+        assertTrue(slowApp.getConstructorTime() < 10_000);
+        AwsProxyRequest req = new AwsProxyRequestBuilder("/hello", "GET").build();
+        long startRequestTime = Instant.now().toEpochMilli();
+        AwsProxyResponse resp = slowApp.handleRequest(req, new MockLambdaContext());
+        long endRequestTime = Instant.now().toEpochMilli();
+        assertTrue(endRequestTime - startRequestTime > SlowAppConfig.SlowDownInit.INIT_SLEEP_TIME_MS - 10_000);
+        assertEquals(200, resp.getStatusCode());
+        Assert.assertEquals(MessageController.HELLO_MESSAGE, resp.getBody());
+    }
+}

aws-serverless-java-container-spring/src/test/java/com/amazonaws/serverless/proxy/spring/springbootslowapp/MessageController.java

@@ -0,0 +1,15 @@
+package com.amazonaws.serverless.proxy.spring.springbootslowapp;
+
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+public class MessageController {
+    public static final String HELLO_MESSAGE = "Hello";
+
+    @RequestMapping(path="/hello", method= RequestMethod.GET)
+    public String hello() {
+        return HELLO_MESSAGE;
+    }
+}

aws-serverless-java-container-spring/src/test/java/com/amazonaws/serverless/proxy/spring/springbootslowapp/SBLambdaHandler.java

@@ -0,0 +1,40 @@
+package com.amazonaws.serverless.proxy.spring.springbootslowapp;
+
+
+import com.amazonaws.serverless.exceptions.ContainerInitializationException;
+import com.amazonaws.serverless.proxy.model.AwsProxyRequest;
+import com.amazonaws.serverless.proxy.model.AwsProxyResponse;
+import com.amazonaws.serverless.proxy.spring.SpringBootLambdaContainerHandler;
+import com.amazonaws.serverless.proxy.spring.SpringBootProxyHandlerBuilder;
+import com.amazonaws.services.lambda.runtime.Context;
+import com.amazonaws.services.lambda.runtime.RequestHandler;
+
+import java.time.Instant;
+
+
+public class SBLambdaHandler
+        implements RequestHandler<AwsProxyRequest, AwsProxyResponse>
+{
+    SpringBootLambdaContainerHandler<AwsProxyRequest, AwsProxyResponse> handler;
+    private long constructorTime;
+
+    public SBLambdaHandler() throws ContainerInitializationException {
+        long startTime = Instant.now().toEpochMilli();
+        handler = new SpringBootProxyHandlerBuilder()
+                .defaultProxy()
+                .asyncInit(startTime)
+                .springBootApplication(TestApplication.class)
+                .buildAndInitialize();
+        constructorTime = Instant.now().toEpochMilli() - startTime;
+    }
+
+    public long getConstructorTime() {
+        return constructorTime;
+    }
+
+    public AwsProxyResponse handleRequest(AwsProxyRequest awsProxyRequest, Context context)
+    {
+        return handler.proxy(awsProxyRequest, context);
+    }
+}
+

aws-serverless-java-container-spring/src/test/java/com/amazonaws/serverless/proxy/spring/springbootslowapp/TestApplication.java

@@ -0,0 +1,25 @@
+package com.amazonaws.serverless.proxy.spring.springbootslowapp;
+
+
+import org.springframework.beans.factory.InitializingBean;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.web.support.SpringBootServletInitializer;
+import org.springframework.context.annotation.ComponentScan;
+import org.springframework.context.annotation.PropertySource;
+import org.springframework.stereotype.Component;
+
+
+@SpringBootApplication
+@ComponentScan(basePackages = "com.amazonaws.serverless.proxy.spring.springbootslowapp")
+@PropertySource("classpath:boot-application.properties")
+public class TestApplication extends SpringBootServletInitializer {
+    @Component
+    public static class SlowDownInit implements InitializingBean {
+        public static final int INIT_SLEEP_TIME_MS = 13_000;
+
+        @Override
+        public void afterPropertiesSet() throws Exception {
+            Thread.sleep(INIT_SLEEP_TIME_MS);
+        }
+    }
+}

aws-serverless-java-container-spring/src/test/java/com/amazonaws/serverless/proxy/spring/springslowapp/LambdaHandler.java

@@ -0,0 +1,35 @@
+package com.amazonaws.serverless.proxy.spring.springslowapp;
+
+import com.amazonaws.serverless.exceptions.ContainerInitializationException;
+import com.amazonaws.serverless.proxy.model.AwsProxyRequest;
+import com.amazonaws.serverless.proxy.model.AwsProxyResponse;
+import com.amazonaws.serverless.proxy.spring.SpringLambdaContainerHandler;
+import com.amazonaws.serverless.proxy.spring.SpringProxyHandlerBuilder;
+import com.amazonaws.services.lambda.runtime.Context;
+import com.amazonaws.services.lambda.runtime.RequestHandler;
+
+import java.time.Instant;
+
+public class LambdaHandler implements RequestHandler<AwsProxyRequest, AwsProxyResponse> {
+    private SpringLambdaContainerHandler<AwsProxyRequest, AwsProxyResponse> handler;
+    private long constructorTime;
+
+    public LambdaHandler() throws ContainerInitializationException {
+        long startTime = Instant.now().toEpochMilli();
+        handler = new SpringProxyHandlerBuilder()
+                .defaultProxy()
+                .asyncInit(startTime)
+                .configurationClasses(SlowAppConfig.class)
+                .buildAndInitialize();
+        constructorTime = Instant.now().toEpochMilli() - startTime;
+    }
+
+    public long getConstructorTime() {
+        return constructorTime;
+    }
+
+    @Override
+    public AwsProxyResponse handleRequest(AwsProxyRequest awsProxyRequest, Context context) {
+        return handler.proxy(awsProxyRequest, context);
+    }
+}

aws-serverless-java-container-spring/src/test/java/com/amazonaws/serverless/proxy/spring/springslowapp/MessageController.java

@@ -0,0 +1,15 @@
+package com.amazonaws.serverless.proxy.spring.springslowapp;
+
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+public class MessageController {
+    public static final String HELLO_MESSAGE = "Hello";
+
+    @RequestMapping(path="/hello", method= RequestMethod.GET)
+    public String hello() {
+        return HELLO_MESSAGE;
+    }
+}

aws-serverless-java-container-spring/src/test/java/com/amazonaws/serverless/proxy/spring/springslowapp/SlowAppConfig.java

@@ -0,0 +1,20 @@
+package com.amazonaws.serverless.proxy.spring.springslowapp;
+
+import org.springframework.beans.factory.InitializingBean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Import;
+import org.springframework.stereotype.Component;
+
+@Configuration
+@Import({MessageController.class})
+public class SlowAppConfig {
+    @Component
+    public static class SlowDownInit implements InitializingBean {
+        public static final int INIT_SLEEP_TIME_MS = 13_000;
+
+        @Override
+        public void afterPropertiesSet() throws Exception {
+            Thread.sleep(INIT_SLEEP_TIME_MS);
+        }
+    }
+}