From 59c233f43158d453fd9f2392a8db4d451e34a46d Mon Sep 17 00:00:00 2001 From: Jon Beilke Date: Sat, 6 Oct 2018 00:33:23 -0500 Subject: [PATCH] avoid needing to import datetime and dealing with timezone vs naive datetimes in tests --- tests/test_kms/test_kms.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/test_kms/test_kms.py b/tests/test_kms/test_kms.py index 218df1d8f..b09459e1d 100644 --- a/tests/test_kms/test_kms.py +++ b/tests/test_kms/test_kms.py @@ -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