Adding e2e tests

Author: leandrodamascena

tests/e2e/event_handler/handlers/data_validation_and_middleware.py

@@ -0,0 +1,27 @@
+from pydantic import BaseModel
+
+from aws_lambda_powertools.event_handler import APIGatewayRestResolver, Response
+from aws_lambda_powertools.event_handler.middlewares import NextMiddleware
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+
+def middleware_auth(app: APIGatewayRestResolver, next_middleware: NextMiddleware):
+    # Return early response
+    return Response(status_code=202, content_type="application/json", body="{}")
+
+
+app = APIGatewayRestResolver(enable_validation=True)
+app.use(middlewares=[middleware_auth])
+
+
+class MyModel(BaseModel):
+    name: str
+
+
+@app.get("/data_validation_middleware")
+def get_data_validation_and_middleware() -> MyModel:
+    return MyModel(name="powertools")
+
+
+def lambda_handler(event, context: LambdaContext):
+    return app.resolve(event, context)

tests/e2e/event_handler/infrastructure.py

@@ -23,6 +23,7 @@ def create_resources(self):
                 functions["ApiGatewayRestHandler"],
                 functions["OpenapiHandler"],
                 functions["OpenapiHandlerWithPep563"],
+                functions["DataValidationAndMiddleware"],
             ],
         )
         self._create_api_gateway_http(function=functions["ApiGatewayHttpHandler"])
@@ -101,6 +102,9 @@ def _create_api_gateway_rest(self, function: list[Function]):
         openapi_schema = apigw.root.add_resource("openapi_schema_with_pep563")
         openapi_schema.add_method("GET", apigwv1.LambdaIntegration(function[2], proxy=True))
 
+        openapi_schema = apigw.root.add_resource("data_validation_middleware")
+        openapi_schema.add_method("GET", apigwv1.LambdaIntegration(function[3], proxy=True))
+
         CfnOutput(self.stack, "APIGatewayRestUrl", value=apigw.url)
 
     def _create_lambda_function_url(self, function: Function):

tests/e2e/event_handler/test_openapi.py

@@ -44,3 +44,18 @@ def test_get_openapi_schema_with_pep563(apigw_rest_endpoint):
     assert "Powertools e2e API" in response.text
     assert "x-amazon-apigateway-gateway-responses" in response.text
     assert response.status_code == 200
+
+
+def test_get_openapi_validation_and_middleware(apigw_rest_endpoint):
+    # GIVEN
+    url = f"{apigw_rest_endpoint}data_validation_middleware"
+
+    # WHEN
+    response = data_fetcher.get_http_response(
+        Request(
+            method="GET",
+            url=url,
+        ),
+    )
+
+    assert response.status_code == 202