modmex-lambda 0.5.0__tar.gz → 0.5.1__tar.gz

{modmex_lambda-0.5.0 → modmex_lambda-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modmex-lambda
-Version: 0.5.0
+Version: 0.5.1
 Summary: Ultra-lightweight AWS Lambda utilities for API Gateway routing and event handling.
 Author: Modmex
 License: MIT

{modmex_lambda-0.5.0 → modmex_lambda-0.5.1}/modmex_lambda/event_handler/gateway_response.py

@@ -20,8 +20,13 @@ class GatewayResponseBuilder:
         self.json_serializer = json_serializer
 
     def serialize(self, event: BaseProxyEvent, cors: CORSConfig | None = None) -> dict[str, Any]:
-        if self.response.is_json() and not isinstance(self.response.body, (str, bytes)):
-            self.response.body = self.json_serializer(self.response.body)
+        if self.response.body is None:
+            self.response.body = ""
+        elif self.response.is_json() and not isinstance(self.response.body, (str, bytes)):
+            if hasattr(self.response.body, "model_dump_json"):
+                self.response.body = self.response.body.model_dump_json()
+            else:
+                self.response.body = self.json_serializer(self.response.body)
 
         self._handle_route_configuration(event, cors)
 

{modmex_lambda-0.5.0 → modmex_lambda-0.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "modmex-lambda"
-version = "0.5.0"
+version = "0.5.1"
 description = "Ultra-lightweight AWS Lambda utilities for API Gateway routing and event handling."
 readme = "README.md"
 requires-python = ">=3.10,<4.0"

{modmex_lambda-0.5.0 → modmex_lambda-0.5.1}/tests/event_handler/test_api_gateway.py

@@ -58,6 +58,11 @@ class AuditService:
         return {"user_id": user_id, "tenant_id": self.tenant_id}
 
 
+class DumpJsonModel:
+    def model_dump_json(self) -> str:
+        return '{"ok":true}'
+
+
 def test_public_api_requires_explicit_api_gateway_resolver() -> None:
     assert modmex_lambda.APIGatewayHttpResolver is APIGatewayHttpResolver
     assert modmex_lambda.APIGatewayRestResolver is APIGatewayRestResolver
@@ -209,6 +214,34 @@ def test_http_resolver_serializes_cookies_in_http_api_shape() -> None:
     assert response_body(response) == {"ok": True}
 
 
+def test_http_resolver_serializes_model_dump_json_body_without_double_encoding() -> None:
+    app = APIGatewayHttpResolver()
+
+    @app.get("/model")
+    def model():
+        return DumpJsonModel()
+
+    response = app.resolve(http_v2_event("GET", "/model"), object())
+
+    assert response["headers"]["Content-Type"] == "application/json"
+    assert response["body"] == '{"ok":true}'
+    assert response_body(response) == {"ok": True}
+
+
+def test_http_resolver_serializes_no_content_response_with_empty_body() -> None:
+    app = APIGatewayHttpResolver()
+
+    @app.delete("/users/<user_id>", status_code=204)
+    def delete_user(user_id: str):
+        return None
+
+    response = app.resolve(http_v2_event("DELETE", "/users/42"), object())
+
+    assert response["statusCode"] == 204
+    assert response["body"] == ""
+    assert response_body(response) is None
+
+
 def test_static_dynamic_any_404_and_405_routing() -> None:
     app = APIGatewayHttpResolver()
 

{modmex_lambda-0.5.0 → modmex_lambda-0.5.1}/tests/event_handler/test_gateway_response.py

@@ -44,3 +44,28 @@ def test_gateway_response_builder_uses_first_origin_and_response_compression_ove
     assert payload["isBase64Encoded"] is False
     assert payload["multiValueHeaders"]["Access-Control-Allow-Origin"] == ["*"]
     assert "Content-Encoding" not in payload["multiValueHeaders"]
+
+
+def test_gateway_response_builder_serializes_none_body_as_empty_string() -> None:
+    event = APIGatewayProxyEvent(
+        {
+            "path": "/users/42",
+            "httpMethod": "DELETE",
+            "headers": {},
+            "multiValueHeaders": {},
+            "requestContext": {},
+        },
+    )
+    response = Response(
+        status_code=204,
+        body=None,
+        content_type=content_types.APPLICATION_JSON,
+    )
+
+    payload = GatewayResponseBuilder(
+        response=response,
+        route=None,
+        json_serializer=lambda value: "null",
+    ).serialize(event)
+
+    assert payload["body"] == ""