diff --git a/services/kms/src/stackit/kms/models/key.py b/services/kms/src/stackit/kms/models/key.py index 105c02a63..c2f413177 100644 --- a/services/kms/src/stackit/kms/models/key.py +++ b/services/kms/src/stackit/kms/models/key.py @@ -80,8 +80,10 @@ class Key(BaseModel): @field_validator("state") def state_validate_enum(cls, value): """Validates the enum""" - if value not in set(["active", "version_not_ready", "deleted"]): - raise ValueError("must be one of enum values ('active', 'version_not_ready', 'deleted')") + if value not in set(["active", "deleted", "not_available", "errors_exist", "no_version"]): + raise ValueError( + "must be one of enum values ('active', 'deleted', 'not_available', 'errors_exist', 'no_version')" + ) return value model_config = ConfigDict( diff --git a/services/kms/src/stackit/kms/models/version.py b/services/kms/src/stackit/kms/models/version.py index 57e75eec3..b3b630906 100644 --- a/services/kms/src/stackit/kms/models/version.py +++ b/services/kms/src/stackit/kms/models/version.py @@ -69,9 +69,18 @@ class Version(BaseModel): @field_validator("state") def state_validate_enum(cls, value): """Validates the enum""" - if value not in set(["active", "key_material_not_ready", "key_material_invalid", "disabled", "destroyed"]): + if value not in set( + [ + "active", + "key_material_not_ready", + "key_material_invalid", + "key_material_unavailable", + "disabled", + "destroyed", + ] + ): raise ValueError( - "must be one of enum values ('active', 'key_material_not_ready', 'key_material_invalid', 'disabled', 'destroyed')" + "must be one of enum values ('active', 'key_material_not_ready', 'key_material_invalid', 'key_material_unavailable', 'disabled', 'destroyed')" ) return value diff --git a/services/kms/src/stackit/kms/models/wrapping_key.py b/services/kms/src/stackit/kms/models/wrapping_key.py index 2da701142..960c8fc42 100644 --- a/services/kms/src/stackit/kms/models/wrapping_key.py +++ b/services/kms/src/stackit/kms/models/wrapping_key.py @@ -69,8 +69,10 @@ class WrappingKey(BaseModel): @field_validator("state") def state_validate_enum(cls, value): """Validates the enum""" - if value not in set(["active", "key_material_not_ready", "expired", "deleting"]): - raise ValueError("must be one of enum values ('active', 'key_material_not_ready', 'expired', 'deleting')") + if value not in set(["active", "key_material_not_ready", "expired", "deleting", "key_material_unavailable"]): + raise ValueError( + "must be one of enum values ('active', 'key_material_not_ready', 'expired', 'deleting', 'key_material_unavailable')" + ) return value model_config = ConfigDict(