Adding 2e2 tests

Author: leandrodamascena

tests/e2e/event_handler/handlers/alb_handler_with_body_none.py

@@ -0,0 +1,17 @@
+from aws_lambda_powertools.event_handler import (
+    ALBResolver,
+    Response,
+)
+
+app = ALBResolver()
+
+
+@app.get("/todos_with_no_body")
+def todos():
+    return Response(
+        status_code=200,
+    )
+
+
+def lambda_handler(event, context):
+    return app.resolve(event, context)

tests/e2e/event_handler/infrastructure.py

@@ -1,4 +1,4 @@
-from typing import Dict, Optional
+from typing import Dict, List, Optional
 
 from aws_cdk import CfnOutput
 from aws_cdk import aws_apigateway as apigwv1
@@ -17,12 +17,12 @@ class EventHandlerStack(BaseInfrastructure):
     def create_resources(self):
         functions = self.create_lambda_functions()
 
-        self._create_alb(function=functions["AlbHandler"])
+        self._create_alb(function=[functions["AlbHandler"], functions["AlbHandlerWithBodyNone"]])
         self._create_api_gateway_rest(function=functions["ApiGatewayRestHandler"])
         self._create_api_gateway_http(function=functions["ApiGatewayHttpHandler"])
         self._create_lambda_function_url(function=functions["LambdaFunctionUrlHandler"])
 
-    def _create_alb(self, function: Function):
+    def _create_alb(self, function: List[Function]):
         vpc = ec2.Vpc.from_lookup(
             self.stack,
             "VPC",
@@ -33,15 +33,19 @@ def _create_alb(self, function: Function):
         alb = elbv2.ApplicationLoadBalancer(self.stack, "ALB", vpc=vpc, internet_facing=True)
         CfnOutput(self.stack, "ALBDnsName", value=alb.load_balancer_dns_name)
 
-        self._create_alb_listener(alb=alb, name="Basic", port=80, function=function)
+        # Function with Body
+        self._create_alb_listener(alb=alb, name="Basic", port=80, function=function[0])
         self._create_alb_listener(
             alb=alb,
             name="MultiValueHeader",
             port=8080,
-            function=function,
+            function=function[0],
             attributes={"lambda.multi_value_headers.enabled": "true"},
         )
 
+        # Function without Body
+        self._create_alb_listener(alb=alb, name="BasicWithoutBody", port=8081, function=function[1])
+
     def _create_alb_listener(
         self,
         alb: elbv2.ApplicationLoadBalancer,

tests/e2e/event_handler/test_response_code.py

@@ -0,0 +1,29 @@
+import pytest
+from requests import Request
+
+from tests.e2e.utils import data_fetcher
+from tests.e2e.utils.auth import build_iam_auth
+
+
+@pytest.fixture
+def alb_basic_without_body_listener_endpoint(infrastructure: dict) -> str:
+    dns_name = infrastructure.get("ALBDnsName")
+    port = infrastructure.get("ALBBasicWithoutBodyListenerPort", "")
+    return f"http://{dns_name}:{port}"
+
+
+@pytest.mark.xdist_group(name="event_handler")
+def test_alb_with_body_empty(alb_basic_without_body_listener_endpoint):
+    # GIVEN url has a trailing slash - it should behave as if there was not one
+    url = f"{alb_basic_without_body_listener_endpoint}/todos_with_no_body"
+
+    # WHEN calling an invalid URL (with trailing slash) expect HTTPError exception from data_fetcher
+    response = data_fetcher.get_http_response(
+        Request(
+            method="GET",
+            url=url,
+            auth=build_iam_auth(url=url, aws_service="lambda"),
+        ),
+    )
+
+    assert response.status_code == 200