Add test for get_request_from_user_context

Author: nkshah2

tests/test_user_context.py

@@ -13,8 +13,12 @@
 # under the License.
 from typing import Any, Dict, List, Optional
 
+from fastapi import FastAPI
+from fastapi.testclient import TestClient
 from pytest import fixture, mark
+
 from supertokens_python import InputAppInfo, SupertokensConfig, init
+from supertokens_python.asyncio import get_request_from_user_context
 from supertokens_python.framework.fastapi import get_middleware
 from supertokens_python.recipe import emailpassword, session
 from supertokens_python.recipe.emailpassword.asyncio import sign_up
@@ -28,9 +32,6 @@
     RecipeInterface as SRecipeInterface,
 )
 
-from fastapi import FastAPI
-from fastapi.testclient import TestClient
-
 from .utils import clean_st, reset, setup_st, sign_in_request, start_st
 
 works = False
@@ -277,3 +278,113 @@ async def create_new_session(
             create_new_session_context_works,
         ]
     )
+
+
+@mark.asyncio
+async def test_get_request_from_user_context(driver_config_client: TestClient):
+    signin_api_context_works, signin_context_works, create_new_session_context_works = (
+        False,
+        False,
+        False,
+    )
+
+    def apis_override_email_password(param: APIInterface):
+        og_sign_in_post = param.sign_in_post
+
+        async def sign_in_post(
+            form_fields: List[FormField],
+            api_options: APIOptions,
+            user_context: Dict[str, Any],
+        ):
+            req = get_request_from_user_context(user_context)
+            if req:
+                assert req.method() == "POST"
+                assert req.get_path() == "/auth/signin"
+                nonlocal signin_api_context_works
+                signin_api_context_works = True
+
+            return await og_sign_in_post(form_fields, api_options, user_context)
+
+        param.sign_in_post = sign_in_post
+        return param
+
+    def functions_override_email_password(param: RecipeInterface):
+        og_sign_in = param.sign_in
+
+        async def sign_in(email: str, password: str, user_context: Dict[str, Any]):
+            req = get_request_from_user_context(user_context)
+            if req:
+                assert req.method() == "POST"
+                assert req.get_path() == "/auth/signin"
+                nonlocal signin_context_works
+                signin_context_works = True
+
+            return await og_sign_in(email, password, user_context)
+
+        param.sign_in = sign_in
+        return param
+
+    def functions_override_session(param: SRecipeInterface):
+        og_create_new_session = param.create_new_session
+
+        async def create_new_session(
+            user_id: str,
+            access_token_payload: Optional[Dict[str, Any]],
+            session_data_in_database: Optional[Dict[str, Any]],
+            disable_anti_csrf: Optional[bool],
+            user_context: Dict[str, Any],
+        ):
+            req = get_request_from_user_context(user_context)
+            if req:
+                assert req.method() == "POST"
+                assert req.get_path() == "/auth/signin"
+                nonlocal create_new_session_context_works
+                create_new_session_context_works = True
+
+            response = await og_create_new_session(
+                user_id,
+                access_token_payload,
+                session_data_in_database,
+                disable_anti_csrf,
+                user_context,
+            )
+            return response
+
+        param.create_new_session = create_new_session
+        return param
+
+    init(
+        supertokens_config=SupertokensConfig("http://localhost:3567"),
+        app_info=InputAppInfo(
+            app_name="SuperTokens Demo",
+            api_domain="http://api.supertokens.io",
+            website_domain="http://supertokens.io",
+        ),
+        framework="fastapi",
+        recipe_list=[
+            emailpassword.init(
+                override=emailpassword.InputOverrideConfig(
+                    apis=apis_override_email_password,
+                    functions=functions_override_email_password,
+                )
+            ),
+            session.init(
+                override=session.InputOverrideConfig(
+                    functions=functions_override_session
+                )
+            ),
+        ],
+    )
+    start_st()
+
+    await sign_up("[email protected]", "validpass123", {"manualCall": True})
+    res = sign_in_request(driver_config_client, "[email protected]", "validpass123")
+
+    assert res.status_code == 200
+    assert all(
+        [
+            signin_api_context_works,
+            signin_context_works,
+            create_new_session_context_works,
+        ]
+    )