👷

Author: jmeisele

src/handler.py

@@ -1,6 +1,6 @@
-import logging
 import json
-from typing import Dict, Any
+import logging
+from typing import Any, Dict
 
 from aws_xray_sdk.core import patch_all, xray_recorder
 

tests/test_handler.py

@@ -1,6 +1,15 @@
+import json
+
 from src.handler import lambda_handler
 
 
 def test_lambda_handler() -> None:
-    result = lambda_handler(event={}, context={})
-    assert result == "Hello from Lambda!"
+    event = {}
+    context = {}
+    result = lambda_handler(event, context)
+    expected = {
+        "statusCode": 200,
+        "headers": {"Content-Type": "application/json"},
+        "body": json.dumps("Hello from lambda"),
+    }
+    assert result == expected