fix: add crash test

Author: rubenfonseca

aws_lambda_powertools/event_handler/api_gateway.py

@@ -241,6 +241,7 @@ def __init__(
         self.headers: Dict[str, Union[str, List[str]]] = headers if headers else {}
         self.cookies = cookies or []
         self.compress = compress
+        self.content_type = content_type
         if content_type:
             self.headers.setdefault("Content-Type", content_type)
 

aws_lambda_powertools/event_handler/bedrock_agent.py

@@ -30,7 +30,7 @@ def build(self, event: BedrockAgentEvent, *args) -> Dict[str, Any]:
                 "httpMethod": event.http_method,
                 "httpStatusCode": self.response.status_code,
                 "responseBody": {
-                    "application/json": {
+                    self.response.content_type: {
                         "body": self.response.body,
                     },
                 },

tests/functional/event_handler/test_bedrock_agent.py

@@ -77,3 +77,34 @@ def claims():
     assert result["response"]["actionGroup"] == "ClaimManagementActionGroup"
     assert result["response"]["httpMethod"] == "GET"
     assert result["response"]["httpStatusCode"] == 404
+
+
+def test_bedrock_agent_event_with_exception():
+    # GIVEN a Bedrock Agent event
+    app = BedrockAgentResolver()
+
+    @app.exception_handler(RuntimeError)
+    def handle_runtime_error(ex: RuntimeError):
+        return Response(
+            status_code=500,
+            content_type=content_types.TEXT_PLAIN,
+            body="Something went wrong",
+        )
+
+    @app.get("/claims")
+    def claims():
+        raise RuntimeError()
+
+    # WHEN calling the event handler
+    result = app(load_event("bedrockAgentEvent.json"), {})
+
+    # THEN process the exception correctly
+    # AND return 500 because of the internal server error
+    assert result["messageVersion"] == "1.0"
+    assert result["response"]["apiPath"] == "/claims"
+    assert result["response"]["actionGroup"] == "ClaimManagementActionGroup"
+    assert result["response"]["httpMethod"] == "GET"
+    assert result["response"]["httpStatusCode"] == 500
+
+    body = result["response"]["responseBody"]["text/plain"]["body"]
+    assert body == "Something went wrong"