Added tests of the new functions

Author: rohe

tests/test_02_jwk.py

@@ -17,6 +17,7 @@
 
 from cryptojwt.exception import DeSerializationNotPossible, UnsupportedAlgorithm
 from cryptojwt.exception import WrongUsage
+from cryptojwt.jwk.hmac import new_sym_key
 
 from cryptojwt.utils import as_unicode, as_bytes
 from cryptojwt.utils import b64e
@@ -578,6 +579,7 @@ def test_appropriate():
     assert _j1.appropriate_for('sign')
     assert _j1.appropriate_for('encrypt') is False
 
+
 def test_thumbprint_ec():
     jwk = key_from_jwk_dict({
         "kty": "EC",
@@ -588,6 +590,7 @@ def test_thumbprint_ec():
     thumbprint = "RCWR9g8NPt9iZeq-lh-qXbiFxXcU0_o1YLitDj3kpg0"
     assert (jwk.thumbprint('SHA-256').decode()) == thumbprint
 
+
 def test_thumbprint_rsa():
     jwk = key_from_jwk_dict({
         "kty": "RSA",
@@ -596,3 +599,11 @@ def test_thumbprint_rsa():
     })
     thumbprint = "Q1wZMrouq_iCnG7mr2y03Zxf7iE9mie-y_Mfh9-Cgk0"
     assert (jwk.thumbprint('SHA-256').decode()) == thumbprint
+
+
+def test_mint_new_sym_key():
+    key = new_sym_key(bytes=24, use='sig', kid='one')
+    assert key
+    assert key.use == 'sig'
+    assert key.kid == 'one'
+    assert len(key.key) == 24

tests/test_03_key_bundle.py

@@ -564,6 +564,7 @@ def test_dump_jwk():
     assert key.kty == 'RSA'
     assert isinstance(key.public_key(), rsa.RSAPublicKey)
 
+
 JWKS_DICT = {"keys": [
     {
         "n": u"zkpUgEgXICI54blf6iWiD2RbMDCOO1jV0VSff1MFFnujM4othfMsad7H1kRo50YM5S_X9TdvrpdOfpz5aBaKFhT6Ziv0nhtcekq1eRl8mjBlvGKCE5XGk-0LFSDwvqgkJoFYInq7bu0a4JEzKs5AyJY75YlGh879k1Uu2Sv3ZZOunfV1O1Orta-NvS-aG_jN5cstVbCGWE20H0vFVrJKNx0Zf-u-aA-syM4uX7wdWgQ-owoEMHge0GmGgzso2lwOYf_4znanLwEuO3p5aabEaFoKNR4K6GjQcjBcYmDEE4CtfRU9AEmhcD1kleiTB9TjPWkgDmT9MXsGxBHf3AKT5w",
@@ -602,7 +603,8 @@ def test_keys():
     b'iA7PvG_DfJIeeqQcuXFmvUGjqBkda8In_uMpZrcodVA',
     b'kLsuyGef1kfw5-t-N9CJLIHx_dpZ79-KemwqjwdrvTI',
     b'8w34j9PLyCVC7VOZZb1tFVf0MOa2KZoy87lICMeD5w8',
-    b'nKzalL5pJOtVAdCtBAU8giNRNimE-XbylWZ4vq6ZlF8'
+    b'nKzalL5pJOtVAdCtBAU8giNRNimE-XbylWZ4vq6ZlF8',
+    b'akXzyGlXg8yLhsCczKb_r8VERLx7-iZBUMIVgg2K7p4'
 ]
 
 
@@ -656,6 +658,13 @@ def test_jwks_url():
     ]
 
 
+KEYSPEC_6 = [
+    {"type": "oct", "bytes": "24", "use": ["enc"], 'kid':'code'},
+    {"type": "oct", "bytes": "24", "use": ["enc"], 'kid': 'token'},
+    {"type": "oct", "bytes": "24", "use": ["enc"], 'kid': 'refresh_token'}
+    ]
+
+
 def test_key_diff_none():
     _kb = build_key_bundle(key_conf=KEYSPEC)
 
@@ -764,3 +773,12 @@ def test_key_rollover():
 
     assert len(kb_1.get(only_active=False)) == 4
     assert len(kb_1.get()) == 2
+
+
+def test_build_key_bundle_sym():
+    _kb = build_key_bundle(key_conf=KEYSPEC_6)
+    assert len(_kb) == 3
+
+    assert len(_kb.get('RSA')) == 0
+    assert len(_kb.get('EC')) == 0
+    assert len(_kb.get('OCT')) == 3