From 21ba18e475edefc6245de1149d7c587c41a6f5dd Mon Sep 17 00:00:00 2001 From: rishabhpoddar Date: Sun, 13 Oct 2024 13:37:34 +0530 Subject: [PATCH] fixes stuff --- tests/test-server/override_logging.py | 2 ++ tests/test-server/test_functions_mapper.py | 8 ++++---- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/test-server/override_logging.py b/tests/test-server/override_logging.py index 41be4e32..b29cf1b4 100644 --- a/tests/test-server/override_logging.py +++ b/tests/test-server/override_logging.py @@ -151,3 +151,5 @@ def transform_logged_data(data: Any, visited: Union[Set[Any], None] = None) -> A return data.to_json() if isinstance(data, IsVerifiedSCV): return "IsVerifiedSCV" + + return data diff --git a/tests/test-server/test_functions_mapper.py b/tests/test-server/test_functions_mapper.py index 5f76f915..df95a83a 100644 --- a/tests/test-server/test_functions_mapper.py +++ b/tests/test-server/test_functions_mapper.py @@ -346,8 +346,8 @@ async def consume_code_post( ) -> Any: o = await api_options.request.json() assert o is not None - if o.get("user_context", {}).get("DO_LINK") is not None: - user_context["DO_LINK"] = o["user_context"]["DO_LINK"] + if o.get("userContext", {}).get("DO_LINK") is not None: + user_context["DO_LINK"] = o["userContext"]["DO_LINK"] return await og( pre_auth_session_id, user_input_code, @@ -417,8 +417,8 @@ async def sign_up_post( if "signUpPOST" in eval_str: n = await api_options.request.json() assert n is not None - if n.get("user_context", {}).get("DO_LINK") is not None: - user_context["DO_LINK"] = n["user_context"]["DO_LINK"] + if n.get("userContext", {}).get("DO_LINK") is not None: + user_context["DO_LINK"] = n["userContext"]["DO_LINK"] return await og_sign_up_post( form_fields, tenant_id,