def test_encryption(self):
"""Verify that fields *are* encrypted when GNUPG_HOME is set"""
with self.settings(GNUPG_HOME=os.path.join('peacecorps', 'tests',
'gpg'),
GPG_RECIPIENTS={
'peacecorps.DonorInfo.xml': 'C68F6B22'}):
di = DonorInfo(agency_tracking_id='TRACK', account=self.account,
xml='Plain Text')
di.save()
# Was *not* saved in plain text in the DB
values = DonorInfo.objects.filter(pk=di.pk).values_list('xml')
byte_str = values[0][0]
self.assertTrue('BEGIN PGP' in byte_str.decode('utf-8'))
# Decodes correctly
from_db = DonorInfo.objects.get(pk=di.pk)
self.assertEqual(from_db.xml, 'Plain Text')
def test_no_encryption(self):
"""With no encryption settings, fields still work"""
with self.settings(GNUPG_HOME=''):
di = DonorInfo(agency_tracking_id='TRACK', account=self.account,
xml='Plain Text')
di.save()
# Was saved in plain text in the DB
values = DonorInfo.objects.filter(pk=di.pk).values_list('xml')
byte_str = values[0][0]
if isinstance(byte_str, memoryview):
byte_str = byte_str.tobytes()
self.assertEqual(byte_str.decode('utf-8'), 'Plain Text')
# Decodes correctly
from_db = DonorInfo.objects.get(pk=di.pk)
self.assertEqual(from_db.xml, 'Plain Text')
