Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add kms boto3 test.

Browse Source
This commit is contained in:
Steve Pulec 2017-03-14 23:42:47 -04:00
parent 0df47574ab
commit 09a4d177f5
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
moto/kms/responses.py
View File

@ -231,7 +231,6 @@ class KmsResponse(BaseResponse):
def decrypt(self): def decrypt(self):
value = self.parameters.get("CiphertextBlob") value = self.parameters.get("CiphertextBlob")
print("value 3", value)
return json.dumps({"Plaintext": base64.b64decode(value).decode("utf-8")}) return json.dumps({"Plaintext": base64.b64decode(value).decode("utf-8")})

12
tests/test_kms/test_kms.py
View File

@ -1,11 +1,12 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import re import re
import boto3
import boto.kms import boto.kms
from boto.exception import JSONResponseError from boto.exception import JSONResponseError
from boto.kms.exceptions import AlreadyExistsException, NotFoundException from boto.kms.exceptions import AlreadyExistsException, NotFoundException
import sure # noqa import sure # noqa
from moto import mock_kms_deprecated from moto import mock_kms, mock_kms_deprecated
from nose.tools import assert_raises from nose.tools import assert_raises
@ -600,3 +601,12 @@ def test__assert_default_policy():
"not-default").should.throw(JSONResponseError) "not-default").should.throw(JSONResponseError)
_assert_default_policy.when.called_with( _assert_default_policy.when.called_with(
"default").should_not.throw(JSONResponseError) "default").should_not.throw(JSONResponseError)
@mock_kms
def test_kms_encrypt_boto3():
client = boto3.client('kms', region_name='us-east-1')
response = client.encrypt(KeyId='foo', Plaintext=b'bar')
response = client.decrypt(CiphertextBlob=response['CiphertextBlob'])
response['Plaintext'].should.equal(b'bar')