def test_enrypt_decrypt_fail_for_invalid_file_type(self):
self.assertRaises(ValidationError, EncryptionService().encrypt_file, self.invalid_file, self.password)
self.assertRaises(ValidationError, EncryptionService().decrypt_file, self.invalid_file, self.password)
def test_encrypt_decrypt_return_false_for_raise_exception_false_for_invalid_input(self):
self.assertEqual(False, EncryptionService(raise_exception=False).encrypt_file(
self.invalid_file, self.password))
self.assertEqual(False, EncryptionService(raise_exception=False).decrypt_file(
self.invalid_file, self.password))
def test_encrypt_decrypt_fail_without_filename(self):
self.assertRaises(ValidationError, EncryptionService().encrypt_file, None, self.password)
self.assertRaises(ValidationError, EncryptionService().decrypt_file, None, self.password)
def test_encrypt_decrypt_fail_without_password(self):
file_name = copy.deepcopy(self.file)
self.assertRaises(ValidationError, EncryptionService().encrypt_file, file_name, None)
self.assertRaises(ValidationError, EncryptionService().decrypt_file, file_name, None)
def test_decrypt_with_wrong_password_fails(self):
file_name = copy.deepcopy(self.file)
encrypted_file = EncryptionService().encrypt_file(file_name, self.password, extension='enc')
self.assertNotEqual(file_name.read(), encrypted_file.read())
decrypted_file = EncryptionService().decrypt_file(encrypted_file, self.wrong_password, extension='enc')
self.assertNotEquals(self.file.read(), decrypted_file.read())
def test_encryption_decryption_roundup(self):
file_name = copy.deepcopy(self.file)
encrypted_file = EncryptionService().encrypt_file(file_name, self.password)
self.assertNotEqual(file_name.read(), encrypted_file.read())
decrypted_file = EncryptionService().decrypt_file(encrypted_file, self.password)
self.assertEqual(self.file.read(), decrypted_file.read())
from djangoencryptfile import EncryptionService
from django.core.files import File
password = '******'
service = EncryptionService(raise_exception=False)
open('code for crypto', 'rb') as inputfile:
usefile = File(inputfile, name='readme.md')
encrypted_file = service.encrypt_file(useFile, password, extension='enc') # it will save readme.md.enc
decrypt_file = service.decrypt_file(encrypted_file, password, extension='.enc') # it will remove .enc extension