fix test failure

Author: KShivendu

supertokens_python/recipe/session/utils.py

@@ -521,13 +521,9 @@ async def update_claims_in_payload_if_needed(
         log_debug_message(
             "update_claims_in_payload_if_needed checking %s", validator.id
         )
-        if (
-            hasattr(validator, "claim")
-            and (validator.claim is not None)
-            and (
-                await resolve(
-                    validator.should_refetch(new_access_token_payload, user_context)
-                )
+        if (validator.claim is not None) and (
+            await resolve(
+                validator.should_refetch(new_access_token_payload, user_context)
             )
         ):
             log_debug_message(

tests/sessions/claims/test_assert_claims.py

@@ -1,3 +1,4 @@
+import json
 from typing import Any, Dict, TypeVar, Union
 from unittest.mock import patch
 
@@ -49,16 +50,19 @@ async def test_should_call_validate_with_the_same_payload_object():
     class DummyClaimValidator(SessionClaimValidator):
         def __init__(self, claim: SessionClaim[Any]):
             super().__init__("claim_validator_id", claim)
-            self.validate_call_count = 0
+            self.validate_calls: Dict[str, int] = {}
 
         async def validate(
             self, payload: JSONObject, user_context: Union[Dict[str, Any], None] = None
         ):
-            self.validate_call_count += 1
+            payload_json = json.dumps(payload)
+            self.validate_calls[payload_json] = (
+                self.validate_calls.get(payload_json, 0) + 1
+            )
             return ClaimValidationResult(is_valid=True)
 
         def should_refetch(self, payload: JSONObject, user_context: Dict[str, Any]):
-            return True
+            return False
 
     dummy_claim = PrimitiveClaim("st-claim", lambda _, __: "Hello world")
 
@@ -72,6 +76,6 @@ def should_refetch(self, payload: JSONObject, user_context: Dict[str, Any]):
         wraps=session.update_access_token_payload,
     ) as mock:
         await session.assert_claims([dummy_claim.validators.dummy_claim_validator])  # type: ignore
-        mock.assert_not_called()
 
-        assert dummy_claim_validator.validate_call_count == 1
+        assert dummy_claim_validator.validate_calls == {json.dumps(payload): 1}
+        mock.assert_not_called()