fixes more tests

supertokens · Sep 25, 2024 · 19a13bf · 19a13bf
1 parent 85221ea
commit 19a13bf
Show file tree

Hide file tree

Showing 7 changed files with 25 additions and 12 deletions.
diff --git a/tests/sessions/claims/test_create_new_session.py b/tests/sessions/claims/test_create_new_session.py
@@ -32,7 +32,7 @@ async def test_create_access_token_payload_with_session_claims(timestamp: int):
     s = await create_new_session(dummy_req, "public", RecipeUserId("someId"))
 
     payload = s.get_access_token_payload()
-    assert len(payload) == 10
+    assert len(payload) == 11
     assert payload["st-true"] == {"v": True, "t": timestamp}
 
 
@@ -45,7 +45,7 @@ async def test_should_create_access_token_payload_with_session_claims_with_an_no
     s = await create_new_session(dummy_req, "public", RecipeUserId("someId"))
 
     payload = s.get_access_token_payload()
-    assert len(payload) == 9
+    assert len(payload) == 10
     assert payload.get("st-true") is None
 
 
@@ -70,6 +70,6 @@ async def test_should_merge_claims_and_passed_access_token_payload_obj(timestamp
     s = await create_new_session(dummy_req, "public", RecipeUserId("someId"))
 
     payload = s.get_access_token_payload()
-    assert len(payload) == 11
+    assert len(payload) == 12
     assert payload["st-true"] == {"v": True, "t": timestamp}
     assert payload["user-custom-claim"] == "foo"
diff --git a/tests/sessions/claims/test_primitive_array_claim.py b/tests/sessions/claims/test_primitive_array_claim.py
@@ -80,9 +80,16 @@ async def test_primitive_claim_matching__add_to_payload():
 async def test_primitive_claim_fetch_value_params_correct():
     claim = PrimitiveArrayClaim("key", sync_fetch_value)
     user_id, ctx = "user_id", {}
-    await claim.build(user_id, RecipeUserId(user_id), DEFAULT_TENANT_ID, ctx)
+    recipe_user_id = RecipeUserId(user_id)
+    await claim.build(user_id, recipe_user_id, DEFAULT_TENANT_ID, ctx)
     assert sync_fetch_value.call_count == 1
-    assert (user_id, DEFAULT_TENANT_ID, ctx) == sync_fetch_value.call_args_list[0][
+    assert (
+        user_id,
+        recipe_user_id,
+        DEFAULT_TENANT_ID,
+        ctx,
+        {},
+    ) == sync_fetch_value.call_args_list[0][
         0
     ]  # extra [0] refers to call params
 

diff --git a/tests/sessions/claims/test_primitive_claim.py b/tests/sessions/claims/test_primitive_claim.py
@@ -49,7 +49,13 @@ async def test_primitive_claim_fetch_value_params_correct():
     user_id, ctx = "user_id", {}
     await claim.build(user_id, RecipeUserId(user_id), DEFAULT_TENANT_ID, ctx)
     assert sync_fetch_value.call_count == 1
-    assert (user_id, DEFAULT_TENANT_ID, ctx) == sync_fetch_value.call_args_list[0][
+    assert (
+        user_id,
+        RecipeUserId(user_id),
+        DEFAULT_TENANT_ID,
+        ctx,
+        {},
+    ) == sync_fetch_value.call_args_list[0][
         0
     ]  # extra [0] refers to call params
 

diff --git a/tests/sessions/claims/test_set_claim_value.py b/tests/sessions/claims/test_set_claim_value.py
@@ -62,7 +62,7 @@ async def test_should_overwrite_claim_value(timestamp: int):
     s = await create_new_session(dummy_req, "public", RecipeUserId("someId"))
 
     payload = s.get_access_token_payload()
-    assert len(payload) == 10
+    assert len(payload) == 11
     assert payload["st-true"] == {"t": timestamp, "v": True}
 
     await s.set_claim_value(TrueClaim, False)
@@ -81,7 +81,7 @@ async def test_should_overwrite_claim_value_using_session_handle(timestamp: int)
     s = await create_new_session(dummy_req, "public", RecipeUserId("someId"))
 
     payload = s.get_access_token_payload()
-    assert len(payload) == 10
+    assert len(payload) == 11
     assert payload["st-true"] == {"t": timestamp, "v": True}
 
     await set_claim_value(s.get_handle(), TrueClaim, False)

diff --git a/tests/sessions/claims/test_verify_session.py b/tests/sessions/claims/test_verify_session.py
@@ -447,7 +447,7 @@ async def test_should_allow_if_assert_claims_returns_no_error(
         assert validators == [validator]
         assert ctx["_default"]["request"]
         recipe_impl_mock.validate_claims.assert_called_once_with(  # type: ignore
-            "test_user_id", {}, [validator], ctx
+            "test_user_id", RecipeUserId("test_user_id"), {}, [validator], ctx
         )
 
 

diff --git a/tests/sessions/claims/utils.py b/tests/sessions/claims/utils.py
@@ -9,8 +9,8 @@
 from supertokens_python.types import RecipeUserId
 from tests.utils import st_init_common_args
 
-TrueClaim = BooleanClaim("st-true", fetch_value=lambda _, __, ___: True)  # type: ignore
-NoneClaim = BooleanClaim("st-none", fetch_value=lambda _, __, ___: None)  # type: ignore
+TrueClaim = BooleanClaim("st-true", fetch_value=lambda _, __, ___, _____, ______: True)
+NoneClaim = BooleanClaim("st-none", fetch_value=lambda _, __, ___, _____, ______: None)
 
 
 def session_functions_override_with_claim(

diff --git a/tests/sessions/test_access_token_version.py b/tests/sessions/test_access_token_version.py
@@ -47,7 +47,7 @@ async def test_access_token_v4():
         False,
     )
     assert res["userId"] == "user-id"
-    assert parsed_info.version == 4
+    assert parsed_info.version == 5
 
 
 async def test_parsing_access_token_v2():