1818
1919from __future__ import annotations
2020
21+ from collections .abc import Callable
2122import datetime
2223import json
2324import ssl
24- from typing import Any , Callable , Literal
25+ from typing import Any , Literal
2526
2627from aiohttp import web
2728from cryptography import x509
@@ -56,7 +57,7 @@ def __class__(self) -> Credentials:
5657 def refresh (self , _ : Callable ) -> None :
5758 """Refreshes the access token."""
5859 self .token = "12345"
59- self .expiry = datetime .datetime .now (datetime .timezone . utc ) + datetime .timedelta (
60+ self .expiry = datetime .datetime .now (datetime .UTC ) + datetime .timedelta (
6061 minutes = 60
6162 )
6263
@@ -70,7 +71,7 @@ def expired(self) -> bool:
7071 """
7172 if self .expiry is None :
7273 return False
73- return not self .expiry > datetime .datetime .now (datetime .timezone . utc )
74+ return not self .expiry > datetime .datetime .now (datetime .UTC )
7475
7576 @property
7677 def universe_domain (self ) -> str :
@@ -94,11 +95,11 @@ def token_state(
9495 if self .expiry is None :
9596 return TokenState .FRESH
9697
97- expired = datetime .datetime .now (datetime .timezone . utc ) >= self .expiry
98+ expired = datetime .datetime .now (datetime .UTC ) >= self .expiry
9899 if expired :
99100 return TokenState .INVALID
100101
101- is_stale = datetime .datetime .now (datetime .timezone . utc ) >= (
102+ is_stale = datetime .datetime .now (datetime .UTC ) >= (
102103 self .expiry - _helpers .REFRESH_THRESHOLD
103104 )
104105 if is_stale :
@@ -117,9 +118,9 @@ def generate_cert(
117118 Generate a private key and cert object to be used in testing.
118119 """
119120 if cert_before is None :
120- cert_before = datetime .datetime .now (datetime .timezone . utc )
121+ cert_before = datetime .datetime .now (datetime .UTC )
121122 if cert_after is None :
122- cert_after = datetime .datetime .now (datetime .timezone . utc ) + datetime .timedelta (hours = 1 )
123+ cert_after = datetime .datetime .now (datetime .UTC ) + datetime .timedelta (hours = 1 )
123124 # generate private key
124125 key = rsa .generate_private_key (public_exponent = 65537 , key_size = 2048 )
125126 common_name = f"{ project } :{ name } "
@@ -168,9 +169,9 @@ def client_key_signed_cert(
168169 Create a PEM encoded certificate that is signed by given public key.
169170 """
170171 if cert_before is None :
171- cert_before = datetime .datetime .now (datetime .timezone . utc )
172+ cert_before = datetime .datetime .now (datetime .UTC )
172173 if cert_expiration is None :
173- cert_expiration = datetime .datetime .now (datetime .timezone . utc ) + datetime .timedelta (hours = 1 )
174+ cert_expiration = datetime .datetime .now (datetime .UTC ) + datetime .timedelta (hours = 1 )
174175 # configure cert subject
175176 subject = issuer = x509 .Name (
176177 [
@@ -234,9 +235,9 @@ def __init__(
234235 cert_expiration : datetime .datetime | None = None ,
235236 ) -> None :
236237 if cert_before is None :
237- cert_before = datetime .datetime .now (datetime .timezone . utc )
238+ cert_before = datetime .datetime .now (datetime .UTC )
238239 if cert_expiration is None :
239- cert_expiration = datetime .datetime .now (datetime .timezone . utc ) + datetime .timedelta (hours = 1 )
240+ cert_expiration = datetime .datetime .now (datetime .UTC ) + datetime .timedelta (hours = 1 )
240241 if dns_names is None :
241242 dns_names = ["abcde.12345.us-central1.sql.goog" ]
242243 if ip_addrs is None :
0 commit comments