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'