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)