From db1f9f4535fb8d9900aec8f31bef11f298e50f5d Mon Sep 17 00:00:00 2001 From: Simo Sorce Date: Tue, 30 May 2023 11:30:18 -0400 Subject: [PATCH] Add test to generate key with algorithm This uses an algorithm that has different input_keysize and output keysize. Signed-off-by: Simo Sorce --- jwcrypto/tests.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/jwcrypto/tests.py b/jwcrypto/tests.py index ebfbb38..b488ed7 100644 --- a/jwcrypto/tests.py +++ b/jwcrypto/tests.py @@ -431,6 +431,15 @@ def test_generate_oct_key(self): e.deserialize(enc, key) self.assertEqual(e.payload.decode('utf-8'), 'test') + # also test key generation with input_keysize != keysize + key = jwk.JWK.generate(kty='oct', alg="A128CBC-HS256") + self.assertEqual(len(base64url_decode(key['k'])), 32) + e = jwe.JWE('test', '{"alg":"A256KW","enc":"A128CBC-HS256"}') + e.add_recipient(key) + enc = e.serialize() + e.deserialize(enc, key) + self.assertEqual(e.payload.decode('utf-8'), 'test') + def test_generate_EC_key(self): # Backwards compat curve key = jwk.JWK.generate(kty='EC', curve='P-256')