test: Fix failing tests for django and flask

Author: KShivendu

tests/auth-react/django3x/mysite/utils.py

@@ -854,7 +854,7 @@ async def authorisation_url_get(
         session.init(override=session.InputOverrideConfig(apis=override_session_apis)),
         emailverification.init(
             mode="REQUIRED",
-            create_and_send_custom_email=ev_create_and_send_custom_email,  # TODO: Is it correct to create a seperate func for this?
+            create_and_send_custom_email=ev_create_and_send_custom_email,
             override=EVInputOverrideConfig(apis=override_email_verification_apis),
         ),
         emailpassword.init(

tests/auth-react/django3x/polls/urls.py

@@ -1,3 +1,5 @@
+import os
+
 from django.urls import path
 
 from . import views
@@ -11,3 +13,18 @@
     path("test/featureFlags", views.test_feature_flags, name="featureFlags"),
     path("beforeeach", views.before_each, name="beforeeach"),
 ]
+
+mode = os.environ.get("APP_MODE", "asgi")
+
+if mode == "asgi":
+    urlpatterns += [
+        path("unverifyEmail", views.unverify_email_api, name="unverifyEmail"),  # type: ignore
+        path("setRole", views.set_role_api, name="setRole"),  # type: ignore
+        path("checkRole", views.check_role_api, name="checkRole"),  # type: ignore
+    ]
+else:
+    urlpatterns += [
+        path("unverifyEmail", views.sync_unverify_email_api, name="unverifyEmail"),
+        path("setRole", views.sync_set_role_api, name="setRole"),
+        path("checkRole", views.sync_check_role_api, name="checkRole"),
+    ]

tests/auth-react/django3x/polls/views.py

@@ -13,19 +13,52 @@
 # under the License.
 import json
 import os
+from typing import List, Dict, Any
 
 from django.conf import settings
 from django.http import HttpRequest, HttpResponse, JsonResponse
 from mysite.store import get_codes, get_url_with_token
 from mysite.utils import custom_init
+
+from supertokens_python.recipe.emailverification import EmailVerificationClaim
 from supertokens_python.recipe.session import SessionContainer
+from supertokens_python.recipe.session.interfaces import SessionClaimValidator
+from supertokens_python.recipe.userroles import UserRoleClaim, PermissionClaim
 
 mode = os.environ.get("APP_MODE", "asgi")
 
+
+async def override_global_claim_validators(
+    gv: List[SessionClaimValidator],
+    _session: SessionContainer,
+    user_context: Dict[str, Any],
+):
+    validators = gv.copy()
+    req = user_context["_default"]["request"]
+    body = await req.json()
+
+    if body.get("role"):
+        info = body["role"]
+        validator = getattr(UserRoleClaim.validators, info["validator"])
+        validators.append(validator(*info["args"]))
+
+    if body.get("permission"):
+        info = body["permission"]
+        validator = getattr(PermissionClaim.validators, info["validator"])
+        validators.append(validator(*info["args"]))
+
+    return validators
+
+
 if mode == "asgi":
     from supertokens_python.recipe.session.framework.django.asyncio import (
         verify_session,
     )
+    from supertokens_python.recipe.userroles.asyncio import (
+        create_new_role_or_add_permissions,
+        add_role_to_user,
+    )
+    from supertokens_python.recipe.emailverification.asyncio import unverify_email
 
     @verify_session()
     async def session_info(request: HttpRequest):  # type: ignore
@@ -39,8 +72,36 @@ async def session_info(request: HttpRequest):  # type: ignore
             }
         )
 
+    @verify_session()
+    async def set_role_api(request: HttpRequest):
+        session_: SessionContainer = request.supertokens  # type: ignore
+        body = json.loads(request.body)
+        await create_new_role_or_add_permissions(body["role"], body["permissions"])
+        await add_role_to_user(session_.get_user_id(), body["role"])
+        await session_.fetch_and_set_claim(UserRoleClaim)
+        await session_.fetch_and_set_claim(PermissionClaim)
+        return JsonResponse({"status": "OK"})
+
+    @verify_session()
+    async def unverify_email_api(request: HttpRequest):
+        session_: SessionContainer = request.supertokens  # type: ignore
+        await unverify_email(session_.get_user_id())
+        await session_.fetch_and_set_claim(EmailVerificationClaim)
+        return JsonResponse({"status": "OK"})
+
+    @verify_session(override_global_claim_validators=override_global_claim_validators)
+    async def check_role_api():  # type: ignore
+        return JsonResponse({"status": "OK"})
+
 else:
     from supertokens_python.recipe.session.framework.django.syncio import verify_session
+    from supertokens_python.recipe.userroles.syncio import (
+        create_new_role_or_add_permissions as sync_create_new_role_or_add_permissions,
+        add_role_to_user as sync_add_role_to_user,
+    )
+    from supertokens_python.recipe.emailverification.syncio import (
+        unverify_email as sync_unverify_email,
+    )
 
     @verify_session()
     def session_info(request: HttpRequest):
@@ -54,6 +115,27 @@ def session_info(request: HttpRequest):
             }
         )
 
