avoid needing to import datetime and dealing with timezone vs naive datetimes in tests

This commit is contained in:
Jon Beilke 2018-10-06 00:33:23 -05:00
parent 21c8914efe
commit 59c233f431

View File

@ -9,7 +9,6 @@ import sure # noqa
from moto import mock_kms, mock_kms_deprecated
from nose.tools import assert_raises
from freezegun import freeze_time
from datetime import datetime, timedelta
@mock_kms_deprecated
@ -659,7 +658,7 @@ def test_schedule_key_deletion():
KeyId=key['KeyMetadata']['KeyId']
)
assert response['KeyId'] == key['KeyMetadata']['KeyId']
assert response['DeletionDate'] == datetime.now() + timedelta(days=30)
assert response['DeletionDate'] == '2015-01-31T12:00:00.000Z'
result = client.describe_key(KeyId=key['KeyMetadata']['KeyId'])
assert result["KeyMetadata"]["Enabled"] == False
@ -677,7 +676,7 @@ def test_schedule_key_deletion_custom():
PendingWindowInDays=7
)
assert response['KeyId'] == key['KeyMetadata']['KeyId']
assert response['DeletionDate'] == datetime.now() + timedelta(days=7)
assert response['DeletionDate'] == '2015-01-08T12:00:00.000Z'
result = client.describe_key(KeyId=key['KeyMetadata']['KeyId'])
assert result["KeyMetadata"]["Enabled"] == False