feat(event-handler): Use shared json Encoder

Also use more concise seperators

Author: michaelbrewer

aws_lambda_powertools/event_handler/api_gateway.py

@@ -5,6 +5,7 @@
 from enum import Enum
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
+from aws_lambda_powertools.shared.json_encoder import Encoder
 from aws_lambda_powertools.utilities.data_classes import ALBEvent, APIGatewayProxyEvent, APIGatewayProxyEventV2
 from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent
 from aws_lambda_powertools.utilities.typing import LambdaContext
@@ -104,7 +105,11 @@ def resolve(self, event, context) -> Dict[str, Any]:
         if isinstance(result, Response):
             response = result
         elif isinstance(result, dict):
-            response = Response(status_code=200, content_type="application/json", body=json.dumps(result))
+            response = Response(
+                status_code=200,
+                content_type="application/json",
+                body=json.dumps(result, separators=(",", ":"), cls=Encoder),
+            )
         else:
             response = Response(*result)
 

tests/functional/event_handler/test_api_gateway.py

@@ -1,12 +1,14 @@
 import base64
 import json
 import zlib
+from decimal import Decimal
 from pathlib import Path
 from typing import Dict, Tuple
 
 import pytest
 
 from aws_lambda_powertools.event_handler.api_gateway import ApiGatewayResolver, ProxyEventType, Response
+from aws_lambda_powertools.shared.json_encoder import Encoder
 from aws_lambda_powertools.utilities.data_classes import ALBEvent, APIGatewayProxyEvent, APIGatewayProxyEventV2
 
 
@@ -293,18 +295,20 @@ def handler(event, context):
 def test_rest_api():
     # GIVEN a function that returns a Dict
     app = ApiGatewayResolver(proxy_type=ProxyEventType.http_api_v1)
+    expected_dict = {"foo": "value", "second": Decimal("100.01")}
 
     @app.get("/my/path")
     def rest_func() -> Dict:
-        return {"foo": "value"}
+        return expected_dict
 
     # WHEN calling the event handler
     result = app(LOAD_GW_EVENT, {})
 
     # THEN automatically process this as a json rest api response
     assert result["statusCode"] == 200
     assert result["headers"]["Content-Type"] == APPLICATION_JSON
-    assert result["body"] == json.dumps({"foo": "value"})
+    expected_str = json.dumps(expected_dict, separators=(",", ":"), indent=None, cls=Encoder)
+    assert result["body"] == expected_str
 
 
 def test_handling_response_type():