fixes tests

Author: rishabhpoddar

supertokens_python/recipe/session/jwks.py

@@ -66,8 +66,6 @@ def get_cached_keys() -> Optional[List[PyJWK]]:
         # if it has a valid cache entry from one of the core URLs. It will only attempt to fetch
         # from the cores again after the entry in the cache is expired
         if cached_keys.is_fresh():
-            if environ.get("SUPERTOKENS_ENV") == "testing":
-                log_debug_message("Returning JWKS from cache")
             return cached_keys.keys
 
     return None
@@ -97,6 +95,8 @@ def get_latest_keys(kid: Optional[str] = None) -> List[PyJWK]:
     with RWLockContext(mutex, read=True):
         matching_keys = find_matching_keys(get_cached_keys(), kid)
         if matching_keys is not None:
+            if environ.get("SUPERTOKENS_ENV") == "testing":
+                log_debug_message("Returning JWKS from cache")
             return matching_keys
         # otherwise unknown kid, will continue to reload the keys
 

tests/sessions/test_jwks.py

@@ -31,6 +31,7 @@
     get_latest_keys,
 )
 from supertokens_python.utils import utf_base64encode
+from tests.utils import min_api_version
 
 from _pytest.logging import LogCaptureFixture
 
@@ -491,6 +492,7 @@ async def test_session_verification_of_jwt_based_on_session_payload(
     assert s_.get_user_id() == "userId"
 
 
+@min_api_version("3.0")
 async def test_session_verification_of_jwt_based_on_session_payload_with_check_db():
     init(**get_st_init_args(recipe_list=[session.init()]))
     start_st()
@@ -500,7 +502,6 @@ async def test_session_verification_of_jwt_based_on_session_payload_with_check_d
     payload = s.get_access_token_payload()
     del payload["iat"]
     del payload["exp"]
-    payload["tId"] = "public"  # tenant id
 
     now = get_timestamp_ms()
     jwt_expiry = now + 10 * 1000  # expiry jwt after 10sec

tests/utils.py

@@ -468,6 +468,7 @@ async def get_api_version():
 
     api_version = asyncio.gather(get_api_version())
     core_version = loop.run_until_complete(api_version)[0]  # type: ignore # pylint: disable=unused-variable
+    teardown_function(_)
     return core_version