Merge pull request #321 from supertokens/test-cicd/fix-failing-tests

test: Fix failing unit tests for access token v3

Author: rishabhpoddar

Makefile

@@ -17,7 +17,7 @@ set-up-hooks:
 	chmod +x .git/hooks/pre-commit
 
 test:
-	pytest --reruns 3 --reruns-delay 5 ./tests/
+	pytest -vv --reruns 3 --reruns-delay 5 ./tests/
 
 dev-install:
 	pip install -r dev-requirements.txt

supertokens_python/framework/django/django_request.py

@@ -21,11 +21,10 @@
 
 if TYPE_CHECKING:
     from supertokens_python.recipe.session.interfaces import SessionContainer
+    from django.http import HttpRequest
 
 
 class DjangoRequest(BaseRequest):
-    from django.http import HttpRequest
-
     def __init__(self, request: HttpRequest):
         super().__init__()
         self.request = request

supertokens_python/framework/fastapi/fastapi_request.py

@@ -20,12 +20,10 @@
 
 if TYPE_CHECKING:
     from supertokens_python.recipe.session.interfaces import SessionContainer
+    from fastapi import Request
 
 
 class FastApiRequest(BaseRequest):
-
-    from fastapi import Request
-
     def __init__(self, request: Request):
         super().__init__()
         self.request = request

supertokens_python/framework/flask/flask_request.py

@@ -19,11 +19,10 @@
 
 if TYPE_CHECKING:
     from supertokens_python.recipe.session.interfaces import SessionContainer
+    from flask.wrappers import Request
 
 
 class FlaskRequest(BaseRequest):
-    from flask.wrappers import Request
-
     def __init__(self, req: Request):
         super().__init__()
         self.request = req

supertokens_python/recipe/session/asyncio/__init__.py

