test logging for APIM status endpoint

Author: nhsd-rebecca-flynn

pathology-api/lambda_handler.py

@@ -112,34 +112,43 @@ def status() -> Response[str]:
 
 @app.post("/FHIR/R4/Bundle")
 def post_result() -> Response[str]:
-    correlation_id = app.current_event.headers.get(_CORRELATION_ID_HEADER)
+    _logger.debug("Post result endpoint called.")
 
-    if not correlation_id:
-        raise ValueError(f"Missing required header: {_CORRELATION_ID_HEADER}")
-    with set_correlation_id(correlation_id):
-        _logger.debug("Post result endpoint called.")
-
-        try:
-            payload = app.current_event.json_body
-        except JSONDecodeError as e:
-            raise ValidationError("Invalid payload provided.") from e
+    try:
+        payload = app.current_event.json_body
+    except JSONDecodeError as e:
+        raise ValidationError("Invalid payload provided.") from e
 
-        _logger.debug("Payload received: %s", payload)
+    _logger.debug("Payload received: %s", payload)
 
-        if payload is None:
-            raise ValidationError(
-                "Resources must be provided as a bundle of type 'document'"
-            )
+    if payload is None:
+        raise ValidationError(
+            "Resources must be provided as a bundle of type 'document'"
+        )
 
-        bundle = Bundle.model_validate(payload, by_alias=True)
+    bundle = Bundle.model_validate(payload, by_alias=True)
 
-        response = handle_request(bundle)
+    response = handle_request(bundle)
 
-        return _with_default_headers(
-            status_code=200,
-            body=response,
-        )
+    return _with_default_headers(
+        status_code=200,
+        body=response,
+    )
 
 
 def handler(data: dict[str, Any], context: LambdaContext) -> dict[str, Any]:
-    return app.resolve(data, context)
+    headers = data.get("headers", {}) or {}
+    correlation_id = headers.get(_CORRELATION_ID_HEADER)
+
+    if not correlation_id:
+        return {
+            "statusCode": 500,
+            "headers": {"Content-Type": "application/fhir+json"},
+            "body": OperationOutcome.create_server_error(
+                f"Missing required header: {_CORRELATION_ID_HEADER}"
+            ).model_dump_json(by_alias=True, exclude_none=True),
+        }
+    with set_correlation_id(correlation_id):
+        response = app.resolve(data, context)
+        response.setdefault("headers", {})[_CORRELATION_ID_HEADER] = correlation_id
+        return response