include pending days input for schedule key deletion and update tests since boto client returns DeletionDate as datetime

This commit is contained in:
Jon Beilke 2018-10-06 00:13:47 -05:00
commit 21c8914efe
2 changed files with 4 additions and 4 deletions

View file

@ -269,11 +269,12 @@ class KmsResponse(BaseResponse):
def schedule_key_deletion(self):
key_id = self.parameters.get('KeyId')
pending_window_in_days = self.parameters.get('PendingWindowInDays')
_assert_valid_key_id(self.kms_backend.get_key_id(key_id))
try:
return json.dumps({
'KeyId': key_id,
'DeletionDate': iso_8601_datetime_without_milliseconds(self.kms_backend.schedule_key_deletion(key_id))
'DeletionDate': iso_8601_datetime_without_milliseconds(self.kms_backend.schedule_key_deletion(key_id, pending_window_in_days))
})
except KeyError:
raise JSONResponseError(404, 'Not Found', body={