def test_encrypt_file_named(self): msg = b'My private message' pw = b'my private key' pubkey = '8W;>i^H0qi|J&$coR5MFpR*Vn' in_file = tempfile.NamedTemporaryFile() in_file.write(msg) in_file.flush() in_file.seek(0) encrypted_file = tempfile.NamedTemporaryFile() decrypted_file = tempfile.NamedTemporaryFile(delete=False) seccure.encrypt_file(in_file.name, encrypted_file.name, pubkey) seccure.decrypt_file(encrypted_file.name, decrypted_file.name, pw) self.assertEqual(msg, decrypted_file.read())
def test_encrypt_file(self): msg = b'My private message' pw = b'my private key' pubkey = '8W;>i^H0qi|J&$coR5MFpR*Vn' encrypted_file = tempfile.NamedTemporaryFile() decrypted_file = tempfile.NamedTemporaryFile(delete=False) with open(decrypted_file.name, 'wb') as fo: fo.write(msg) with open(decrypted_file.name, 'rb') as fi: with open(encrypted_file.name, 'wb') as fo: seccure.encrypt_file(fi, fo, pubkey) with open(decrypted_file.name, 'wb') as fo: with open(encrypted_file.name, 'rb') as fi: seccure.decrypt_file(fi, fo, pw) with open(decrypted_file.name, 'rb') as fi: self.assertEqual(msg, fi.read())