11import json
22import secrets
3- from datetime import datetime , timedelta , timezone
3+ from datetime import datetime , timedelta
4+ from datetime .timezone import utc
45
56import pytest
67from pydantic import ValidationError
@@ -26,7 +27,7 @@ def test_token_sign_verify_roundtrip(secret_key):
2627 token1 = signing .AuthToken (
2728 url = "https://example.com/url" ,
2829 user = "user" ,
29- expiry = datetime .now (timezone . utc ) + timedelta (minutes = 1 ),
30+ expiry = datetime .now (utc ) + timedelta (minutes = 1 ),
3031 )
3132 token_string = token1 .sign (secret_key , "salt" )
3233
@@ -39,7 +40,7 @@ def test_token_object_url_invalid():
3940 signing .AuthToken (
4041 url = "bad" ,
4142 user = "user" ,
42- expiry = datetime .now (timezone . utc ) + timedelta (minutes = 1 ),
43+ expiry = datetime .now (utc ) + timedelta (minutes = 1 ),
4344 )
4445
4546
@@ -48,15 +49,15 @@ def test_token_object_expired():
4849 signing .AuthToken (
4950 url = "https://example.com/url" ,
5051 user = "user" ,
51- expiry = datetime .now (timezone . utc ) - timedelta (minutes = 1 ),
52+ expiry = datetime .now (utc ) - timedelta (minutes = 1 ),
5253 )
5354
5455
5556def test_token_verify_mismatched_secrets ():
5657 payload = dict (
5758 url = "https://example.com/url" ,
5859 user = "user" ,
59- expiry = datetime .now (timezone . utc ) + timedelta (minutes = 1 ),
60+ expiry = datetime .now (utc ) + timedelta (minutes = 1 ),
6061 )
6162 token = create_raw_token (payload , "secret1" * 10 )
6263
@@ -76,7 +77,7 @@ def test_token_verify_expired(secret_key):
7677 payload = dict (
7778 url = "https://example.com/url" ,
7879 user = "user" ,
79- expiry = datetime .now (timezone . utc ) - timedelta (minutes = 1 ),
80+ expiry = datetime .now (utc ) - timedelta (minutes = 1 ),
8081 )
8182 token = create_raw_token (payload , secret_key )
8283 with pytest .raises (signing .AuthToken .Expired ):
@@ -87,7 +88,7 @@ def test_token_verify_wrong_all_the_things(secret_key):
8788 payload = dict (
8889 url = "bad url" ,
8990 # missing user
90- expiry = datetime .now (timezone . utc ) - timedelta (minutes = 1 ),
91+ expiry = datetime .now (utc ) - timedelta (minutes = 1 ),
9192 )
9293 token = create_raw_token (payload , secret_key )
9394 with pytest .raises (ValidationError ) as exc_info :
0 commit comments