fix: update exception handling in PutLambdaEndpoint API (#18)

Author: bchampp

examples/cli.py

@@ -226,7 +226,7 @@ def generate_sam_template():
                 "Timeout": 60,
                 "MemorySize": 128,
                 "Environment": {
-                    "Variables": {"DEX_ENDPOINT": {"Ref": "LambdaEndpoint"}}
+                    "Variables": {"AWS_ENDPOINT_URL_LAMBDA": {"Ref": "LambdaEndpoint"}}
                 },
             }
         },
@@ -353,7 +353,9 @@ def deploy_function(example_name: str, function_name: str | None = None):
         "Description": example_config["description"],
         "Timeout": 60,
         "MemorySize": 128,
-        # "Environment": {"Variables": {"AWS_ENDPOINT_URL_LAMBDA": config["lambda_endpoint"]}},
+        "Environment": {
+            "Variables": {"AWS_ENDPOINT_URL_LAMBDA": config["lambda_endpoint"]}
+        },
         "DurableConfig": example_config["durableConfig"],
     }
 

src/aws_durable_execution_sdk_python_testing/web/handlers.py

@@ -790,8 +790,8 @@ def handle(self, parsed_route: Route, request: HTTPRequest) -> HTTPResponse:  #
             region_name = body.get("RegionName", "us-east-1")
 
             if not endpoint_url:
-                return HTTPResponse.create_json(
-                    400, {"error": "EndpointUrl is required"}
+                return self._handle_aws_exception(
+                    InvalidParameterValueException("EndpointUrl is required")
                 )
 
             # Update the invoker's Lambda endpoint
@@ -802,7 +802,5 @@ def handle(self, parsed_route: Route, request: HTTPRequest) -> HTTPResponse:  #
                 {"message": "Lambda endpoint updated successfully"}
             )
 
-        except (AttributeError, TypeError) as e:
-            return HTTPResponse.create_json(
-                500, {"error": f"Failed to update Lambda endpoint: {e!s}"}
-            )
+        except Exception as e:  # noqa: BLE001
+            return self._handle_framework_exception(e)

tests/web/handlers_test.py

@@ -2109,7 +2109,8 @@ def test_update_lambda_endpoint_handler_missing_endpoint_url():
     response = handler.handle(update_route, request)
 
     assert response.status_code == 400
-    assert response.body == {"error": "EndpointUrl is required"}
+    assert response.body["Type"] == "InvalidParameterValueException"
+    assert response.body["message"] == "EndpointUrl is required"
 
 
 def test_update_lambda_endpoint_handler_default_region():