44
55
66import base64
7- import calendar
7+ import datetime
88import json
99import os
1010import time
1111
12- import iso8601
1312import pretend
1413import pytest
1514
@@ -46,7 +45,7 @@ def test_generate(self, secret, now, iv, src, token, backend):
4645 f = Fernet (secret .encode ("ascii" ), backend = backend )
4746 actual_token = f ._encrypt_from_parts (
4847 src .encode ("ascii" ),
49- calendar . timegm ( iso8601 . parse_date (now ).utctimetuple ()),
48+ int ( datetime . datetime . fromisoformat (now ).timestamp ()),
5049 bytes (iv ),
5150 )
5251 assert actual_token == token .encode ("ascii" )
@@ -60,7 +59,7 @@ def test_verify(
6059 ):
6160 # secret & token are both str
6261 f = Fernet (secret .encode ("ascii" ), backend = backend )
63- current_time = calendar . timegm ( iso8601 . parse_date (now ).utctimetuple ())
62+ current_time = int ( datetime . datetime . fromisoformat (now ).timestamp ())
6463 payload = f .decrypt_at_time (
6564 token , # str
6665 ttl = ttl_sec ,
@@ -86,7 +85,7 @@ def test_verify(
8685 @json_parametrize (("secret" , "token" , "now" , "ttl_sec" ), "invalid.json" )
8786 def test_invalid (self , secret , token , now , ttl_sec , backend , monkeypatch ):
8887 f = Fernet (secret .encode ("ascii" ), backend = backend )
89- current_time = calendar . timegm ( iso8601 . parse_date (now ).utctimetuple ())
88+ current_time = int ( datetime . datetime . fromisoformat (now ).timestamp ())
9089 with pytest .raises (InvalidToken ):
9190 f .decrypt_at_time (
9291 token .encode ("ascii" ),
0 commit comments