Test all the things

Author: Michael Davis

jose/algorithms/__init__.py

@@ -10,8 +10,6 @@ def get_algorithm_object(algorithm):
     """
     Returns an algorithm object for the given algorithm.
     """
-    if algorithm not in ALGORITHMS.SUPPORTED:
-        raise JWSError('Algorithm not supported: %s' % algorithm)
 
     if algorithm == ALGORITHMS.HS256:
         return HMACAlgorithm(HMACAlgorithm.SHA256)
@@ -31,4 +29,4 @@ def get_algorithm_object(algorithm):
     if algorithm == ALGORITHMS.RS512:
         return RSAAlgorithm(RSAAlgorithm.SHA512)
 
-    raise JWSError('No algorithm found for %s.' % algorithm)
+    raise JWSError('Algorithm not supported: %s' % algorithm)

tests/test_jws.py

@@ -46,6 +46,10 @@ def test_claims_not_json(self):
         with pytest.raises(JWSError):
             jws.verify(token, 'secret', ['HS256'])
 
+    def test_invalid_key(self, claims):
+        with pytest.raises(JWSError):
+            jws.sign(claims, 'secret', algorithm='RS256')
+
 
 class TestHMAC: