test: lambda error responses

Author: monroeclinton

tests/unit/local/lambda_service/test_lambda_error_responses.py

@@ -97,6 +97,19 @@ def test_generic_method_not_allowed(self, service_response_mock):
             405,
         )
 
+    @patch("samcli.local.services.base_local_service.BaseLocalService.service_response")
+    def test_validation_exception(self, service_response_mock):
+        service_response_mock.return_value = "ValidationException"
+
+        response = LambdaErrorResponses.validation_exception("ValidationException")
+
+        self.assertEqual(response, "ValidationException")
+        service_response_mock.assert_called_once_with(
+            '{"Type": "User", "Message": "ValidationException"}',
+            {"x-amzn-errortype": "ValidationException", "Content-Type": "application/json"},
+            400,
+        )
+
     @patch("samcli.local.services.base_local_service.BaseLocalService.service_response")
     def test_durable_execution_not_found(self, service_response_mock):
         service_response_mock.return_value = "DurableExecutionNotFound"
@@ -109,3 +122,16 @@ def test_durable_execution_not_found(self, service_response_mock):
             {"x-amzn-errortype": "ResourceNotFound", "Content-Type": "application/json"},
             404,
         )
+
+    @patch("samcli.local.services.base_local_service.BaseLocalService.service_response")
+    def test_container_creation_failed(self, service_response_mock):
+        service_response_mock.return_value = "ContainerCreationFailed"
+
+        response = LambdaErrorResponses.container_creation_failed("Container creation failed: test message")
+
+        self.assertEqual(response, "ContainerCreationFailed")
+        service_response_mock.assert_called_once_with(
+            '{"Type": "LocalService", "Message": "Container creation failed: test message"}',
+            {"x-amzn-errortype": "ContainerCreationFailed", "Content-Type": "application/json"},
+            501,
+        )