compat for older freezegun version

2 files changed, 5 insertions, 3 deletions

diff --git a/tests/test_jws.py b/tests/test_jws.py
index 9938311..4f0da50 100644
--- a/tests/test_jws.py
+++ b/tests/test_jws.py
@@ -1,3 +1,4 @@
+from datetime import timedelta
 from functools import partial
 
 import pytest
@@ -79,7 +80,7 @@ class TestTimedJWSSerializer(TestJWSSerializer, TestTimedSerializer):
         signed = serializer.dumps(value)
         freeze.tick()
         assert serializer.loads(signed) == value
-        freeze.tick(10)
+        freeze.tick(timedelta(seconds=10))
 
         with pytest.raises(SignatureExpired) as exc_info:
             serializer.loads(signed)
diff --git a/tests/test_timed.py b/tests/test_timed.py
index 36c1a86..805ff42 100644
--- a/tests/test_timed.py
+++ b/tests/test_timed.py
@@ -1,4 +1,5 @@
 from datetime import datetime
+from datetime import timedelta
 from functools import partial
 
 import pytest
@@ -33,7 +34,7 @@ class TestTimestampSigner(FreezeMixin, TestSigner):
         signed = signer.sign("value")
         freeze.tick()
         assert signer.unsign(signed, max_age=10) == b"value"
-        freeze.tick(10)
+        freeze.tick(timedelta(seconds=10))
 
         with pytest.raises(SignatureExpired) as exc_info:
             signer.unsign(signed, max_age=10)
@@ -72,7 +73,7 @@ class TestTimedSerializer(FreezeMixin, TestSerializer):
         signed = serializer.dumps(value)
         freeze.tick()
         assert serializer.loads(signed, max_age=10) == value
-        freeze.tick(10)
+        freeze.tick(timedelta(seconds=10))
 
         with pytest.raises(SignatureExpired) as exc_info:
             serializer.loads(signed, max_age=10)