def test_jwkset(self): ''' nose2 jose.tests.test_jwk.TestJwk.test_jwkset ''' jwkset = JwkSet() jwkset.keys.append(Jwk(kid='kidRsa', kty=keys.KeyTypeEnum.RSA)) jwkset.keys.append(Jwk(kid='kidEc', kty=keys.KeyTypeEnum.EC)) jwkset.keys.append(Jwk(kid='kidOct', kty=keys.KeyTypeEnum.OCT)) self.assertEqual(jwkset.get_key(kty=KeyTypeEnum.RSA).kid, 'kidRsa') self.assertEqual(jwkset.get_key(kty=KeyTypeEnum.EC).kid, 'kidEc') self.assertEqual(jwkset.get_key(kty=KeyTypeEnum.OCT).kid, 'kidOct')
class TestEntity(KeyOwner): def __init__(self, identifier, jku): self.identifier = identifier self.jku = jku self.jwkset = JwkSet( keys=[ Jwk.generate(KeyTypeEnum.RSA), Jwk.generate(KeyTypeEnum.EC), Jwk.generate(KeyTypeEnum.OCT), ] ) def get_key(self, crypto, *args, **kwargs): return self.jwkset.get_key( crypto.key_type, kid=crypto.kid )