feat: pick_random_key algorithm parameter is optional

Author: azmeuk

src/joserfc/_keys.py

@@ -152,8 +152,8 @@ def get_by_kid(self, kid: str | None = None, parameters: KeyParameters | None =
             return keys[0]
         raise InvalidKeyIdError(f"No key for kid: '{kid}'")
 
-    def pick_random_key(self, algorithm: str, parameters: KeyParameters | None = None) -> t.Optional[Key]:
-        key_types = self.algorithm_keys.get(algorithm)
+    def pick_random_key(self, algorithm: str | None = None, parameters: KeyParameters | None = None) -> t.Optional[Key]:
+        key_types = self.algorithm_keys.get(algorithm) if algorithm else None
         if key_types:
             keys = [k for k in self.keys if k.key_type in key_types]
         else:

tests/jwk/test_jwk_set.py

@@ -52,6 +52,8 @@ def test_random_key(self):
         self.assertIsNotNone(key1)
         key2 = key_set.pick_random_key("RS256")
         self.assertIsNone(key2)
+        key3 = key_set.pick_random_key()
+        self.assertIsNotNone(key3)
 
     def test_key_set_methods(self):
         key_set = KeySet.generate_key_set("oct", 8)