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

cant manipulate time in server mode tests

Browse Source
This commit is contained in:
Jon Beilke 2018-10-08 08:29:21 -05:00
parent 398dcd8230
commit c2595b2eef
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_kms/test_kms.py
View File

@ -655,12 +655,19 @@ 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')
if os.environ.get('TEST_SERVER_MODE', 'false').lower() == 'false':
with freeze_time("2015-01-01 12:00:00"): with freeze_time("2015-01-01 12:00:00"):
response = client.schedule_key_deletion( response = client.schedule_key_deletion(
KeyId=key['KeyMetadata']['KeyId'] KeyId=key['KeyMetadata']['KeyId']
) )
assert response['KeyId'] == key['KeyMetadata']['KeyId'] assert response['KeyId'] == key['KeyMetadata']['KeyId']
assert response['DeletionDate'] == datetime(2015, 1, 31, 12, 0, tzinfo=tzlocal()) assert response['DeletionDate'] == datetime(2015, 1, 31, 12, 0, tzinfo=tzlocal())
else:
# Can't manipulate time in server mode
response = client.schedule_key_deletion(
KeyId=key['KeyMetadata']['KeyId']
)
assert response['KeyId'] == key['KeyMetadata']['KeyId']
result = client.describe_key(KeyId=key['KeyMetadata']['KeyId']) result = client.describe_key(KeyId=key['KeyMetadata']['KeyId'])
assert result["KeyMetadata"]["Enabled"] == False assert result["KeyMetadata"]["Enabled"] == False
@ -672,6 +679,7 @@ 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')
if os.environ.get('TEST_SERVER_MODE', 'false').lower() == 'false':
with freeze_time("2015-01-01 12:00:00"): with freeze_time("2015-01-01 12:00:00"):
response = client.schedule_key_deletion( response = client.schedule_key_deletion(
KeyId=key['KeyMetadata']['KeyId'], KeyId=key['KeyMetadata']['KeyId'],
@ -679,6 +687,13 @@ def test_schedule_key_deletion_custom():
) )
assert response['KeyId'] == key['KeyMetadata']['KeyId'] assert response['KeyId'] == key['KeyMetadata']['KeyId']
assert response['DeletionDate'] == datetime(2015, 1, 8, 12, 0, tzinfo=tzlocal()) assert response['DeletionDate'] == datetime(2015, 1, 8, 12, 0, tzinfo=tzlocal())
else:
# Can't manipulate time in server mode
response = client.schedule_key_deletion(
KeyId=key['KeyMetadata']['KeyId'],
PendingWindowInDays=7
)
assert response['KeyId'] == key['KeyMetadata']['KeyId']
result = client.describe_key(KeyId=key['KeyMetadata']['KeyId']) result = client.describe_key(KeyId=key['KeyMetadata']['KeyId'])
assert result["KeyMetadata"]["Enabled"] == False assert result["KeyMetadata"]["Enabled"] == False