def test_KeyModel(self):
"""Test that adding a well-formed Key works."""
self.assertEqual(models.Key.objects.count(), 0)
key_1 = models.Key(name="testkey1",
data="abc123",
algorithm="MD5")
key_1.save()
self.assertEqual(models.Key.objects.count(), 1)
def test_FernetKeyDecryptionFailure(self):
"""Test encrypt/decryption when Fernet key changes."""
original_tsig_key = 'oGyDayyZ2mDUiJCuTUODnA=='
key_1 = models.Key(name='testencryptedkey1',
data=original_tsig_key,
algorithm='MD5')
key_1.save()
new_fkey = Fernet(Fernet.generate_key())
with self.assertRaises(InvalidToken):
decrypted_tsig_key = new_fkey.decrypt(bytes(key_1.data))
def test_FernetKeyDecryptionSuccess(self):
"""Test encrypt/decryption when Fernet key is generated by Django."""
original_tsig_key = 'oGyDayyZ2mDUiJCuTUODnA=='
key_1 = models.Key(name='testencryptedkey1',
data=original_tsig_key,
algorithm='MD5')
key_1.save()
decrypt_key = Fernet(settings.FERNET_KEY)
decrypted_tsig_key = decrypt_key.decrypt(bytes(key_1.data))
self.assertEqual(original_tsig_key, decrypted_tsig_key)