Added unit test for Jersey principal injection.

Author: sapessi

aws-serverless-java-container-core/src/main/java/com/amazonaws/serverless/proxy/internal/testutils/AwsProxyRequestBuilder.java

@@ -168,6 +168,10 @@ public AwsProxyRequestBuilder authorizerPrincipal(String principal) {
             this.request.getRequestContext().setAuthorizer(new ApiGatewayAuthorizerContext());
         }
         this.request.getRequestContext().getAuthorizer().setPrincipalId(principal);
+        if (this.request.getRequestContext().getAuthorizer().getClaims() == null) {
+            this.request.getRequestContext().getAuthorizer().setClaims(new CognitoAuthorizerClaims());
+        }
+        this.request.getRequestContext().getAuthorizer().getClaims().setSubject(principal);
         return this;
     }
 

aws-serverless-java-container-jersey/src/test/java/com/amazonaws/serverless/proxy/jersey/EchoJerseyResource.java

@@ -25,6 +25,7 @@
 import javax.ws.rs.core.Context;
 import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.Response;
+import javax.ws.rs.core.SecurityContext;
 import javax.ws.rs.core.UriInfo;
 import java.util.Enumeration;
 import java.util.Random;
@@ -37,6 +38,9 @@ public class EchoJerseyResource {
     public static final String SERVLET_RESP_HEADER_KEY = "X-HttpServletResponse";
     public static final String EXCEPTION_MESSAGE = "Fake exception";
 
+    @Context
+    SecurityContext securityCtx;
+
     @Path("/headers") @GET
     @Produces(MediaType.APPLICATION_JSON)
     public MapResponseModel echoHeaders(@Context ContainerRequestContext context) {
@@ -48,6 +52,16 @@ public MapResponseModel echoHeaders(@Context ContainerRequestContext context) {
         return headers;
     }
 
+    @Path("/security-context") @GET
+    @Produces(MediaType.APPLICATION_JSON)
+    public SingleValueModel getPrincipal() {
+        SingleValueModel output = new SingleValueModel();
+        if (securityCtx != null) {
+            output.setValue(securityCtx.getUserPrincipal().getName());
+        }
+        return output;
+    }
+
     @Path("/servlet-headers") @GET
     @Produces(MediaType.APPLICATION_JSON)
     public MapResponseModel echoServletHeaders(@Context HttpServletRequest context) {

aws-serverless-java-container-jersey/src/test/java/com/amazonaws/serverless/proxy/jersey/JerseyAwsProxyTest.java

@@ -54,6 +54,7 @@ public class JerseyAwsProxyTest {
     private static final String QUERY_STRING_KEY = "identifier";
     private static final String QUERY_STRING_NON_ENCODED_VALUE = "Space Test";
     private static final String QUERY_STRING_ENCODED_VALUE = "Space%20Test";
+    private static final String USER_PRINCIPAL = "user1";
 
 
     private static ObjectMapper objectMapper = new ObjectMapper();
@@ -302,6 +303,16 @@ public void stripBasePath_route_shouldReturn404() {
         handler.stripBasePath("");
     }
 
+    @Test
+    public void securityContext_injectPrincipal_expectPrincipalName() {
+        AwsProxyRequest request = new AwsProxyRequestBuilder("/echo/security-context", "GET")
+                                          .authorizerPrincipal(USER_PRINCIPAL).build();
+
+        AwsProxyResponse resp = handler.proxy(request, lambdaContext);
+        assertEquals(200, resp.getStatusCode());
+        validateSingleValueModel(resp, USER_PRINCIPAL);
+    }
+
     private void validateMapResponseModel(AwsProxyResponse output) {
         validateMapResponseModel(output, CUSTOM_HEADER_KEY, CUSTOM_HEADER_VALUE);
     }