def test_file_field(self):
customer = Customer(**CUSTOMER__KWARGS)
customer.save()
avatar = Avatar()
avatar.customer = customer
avatar.image.save('test_file_secret_data',
ContentFile('Super secret data'),
save=False)
avatar.save()
avatar_2: Avatar = Avatar.objects.last()
assert_equal(avatar_2.image.read(), b'Super secret data')
avatar_2._anonymize_obj(base_encryption_key='LoremIpsumDolorSitAmet')
avatar_3: Avatar = Avatar.objects.last()
assert_not_equal(avatar_3.image.read(), b'Super secret data')
# Cleanup
avatar_3.image.delete()
avatar_3.delete()
def test_file_field_real_file(self):
anonymizer = anonymizer_register[Avatar]
anonymizer.image.replacement_file = 'test_file'
customer = Customer(**CUSTOMER__KWARGS)
customer.save()
avatar = Avatar()
avatar.customer = customer
avatar.image.save('test_file_real', ContentFile('Super secret data'))
avatar_2: Avatar = Avatar.objects.last()
assert_equal(avatar_2.image.read(), b'Super secret data')
avatar_2._anonymize_obj(base_encryption_key='LoremIpsumDolorSitAmet')
avatar_3: Avatar = Avatar.objects.last()
assert_not_equal(avatar_3.image.read(), b'Super secret data')
anonymizer.image.replacement_file = None
# Cleanup
avatar_3.image.delete()
avatar_3.delete()
def setUpTestData(cls):
cls.customer: Customer = Customer(**CUSTOMER__KWARGS)
cls.customer.save()
def setUpTestData(cls):
cls.customer: Customer = Customer(**CUSTOMER__KWARGS)
cls.customer.save()
cls.base_encryption_key = 'LoremIpsum'