Fixing UT for key_type on upload schema

2020-09-28 19:03:21 -07:00
parent 7a226241db
commit aaff0f7581
2 changed files with 8 additions and 1 deletions
--- a/lemur/certificates/schemas.py
+++ b/lemur/certificates/schemas.py
@ -376,7 +376,13 @@ class CertificateUploadInputSchema(CertificateCreationSchema):

    @pre_load
    def load_data(self, data):
-        data["key_type"] = utils.get_key_type_from_certificate(data["body"])
+        if data.get("body"):
+            try:
+                data["key_type"] = utils.get_key_type_from_certificate(data["body"])
+            except ValueError:
+                raise ValidationError(
+                    "Public certificate presented is not valid.", field_names=["body"]
+                )


 class CertificateExportInputSchema(LemurInputSchema):