diff --git a/test/sasl/test_msk.py b/test/sasl/test_msk.py index 297ca84ce..e9f1325f3 100644 --- a/test/sasl/test_msk.py +++ b/test/sasl/test_msk.py @@ -1,5 +1,6 @@ import datetime import json +import sys from kafka.sasl.msk import AwsMskIamClient @@ -10,7 +11,10 @@ def client_factory(token=None): - now = datetime.datetime.utcfromtimestamp(1629321911) + if sys.version_info >= (3, 3): + now = datetime.datetime.fromtimestamp(1629321911, datetime.timezone.utc) + else: + now = datetime.datetime.utcfromtimestamp(1629321911) with mock.patch('kafka.sasl.msk.datetime') as mock_dt: mock_dt.datetime.utcnow = mock.Mock(return_value=now) return AwsMskIamClient(