From aa75013caa20556067a6fbda4c1a51b023f85732 Mon Sep 17 00:00:00 2001 From: Viraj Kanwade Date: Tue, 31 Dec 2024 20:59:48 -0800 Subject: [PATCH] fix: [#554] fixed tests/test_session.py::test_token_cookie_expires dst issue --- tests/test_session.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/test_session.py b/tests/test_session.py index e7b9f8f2d..264bad222 100644 --- a/tests/test_session.py +++ b/tests/test_session.py @@ -13,7 +13,7 @@ # under the License. import asyncio -from datetime import datetime, timedelta +from datetime import datetime, timedelta, timezone from typing import Any, Dict, List, Optional from unittest.mock import MagicMock @@ -664,13 +664,13 @@ async def test_token_cookie_expires( if c.name == "sAccessToken": # 100 years (set by the SDK) # some time must have elasped since the cookie was set. So less than current time assert ( - datetime.fromtimestamp(c.expires or 0) - timedelta(days=365.25 * 100) - < datetime.now() + datetime.fromtimestamp(c.expires or 0, tz=timezone.utc) - timedelta(days=365.25 * 100) + < datetime.now(tz=timezone.utc) ) if c.name == "sRefreshToken": # 100 days (set by the core) assert ( - datetime.fromtimestamp(c.expires or 0) - timedelta(days=100) - < datetime.now() + datetime.fromtimestamp(c.expires or 0, tz=timezone.utc) - timedelta(days=100) + < datetime.now(tz=timezone.utc) ) assert response.headers["anti-csrf"] != "" @@ -694,13 +694,13 @@ async def test_token_cookie_expires( if c.name == "sAccessToken": # 100 years (set by the SDK) # some time must have elasped since the cookie was set. So less than current time assert ( - datetime.fromtimestamp(c.expires or 0) - timedelta(days=365.25 * 100) - < datetime.now() + datetime.fromtimestamp(c.expires or 0, tz=timezone.utc) - timedelta(days=365.25 * 100) + < datetime.now(tz=timezone.utc) ) if c.name == "sRefreshToken": # 100 days (set by the core) assert ( - datetime.fromtimestamp(c.expires or 0) - timedelta(days=100) - < datetime.now() + datetime.fromtimestamp(c.expires or 0, tz=timezone.utc) - timedelta(days=100) + < datetime.now(tz=timezone.utc) ) assert response.headers["anti-csrf"] != ""