diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/contrib/test_algorithms.py | 5 | ||||
| -rw-r--r-- | tests/test_algorithms.py | 8 |
2 files changed, 7 insertions, 6 deletions
diff --git a/tests/contrib/test_algorithms.py b/tests/contrib/test_algorithms.py index 6d5ca75..b774ad1 100644 --- a/tests/contrib/test_algorithms.py +++ b/tests/contrib/test_algorithms.py @@ -1,5 +1,6 @@ import base64 +from jwt.exceptions import InvalidAsymmetricKeyError from jwt.utils import force_bytes, force_unicode import pytest @@ -36,7 +37,7 @@ class TestPycryptoAlgorithms: def test_rsa_should_reject_non_string_key(self): algo = RSAAlgorithm(RSAAlgorithm.SHA256) - with pytest.raises(TypeError): + with pytest.raises(InvalidAsymmetricKeyError): algo.prepare_key(None) def test_rsa_sign_should_generate_correct_signature_value(self): @@ -117,7 +118,7 @@ class TestEcdsaAlgorithms: def test_ec_should_reject_non_string_key(self): algo = ECAlgorithm(ECAlgorithm.SHA256) - with pytest.raises(TypeError): + with pytest.raises(InvalidAsymmetricKeyError): algo.prepare_key(None) def test_ec_should_accept_unicode_key(self): diff --git a/tests/test_algorithms.py b/tests/test_algorithms.py index 11d8cd0..69501ee 100644 --- a/tests/test_algorithms.py +++ b/tests/test_algorithms.py @@ -58,11 +58,11 @@ class TestAlgorithms: def test_hmac_should_reject_nonstring_key(self): algo = HMACAlgorithm(HMACAlgorithm.SHA256) - with pytest.raises(TypeError) as context: + with pytest.raises(InvalidKeyError) as context: algo.prepare_key(object()) exception = context.value - assert str(exception) == 'Expected a string value' + assert str(exception) == 'HMAC secret key must be a string type.' def test_hmac_should_accept_unicode_key(self): algo = HMACAlgorithm(HMACAlgorithm.SHA256) @@ -144,7 +144,7 @@ class TestAlgorithms: def test_rsa_should_reject_non_string_key(self): algo = RSAAlgorithm(RSAAlgorithm.SHA256) - with pytest.raises(TypeError): + with pytest.raises(InvalidKeyError): algo.prepare_key(None) @pytest.mark.skipif(not has_crypto, reason='Not supported without cryptography library') @@ -358,7 +358,7 @@ class TestAlgorithms: def test_ec_should_reject_non_string_key(self): algo = ECAlgorithm(ECAlgorithm.SHA256) - with pytest.raises(TypeError): + with pytest.raises(InvalidKeyError): algo.prepare_key(None) @pytest.mark.skipif(not has_crypto, reason='Not supported without cryptography library') |