catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

add kms:GenerateRandom and tests

Browse source
This commit is contained in:
mattsb42-aws 2019-08-27 20:55:58 -07:00
commit f7043e1eaf
2 changed files with 26 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
moto/kms/responses.py
View file

@ -2,7 +2,9 @@ from __future__ import unicode_literals
import base64
import json
import os
import re
import six
from moto.core.responses import BaseResponse
@ -384,6 +386,15 @@ class KmsResponse(BaseResponse):
return json.dumps(result)
def generate_random(self):
number_of_bytes = self.parameters.get("NumberOfBytes")
entropy = os.urandom(number_of_bytes)
response_entropy = base64.b64encode(entropy).decode("utf-8")
return json.dumps({"Plaintext": response_entropy})
def _assert_valid_key_id(key_id):
if not re.match(r'^[A-F0-9]{8}-[A-F0-9]{4}-[A-F0-9]{4}-[A-F0-9]{4}-[A-F0-9]{12}$', key_id, re.IGNORECASE):