+    @verify_session()
+    def sync_set_role_api(request: HttpRequest):
+        session_: SessionContainer = request.supertokens  # type: ignore
+        body = json.loads(request.body)
+        sync_create_new_role_or_add_permissions(body["role"], body["permissions"])
+        sync_add_role_to_user(session_.get_user_id(), body["role"])
+        session_.sync_fetch_and_set_claim(UserRoleClaim)
+        session_.sync_fetch_and_set_claim(PermissionClaim)
+        return JsonResponse({"status": "OK"})
+
+    @verify_session()
+    def sync_unverify_email_api(request: HttpRequest):
+        session_: SessionContainer = request.supertokens  # type: ignore
+        sync_unverify_email(session_.get_user_id())
+        session_.sync_fetch_and_set_claim(EmailVerificationClaim)
+        return JsonResponse({"status": "OK"})
+
+    @verify_session(override_global_claim_validators=override_global_claim_validators)
+    def sync_check_role_api():
+        return JsonResponse({"status": "OK"})
+
 
 def ping(request: HttpRequest):
     return HttpResponse("success")
@@ -87,5 +169,12 @@ def before_each(request: HttpRequest):
 
 def test_feature_flags(request: HttpRequest):
     return JsonResponse(
-        {"available": ["passwordless", "thirdpartypasswordless", "generalerror"]}
+        {
+            "available": [
+                "passwordless",
+                "thirdpartypasswordless",
+                "generalerror",
+                "userroles",
+            ]
+        }
     )

tests/auth-react/flask-server/app.py

@@ -17,6 +17,7 @@
 from dotenv import load_dotenv
 from flask import Flask, g, jsonify, make_response, request
 from flask_cors import CORS
+
 from supertokens_python import (
     InputAppInfo,
     Supertokens,
@@ -47,10 +48,14 @@
     InputFormField,
     User,
 )
-from supertokens_python.recipe.emailverification import EmailVerificationRecipe
+from supertokens_python.recipe.emailverification import (
+    EmailVerificationRecipe,
+    EmailVerificationClaim,
+)
 from supertokens_python.recipe.emailverification import (
     InputOverrideConfig as EVInputOverrideConfig,
 )
+from supertokens_python.recipe.emailverification.asyncio import unverify_email
 from supertokens_python.recipe.emailverification.interfaces import (
     APIInterface as EmailVerificationAPIInterface,
 )
@@ -75,6 +80,7 @@
 from supertokens_python.recipe.session.interfaces import (
     APIInterface as SessionAPIInterface,
     SessionContainer,
+    SessionClaimValidator,
 )
 from supertokens_python.recipe.session.interfaces import APIOptions as SAPIOptions
 from supertokens_python.recipe.thirdparty import ThirdPartyRecipe
@@ -104,6 +110,11 @@
 from supertokens_python.recipe.thirdpartypasswordless.interfaces import (
     APIInterface as ThirdpartyPasswordlessAPIInterface,
 )
+from supertokens_python.recipe.userroles import UserRoleClaim, PermissionClaim
+from supertokens_python.recipe.userroles.asyncio import (
+    create_new_role_or_add_permissions,
+    add_role_to_user,
+)
 from supertokens_python.types import GeneralErrorResponse
 from typing_extensions import Literal
 from supertokens_python.recipe.emailverification.types import User as EVUser
@@ -1014,10 +1025,59 @@ def test_get_device():
 
 @app.get("/test/featureFlags")  # type: ignore
 def test_feature_flags():
-    available = ["passwordless", "thirdpartypasswordless", "generalerror"]
+    available = ["passwordless", "thirdpartypasswordless", "generalerror", "userroles"]
     return jsonify({"available": available})
 
 
+@app.get("/unverifyEmail")  # type: ignore
+@verify_session()
+async def unverify_email_api():
+    session_: SessionContainer = g.supertokens  # type: ignore
+    await unverify_email(session_.get_user_id())
+    await session_.fetch_and_set_claim(EmailVerificationClaim)
+    return jsonify({"status": "OK"})
+
+
+@app.route("/setRole", methods=["POST"])  # type: ignore
+@verify_session()
+async def verify_email_api():
+    session_: SessionContainer = g.supertokens  # type: ignore
+    body: Dict[str, Any] = request.get_json()  # type: ignore
+    await create_new_role_or_add_permissions(body["role"], body["permissions"])
+    await add_role_to_user(session_.get_user_id(), body["role"])
+    await session_.fetch_and_set_claim(UserRoleClaim)
+    await session_.fetch_and_set_claim(PermissionClaim)
+    return jsonify({"status": "OK"})
+
+
+async def override_global_claim_validators(
+    gv: List[SessionClaimValidator],
+    _session: SessionContainer,
+    user_context: Dict[str, Any],
+):
+    validators = gv.copy()
+    req = user_context["_default"]["request"]
+    body = await req.json()
+
+    if body.get("role"):
+        info = body["role"]
+        validator = getattr(UserRoleClaim.validators, info["validator"])
+        validators.append(validator(*info["args"]))
+
+    if body.get("permission"):
+        info = body["permission"]
+        validator = getattr(PermissionClaim.validators, info["validator"])
+        validators.append(validator(*info["args"]))
+
+    return validators
+
+
+@app.route("/checkRole", methods=["POST"])  # type: ignore
+@verify_session(override_global_claim_validators=override_global_claim_validators)
+async def check_role_api():
+    return jsonify({"status": "OK"})
+
+
 @app.route("/", defaults={"path": ""})  # type: ignore
 @app.route("/<path:path>")  # type: ignore
 def index(_: str):