feat(types): Added boolean types for registry value and values options

added tests for boolean value and values in registry
authlib · Jun 15, 2024 · 5e58bb3 · 5e58bb3
1 parent 2250618
commit 5e58bb3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/joserfc/rfc7519/registry.py b/src/joserfc/rfc7519/registry.py
@@ -12,8 +12,8 @@
 class ClaimsOption(TypedDict, total=False):
     essential: bool
     allow_blank: Optional[bool]
-    value: Union[str, int]
-    values: Union[List[Union[str, int]], List[str], List[int]]
+    value: Union[str, int, bool]
+    values: Union[List[Union[str, int, bool]], List[str], List[int], List[bool]]
 
 
 class ClaimsRegistry:

diff --git a/tests/jwt/test_claims.py b/tests/jwt/test_claims.py
@@ -65,12 +65,20 @@ def test_option_value(self):
         claims_requests = jwt.JWTClaimsRegistry(sub={"essential": True, "value": "123"})
         self.assertRaises(InvalidClaimError, claims_requests.validate, {"sub": "a"})
         claims_requests.validate({"sub": "123"})
+        claims_requests = jwt.JWTClaimsRegistry(sub={"essential":True,"value":True})
+        claims_requests.validate({"sub":True})
+        self.assertRaises(InvalidClaimError,claims_requests.validate,{"sub":False})
+        claims_requests = jwt.JWTClaimsRegistry(sub={"essential":True,"value":False})
+        claims_requests.validate({"sub":False})
+        self.assertRaises(InvalidClaimError,claims_requests.validate,{"sub":True})
 
     def test_option_values(self):
-        claims_requests = jwt.JWTClaimsRegistry(sub={"essential": True, "values": ["1", "2"]})
+        claims_requests = jwt.JWTClaimsRegistry(sub={"essential": True, "values": ["1", "2", True, False]})
         self.assertRaises(InvalidClaimError, claims_requests.validate, {"sub": "a"})
         claims_requests.validate({"sub": "1"})
         claims_requests.validate({"sub": "2"})
+        claims_requests.validate({"sub":True})
+        claims_requests.validate({"sub":False})
 
     def test_int_claims(self):
         now = int(time.time())