def test_store_keys_wrong_passphrase(self):
key = Key()
key.create_key("rsa", 2048)
pem = key.serialize("test_store_keys_wrong_passphrase")
with self.assertRaisesMessage(ValueError,
"Bad decrypt. Incorrect password?"):
key.load(pem, "test_store_keys_passphrase")
def test_serialize_keys_passphrase(self):
key = Key()
key.create_key("rsa", 4096)
pem = key.serialize("test_store_keys_passphrase")
prvkey = key.load(pem, "test_store_keys_passphrase")
self.assertIsInstance(prvkey.key, rsa.RSAPrivateKey)
self.assertEqual(prvkey.key.key_size, 4096)
def test_store_keys_no_passphrase(self):
key = Key()
key.create_key("ed25519", None)
pem = key.serialize()
key = Key()
prvkey = key.load(pem)
self.assertIsInstance(prvkey.key, ed25519.Ed25519PrivateKey)
def test_store_keys_no_passphrase(self):
key = Key()
key.create_key("rsa", 2048)
pem = key.serialize()
key = Key()
prvkey = key.load(pem)
self.assertIsInstance(prvkey.key, rsa.RSAPrivateKey)
self.assertEqual(prvkey.key.key_size, 2048)
