def test_invalid_private_key(self):
with pytest.raises((SecurityError, TypeError)):
PrivateKey(None)
with pytest.raises(SecurityError):
PrivateKey('')
with pytest.raises(SecurityError):
PrivateKey('foo')
with pytest.raises(SecurityError):
PrivateKey(KEY1[:20] + KEY1[21:])
with pytest.raises(SecurityError):
PrivateKey(CERT1)
def register_auth(key=None,
cert=None,
store=None,
digest='sha1',
serializer='json'):
"""Register security serializer."""
s = SecureSerializer(key and PrivateKey(key),
cert and Certificate(cert),
store and FSCertStore(store),
digest=digest,
serializer=serializer)
registry.register('auth',
s.serialize,
s.deserialize,
content_type='application/data',
content_encoding='utf-8')
def _get_s(self, key, cert, certs):
store = CertStore()
for c in certs:
store.add_cert(Certificate(c))
return SecureSerializer(PrivateKey(key), Certificate(cert), store)
def test_sign(self):
pkey = PrivateKey(KEY1)
pkey.sign('test', bytes_if_py2('sha1'))
with pytest.raises(ValueError):
pkey.sign('test', bytes_if_py2('unknown'))
def test_valid_private_key(self):
PrivateKey(KEY1)
PrivateKey(KEY2)
def test_sign(self):
pkey = PrivateKey(KEY1)
pkey.sign('test', b'sha1')
with self.assertRaises(ValueError):
pkey.sign('test', b'unknown')
def test_sign(self):
pkey = PrivateKey(KEY1)
pkey.sign('test', 'sha1')
self.assertRaises(ValueError, pkey.sign, 'test', 'unknown')
def test_sign(self):
pkey = PrivateKey(KEY1)
pkey.sign('test', 'sha1')
self.assertRaises(ValueError, pkey.sign, 'test', 'unknown')
def test_sign(self):
pkey = PrivateKey(KEY1)
pkey.sign('test', b'sha1')
with pytest.raises(ValueError):
pkey.sign('test', b'unknown')
def test_sign(self):
pkey = PrivateKey(KEY1)
pkey.sign(ensure_bytes('test'), get_digest_algorithm())
with pytest.raises(AttributeError):
pkey.sign(ensure_bytes('test'), get_digest_algorithm('unknown'))
def test_valid_private_key(self):
PrivateKey(KEY1)
PrivateKey(KEY2)
PrivateKey(ENCKEY1, KEYPASSWORD)
PrivateKey(ENCKEY2, KEYPASSWORD)
def test_sign(self):
pkey = PrivateKey(KEY1)
pkey.sign(ensure_bytes('test'), get_digest_algorithm())
with pytest.raises(AttributeError):
pkey.sign(ensure_bytes('test'), get_digest_algorithm('unknown'))
