test: Fix failing frontend integration tests for flask and django

Author: KShivendu

tests/frontendIntegration/django2x/polls/urls.py

@@ -16,6 +16,7 @@
     path("", views.get_info, name="/"),  # type: ignore
     path("update-jwt", views.update_jwt, name="update_jwt"),  # type: ignore
     path("update-jwt-with-handle", views.update_jwt_with_handle, name="update_jwt_with_handle"),  # type: ignore
+    path("session-claims-error", views.session_claim_error_api, name="session_claim_error_api"),  # type: ignore
     path("403-without-body", views.without_body_403, name="without_body_403"),  # type: ignore
     path("testing", views.testing, name="testing"),  # type: ignore
     path("logout", views.logout, name="logout"),  # type: ignore

tests/frontendIntegration/django2x/polls/views.py

@@ -31,7 +31,13 @@
     SessionRecipe,
 )
 from supertokens_python.recipe.session.framework.django.syncio import verify_session
-from supertokens_python.recipe.session.interfaces import APIInterface, RecipeInterface
+from supertokens_python.recipe.session.interfaces import (
+    APIInterface,
+    RecipeInterface,
+    ClaimValidationResult,
+    SessionClaimValidator,
+    JSONObject,
+)
 from supertokens_python.recipe.session.syncio import (
     create_new_session,
     get_session,
@@ -440,6 +446,22 @@ def update_jwt_with_handle(request: HttpRequest):
     return HttpResponse("")
 
 
+def gcv_for_session_claim_err(*_):  # type: ignore
+    class CustomValidator(SessionClaimValidator):
+        def should_refetch(self, payload: JSONObject, user_context: Dict[str, Any]):
+            return False
+
+        async def validate(self, payload: JSONObject, user_context: Dict[str, Any]):
+            return ClaimValidationResult(False, {"message": "testReason"})
+
+    return [CustomValidator("test-claim-failing")]
+
+
+@verify_session(override_global_claim_validators=gcv_for_session_claim_err)  # type: ignore
+def session_claim_error_api(request: HttpRequest):
+    return JsonResponse({})
+
+
 def without_body_403(request: HttpRequest):
     if request.method == "POST":
         return HttpResponse("", status=403)

tests/frontendIntegration/django3x/polls/urls.py

@@ -16,6 +16,7 @@
     path("", views.get_info, name="/"),  # type: ignore
     path("update-jwt", views.update_jwt, name="update_jwt"),  # type: ignore
     path("update-jwt-with-handle", views.update_jwt_with_handle, name="update_jwt_with_handle"),  # type: ignore
+    path("session-claims-error", views.session_claim_error_api, name="session_claim_error_api"),  # type: ignore
     path("403-without-body", views.without_body_403, name="without_body_403"),  # type: ignore
     path("testing", views.testing, name="testing"),  # type: ignore
     path("logout", views.logout, name="logout"),  # type: ignore

tests/frontendIntegration/django3x/polls/views.py

@@ -11,7 +11,13 @@
 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 # License for the specific language governing permissions and limitations
 # under the License.
-from supertokens_python.recipe.session.interfaces import APIInterface, RecipeInterface
+from supertokens_python.recipe.session.interfaces import (
+    APIInterface,
+    RecipeInterface,
+    ClaimValidationResult,
+    JSONObject,
+    SessionClaimValidator,
+)
 from typing import Dict, Union, Any
 import json
 import os
@@ -444,6 +450,22 @@ async def update_jwt_with_handle(request: HttpRequest):
     return HttpResponse("")
 
 
+def gcv_for_session_claim_err(*_):  # type: ignore
+    class CustomValidator(SessionClaimValidator):
+        def should_refetch(self, payload: JSONObject, user_context: Dict[str, Any]):
+            return False
+
+        async def validate(self, payload: JSONObject, user_context: Dict[str, Any]):
+            return ClaimValidationResult(False, {"message": "testReason"})
+
+    return [CustomValidator("test-claim-failing")]
+
+
+@verify_session(override_global_claim_validators=gcv_for_session_claim_err)  # type: ignore
+async def session_claim_error_api(request: HttpRequest):
+    return JsonResponse({})
+
+
 async def without_body_403(request: HttpRequest):
     if request.method == "POST":
         return HttpResponse("", status=403)

tests/frontendIntegration/flask-server/app.py

@@ -27,7 +27,13 @@
 from supertokens_python.recipe import session
 from supertokens_python.recipe.session import InputErrorHandlers, SessionRecipe
 from supertokens_python.recipe.session.framework.flask import verify_session
-from supertokens_python.recipe.session.interfaces import APIInterface, RecipeInterface
+from supertokens_python.recipe.session.interfaces import (
+    APIInterface,
+    RecipeInterface,
+    JSONObject,
+    SessionClaimValidator,
+    ClaimValidationResult,
+)
 from supertokens_python.recipe.session.syncio import (
     create_new_session,
     SessionContainer,
@@ -384,6 +390,24 @@ def update_jwt_with_handle_post():
     return resp
 
 
+def gcv_for_session_claim_err(*_):  # type: ignore
+    class CustomValidator(SessionClaimValidator):
+        def should_refetch(self, payload: JSONObject, user_context: Dict[str, Any]):
+            return False
+
+        async def validate(self, payload: JSONObject, user_context: Dict[str, Any]):
+            return ClaimValidationResult(False, {"message": "testReason"})
+
+    return [CustomValidator("test-claim-failing")]
+
+
+@app.route("/session-claims-error", methods=["POST"])  # type: ignore
+@verify_session(override_global_claim_validators=gcv_for_session_claim_err)  # type: ignore
+def session_claim_error_api():
+    # return empty json response
+    return jsonify({})
+
+
 @app.route("/403-without-body", methods=["POST"])  # type: ignore
 def without_body_403():
     # send 403 without body