fix(jwk): add __bool__ on KeySet

authlib · Aug 6, 2023 · 4d9b838 · 4d9b838
1 parent 752fae7
commit 4d9b838
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/joserfc/_keys.py b/src/joserfc/_keys.py
@@ -112,6 +112,9 @@ def __init__(self, keys: t.List[Key]):
     def __iter__(self) -> t.Iterator[Key]:
         return iter(self.keys)
 
+    def __bool__(self) -> bool:
+        return bool(self.keys)
+
     def as_dict(self, private: t.Optional[bool] = None, **params: t.Any) -> KeySetSerialization:
         keys: t.List[DictKey] = []
 

diff --git a/tests/jwk/test_jwk_set.py b/tests/jwk/test_jwk_set.py
@@ -58,3 +58,10 @@ def test_key_set_methods(self):
         key_set = KeySet.generate_key_set("oct", 8, count=1)
         k2 = key_set.get_by_kid()
         self.assertIsInstance(k2, OctKey)
+
+    def test_key_set_bool(self):
+        key_set = KeySet([])
+        self.assertFalse(key_set)
+
+        key_set = KeySet([OctKey.generate_key()])
+        self.assertTrue(key_set)