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

add freezegun and test DeletionDate for chedule_key_deletion

Browse Source
This commit is contained in:
Jon Beilke 2018-10-05 16:21:16 -05:00
parent 15c24e49f0
commit 7e96203020
2 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
moto/kms/models.py
View File

@ -4,6 +4,7 @@ import boto.kms
from moto.core import BaseBackend, BaseModel from moto.core import BaseBackend, BaseModel
from .utils import generate_key_id from .utils import generate_key_id
from collections import defaultdict from collections import defaultdict
from datetime import datetime, timedelta
class Key(BaseModel): class Key(BaseModel):

26
tests/test_kms/test_kms.py
View File

@ -8,6 +8,8 @@ from boto.kms.exceptions import AlreadyExistsException, NotFoundException
import sure # noqa import sure # noqa
from moto import mock_kms, mock_kms_deprecated from moto import mock_kms, mock_kms_deprecated
from nose.tools import assert_raises from nose.tools import assert_raises
from freezegun import freeze_time
from datetime import datetime, timedelta
@mock_kms_deprecated @mock_kms_deprecated
@ -652,11 +654,12 @@ def test_enable_key():
def test_schedule_key_deletion(): def test_schedule_key_deletion():
client = boto3.client('kms', region_name='us-east-1') client = boto3.client('kms', region_name='us-east-1')
key = client.create_key(description='schedule-key-deletion') key = client.create_key(description='schedule-key-deletion')
response = client.schedule_key_deletion( with freeze_time("2015-01-01 12:00:00"):
KeyId=key['KeyMetadata']['KeyId'] response = client.schedule_key_deletion(
) KeyId=key['KeyMetadata']['KeyId']
assert response['KeyId'] == 'schedule-key-deletion' )
assert response['DeletionDate'] == datetime.now() + timedelta(days=30) assert response['KeyId'] == 'schedule-key-deletion'
assert response['DeletionDate'] == datetime.now() + timedelta(days=30)
result = client.describe_key(KeyId='schedule-key-deletion') result = client.describe_key(KeyId='schedule-key-deletion')
assert result["KeyMetadata"]["Enabled"] == False assert result["KeyMetadata"]["Enabled"] == False
@ -668,12 +671,13 @@ def test_schedule_key_deletion():
def test_schedule_key_deletion_custom(): def test_schedule_key_deletion_custom():
client = boto3.client('kms', region_name='us-east-1') client = boto3.client('kms', region_name='us-east-1')
key = client.create_key(description='schedule-key-deletion') key = client.create_key(description='schedule-key-deletion')
response = client.schedule_key_deletion( with freeze_time("2015-01-01 12:00:00"):
KeyId=key['KeyMetadata']['KeyId'], response = client.schedule_key_deletion(
PendingWindowInDays=7 KeyId=key['KeyMetadata']['KeyId'],
) PendingWindowInDays=7
assert response['KeyId'] == 'schedule-key-deletion' )
assert response['DeletionDate'] == datetime.now() + timedelta(days=7) assert response['KeyId'] == 'schedule-key-deletion'
assert response['DeletionDate'] == datetime.now() + timedelta(days=7)
result = client.describe_key(KeyId='schedule-key-deletion') result = client.describe_key(KeyId='schedule-key-deletion')
assert result["KeyMetadata"]["Enabled"] == False assert result["KeyMetadata"]["Enabled"] == False