Update JWK tests

Author: Michael Davis

tests/test_jwk.py

@@ -0,0 +1,118 @@
+
+from jose import jwk
+from jose.exceptions import JWKError
+
+import pytest
+
+hmac_key = {
+    "kty": "oct",
+    "kid": "018c0ae5-4d9b-471b-bfd6-eef314bc7037",
+    "use": "sig",
+    "alg": "HS256",
+    "k": "hJtXIZ2uSN5kbQfbtTNWbpdmhkV8FJG-Onbc6mxCcYg"
+}
+
+rsa_key = {
+    "kty": "RSA",
+    "kid": "[email protected]",
+    "use": "sig",
+    "n": "n4EPtAOCc9AlkeQHPzHStgAbgs7bTZLwUBZdR8_KuKPEHLd4rHVTeT-O-XV2jRojdNhxJWTDvNd7nqQ0VEiZQHz_AJmSCpMaJMRBSFKrKb2wqVwGU_NsYOYL-QtiWN2lbzcEe6XC0dApr5ydQLrHqkHHig3RBordaZ6Aj-oBHqFEHYpPe7Tpe-OfVfHd1E6cS6M1FZcD1NNLYD5lFHpPI9bTwJlsde3uhGqC0ZCuEHg8lhzwOHrtIQbS0FVbb9k3-tVTU4fg_3L_vniUFAKwuCLqKnS2BYwdq_mzSnbLY7h_qixoR7jig3__kRhuaxwUkRz5iaiQkqgc5gHdrNP5zw",
+    "e": "AQAB"
+}
+
+ec_key = {
+    "kty": "EC",
+    "kid": "[email protected]",
+    "use": "sig",
+    "crv": "P-521",
+    "x": "AHKZLLOsCOzz5cY97ewNUajB957y-C-U88c3v13nmGZx6sYl_oJXu9A5RkTKqjqvjyekWF-7ytDyRXYgCF5cj0Kt",
+    "y": "AdymlHvOiLxXkEhayXQnNCvDX4h9htZaCJN34kfmC6pV5OhQHiraVySsUdaQkAgDPrwQrJmbnX9cwlGfP-HqHZR1"
+}
+
+
+class TestJWK:
+
+    def test_interface(self):
+
+        key = jwk.Key()
+
+        with pytest.raises(NotImplementedError):
+            key._process_jwk(None)
+
+        with pytest.raises(NotImplementedError):
+            key.sign('')
+
+        with pytest.raises(NotImplementedError):
+            key.verify('', '')
+
+    def test_invalid_hash_alg(self):
+        with pytest.raises(JWKError):
+            key = jwk.HMACKey(hmac_key, 'RS512')
+
+        with pytest.raises(JWKError):
+            key = jwk.RSAKey(rsa_key, 'HS512')
+
+        with pytest.raises(JWKError):
+            key = jwk.ECKey(ec_key, 'RS512')
+
+    def test_invalid_jwk(self):
+
+        with pytest.raises(JWKError):
+            key = jwk.HMACKey(rsa_key, 'HS512')
+
+        with pytest.raises(JWKError):
+            key = jwk.HMACKey(hmac_key, 'RS512')
+
+        with pytest.raises(JWKError):
+            key = jwk.HMACKey(rsa_key, 'EC512')
+
+    def test_RSAKey_errors(self):
+
+        rsa_key = {
+            "kty": "RSA",
+            "kid": "[email protected]",
+            "use": "sig",
+            "n": "n4EPtAOCc9AlkeQHPzHStgAbgs7bTZLwUBZdR8_KuKPEHLd4rHVTeT-O-XV2jRojdNhxJWTDvNd7nqQ0VEiZQHz_AJmSCpMaJMRBSFKrKb2wqVwGU_NsYOYL-QtiWN2lbzcEe6XC0dApr5ydQLrHqkHHig3RBordaZ6Aj-oBHqFEHYpPe7Tpe-OfVfHd1E6cS6M1FZcD1NNLYD5lFHpPI9bTwJlsde3uhGqC0ZCuEHg8lhzwOHrtIQbS0FVbb9k3-tVTU4fg_3L_vniUFAKwuCLqKnS2BYwdq_mzSnbLY7h_qixoR7jig3__kRhuaxwUkRz5iaiQkqgc5gHdrNP5zw",
+            "e": "AQAB"
+        }
+
+        with pytest.raises(JWKError):
+            key = jwk.RSAKey(rsa_key, 'HS256')
+
+        rsa_key = {
+            "kty": "oct",
+            "kid": "[email protected]",
+            "use": "sig",
+            "n": "n4EPtAOCc9AlkeQHPzHStgAbgs7bTZLwUBZdR8_KuKPEHLd4rHVTeT-O-XV2jRojdNhxJWTDvNd7nqQ0VEiZQHz_AJmSCpMaJMRBSFKrKb2wqVwGU_NsYOYL-QtiWN2lbzcEe6XC0dApr5ydQLrHqkHHig3RBordaZ6Aj-oBHqFEHYpPe7Tpe-OfVfHd1E6cS6M1FZcD1NNLYD5lFHpPI9bTwJlsde3uhGqC0ZCuEHg8lhzwOHrtIQbS0FVbb9k3-tVTU4fg_3L_vniUFAKwuCLqKnS2BYwdq_mzSnbLY7h_qixoR7jig3__kRhuaxwUkRz5iaiQkqgc5gHdrNP5zw",
+            "e": "AQAB"
+        }
+
+        with pytest.raises(JWKError):
+            key = jwk.RSAKey(rsa_key, 'RS256')
+
+
+
+    def test_construct_from_jwk(self):
+
+        hmac_key = {
+            "kty": "oct",
+            "kid": "018c0ae5-4d9b-471b-bfd6-eef314bc7037",
+            "use": "sig",
+            "alg": "HS256",
+            "k": "hJtXIZ2uSN5kbQfbtTNWbpdmhkV8FJG-Onbc6mxCcYg"
+        }
+
+        key = jwk.construct(hmac_key)
+        assert isinstance(key, jwk.Key)
+
+    def test_construct_from_jwk_missing_alg(self):
+
+        hmac_key = {
+            "kty": "oct",
+            "kid": "018c0ae5-4d9b-471b-bfd6-eef314bc7037",
+            "use": "sig",
+            "k": "hJtXIZ2uSN5kbQfbtTNWbpdmhkV8FJG-Onbc6mxCcYg"
+        }
+
+        with pytest.raises(JWKError):
+            key = jwk.construct(hmac_key)

tests/test_jws.py

@@ -1,5 +1,4 @@
 
-
 from jose import jws
 from jose.constants import ALGORITHMS
 from jose.exceptions import JWSError