1- from datetime import UTC , datetime , timedelta
1+ from datetime import datetime , timedelta
22from typing import Optional
33
44import pytest
@@ -31,7 +31,7 @@ async def get_token(self) -> AccessToken:
3131 token = AccessToken (
3232 username = self ._username ,
3333 public_key = self .local_public_key .to_bytes (),
34- expiration_time = str (datetime .now (UTC ) + timedelta (minutes = 1 )),
34+ expiration_time = str (datetime .now () + timedelta (minutes = 1 )),
3535 )
3636 token .signature = MockAuthorizer ._authority_private_key .sign (self ._token_to_bytes (token ))
3737 return token
@@ -52,7 +52,7 @@ def is_token_valid(self, access_token: AccessToken) -> bool:
5252 if expiration_time .tzinfo is not None :
5353 logger .exception (f"Expected to have no timezone for expiration time: { access_token .expiration_time } " )
5454 return False
55- if expiration_time < datetime .now (UTC ):
55+ if expiration_time < datetime .now ():
5656 logger .exception ("Access token has expired" )
5757 return False
5858
@@ -62,7 +62,7 @@ def is_token_valid(self, access_token: AccessToken) -> bool:
6262
6363 def does_token_need_refreshing (self , access_token : AccessToken ) -> bool :
6464 expiration_time = datetime .fromisoformat (access_token .expiration_time )
65- return expiration_time < datetime .now (UTC ) + self ._MAX_LATENCY
65+ return expiration_time < datetime .now () + self ._MAX_LATENCY
6666
6767 @staticmethod
6868 def _token_to_bytes (access_token : AccessToken ) -> bytes :
0 commit comments