@@ -93,7 +93,13 @@ async def create_new_session_without_request_response(
     claims_added_by_other_recipes = (
         SessionRecipe.get_instance().get_claims_added_by_other_recipes()
     )
-    final_access_token_payload = access_token_payload
+    app_info = SessionRecipe.get_instance().app_info
+    issuer = (
+        app_info.api_domain.get_as_string_dangerous()
+        + app_info.api_base_path.get_as_string_dangerous()
+    )
+
+    final_access_token_payload = {**access_token_payload, "iss": issuer}
 
     for claim in claims_added_by_other_recipes:
         update = await claim.build(user_id, user_context)

supertokens_python/recipe/session/exceptions.py

@@ -28,6 +28,7 @@ def raise_token_theft_exception(user_id: str, session_handle: str) -> NoReturn:
 def raise_try_refresh_token_exception(ex: Union[str, Exception]) -> NoReturn:
     if isinstance(ex, SuperTokensError):
         raise ex
+
     raise TryRefreshTokenError(ex) from None
 
 
@@ -36,16 +37,18 @@ def raise_unauthorised_exception(
     clear_tokens: bool = True,
     response_mutators: Optional[List[ResponseMutator]] = None,
 ) -> NoReturn:
-    if response_mutators is None:
-        response_mutators = []
-
     err = UnauthorisedError(msg, clear_tokens)
-    err.response_mutators.extend(UnauthorisedError.response_mutators)
+
+    if response_mutators is not None:
+        err.response_mutators.extend(response_mutators)
+
     raise err
 
 
 class SuperTokensSessionError(SuperTokensError):
-    response_mutators: List[ResponseMutator] = []
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+        self.response_mutators: List[ResponseMutator] = []
 
 
 class TokenTheftError(SuperTokensSessionError):

supertokens_python/recipe/session/jwks.py

@@ -46,8 +46,8 @@ def reload(self):
             with urllib.request.urlopen(self.uri, timeout=self.timeout_sec) as response:
                 self.jwk_set = PyJWKSet.from_dict(json.load(response))  # type: ignore
                 self.last_fetch_time = get_timestamp_ms()
-        except URLError as e:
-            raise JWKSRequestError(f'Failed to fetch data from the url, err: "{e}"')
+        except URLError:
+            raise JWKSRequestError("Failed to fetch jwk set from the configured uri")
 
     def is_cooling_down(self) -> bool:
         return (self.last_fetch_time > 0) and (
@@ -81,15 +81,17 @@ def get_matching_key_from_jwt(self, token: str) -> PyJWK:
 
         try:
             return self.jwk_set[kid]  # type: ignore
-        except IndexError:
+        except KeyError:
             if not self.is_cooling_down():
                 # One more attempt to fetch the latest keys
                 # and then try to find the key again.
                 self.reload()
                 try:
                     return self.jwk_set[kid]  # type: ignore
-                except IndexError:
+                except KeyError:
                     pass
+        except Exception:
+            raise JWKSKeyNotFoundError("No key found for the given kid")
 
         raise JWKSKeyNotFoundError("No key found for the given kid")
 

supertokens_python/recipe/session/recipe.py

@@ -96,13 +96,6 @@ def __init__(
         expose_access_token_to_frontend_in_cookie_based_auth: Union[bool, None] = None,
     ):
         super().__init__(recipe_id, app_info)
-        self.openid_recipe = OpenIdRecipe(
-            recipe_id,
-            app_info,
-            None,
-            None,
-            override.openid_feature if override is not None else None,
-        )
         self.config = validate_and_normalise_user_input(
             app_info,
             cookie_domain,
@@ -117,6 +110,13 @@ def __init__(
             use_dynamic_access_token_signing_key,
             expose_access_token_to_frontend_in_cookie_based_auth,
         )
+        self.openid_recipe = OpenIdRecipe(
+            recipe_id,
+            app_info,
+            None,
+            None,
+            override.openid_feature if override is not None else None,
+        )
         log_debug_message("session init: anti_csrf: %s", self.config.anti_csrf)
         if self.config.cookie_domain is not None:
             log_debug_message(
@@ -221,7 +221,10 @@ async def handle_api_request(
     async def handle_error(
         self, request: BaseRequest, err: SuperTokensError, response: BaseResponse
     ) -> BaseResponse:
-        if isinstance(err, SuperTokensSessionError):
+        if (
+            isinstance(err, SuperTokensSessionError)
+            and err.response_mutators is not None
+        ):
             for mutator in err.response_mutators:
                 mutator(response)
 

supertokens_python/recipe/session/session_class.py

@@ -70,7 +70,9 @@ async def attach_to_request_response(
                     anti_csrf_response_mutator(self.anti_csrf_token)
                 )
 
-        request.set_session(self)
+        request.set_session(
+            self
+        )  # Although this is called in recipe/session/framework/**/__init__.py. It's required in case of python because functions like create_new_session(req, "user-id") can be called in the framework view handler as well
 
     async def revoke_session(self, user_context: Union[Any, None] = None) -> None:
         if user_context is None:

supertokens_python/recipe/session/session_functions.py

@@ -194,7 +194,7 @@ async def get_session(
             if time_created <= time.time() - JWKCacheMaxAgeInMs:
                 raise e
         else:
-            # Since v3 (and above) tokens contain a kid we can trust the cache-refresh mechanism of the pyjwt library
+            # Since v3 (and above) tokens contain a kid we can trust the cache refresh mechanism built on top of the pyjwt lib
             # This means we do not need to call the core since the signature wouldn't pass verification anyway.
             raise e
 
@@ -281,13 +281,26 @@ async def get_session(
                 response["session"]["handle"],
                 response["session"]["userId"],
                 response["session"]["userDataInJWT"],
-                response["accessToken"]["expiry"],
+                (
+                    response.get("accessToken", {}).get(
+                        "expiry"
+                    )  # if we got a new accesstoken we take the expiry time from there
+                    or (
+                        access_token_info is not None
+                        and access_token_info.get("expiryTime")
+                    )  # if we didn't get a new access token but could validate the token take that info (alwaysCheckCore === true, or parentRefreshTokenHash1 !== null)
+                    or parsed_access_token.payload[
+                        "expiryTime"
+                    ]  # if the token didn't pass validation, but we got here, it means it was a v2 token that we didn't have the key cached for.
+                ),  # This will throw error if others are none and 'expiryTime' key doesn't exist in the payload
             ),
             GetSessionAPIResponseAccessToken(
                 response["accessToken"]["token"],
                 response["accessToken"]["expiry"],
                 response["accessToken"]["createdTime"],
-            ),
+            )
+            if "accessToken" in response
+            else None,
         )
     if response["status"] == "UNAUTHORISED":
         log_debug_message("getSession: Returning UNAUTHORISED because of core response")