def test_delete_file(self):
self.mock.StubOutWithMock(FieldFile, 'delete')
FieldFile.delete()
item = PodcastItem(
file=SimpleUploadedFile('/ep123.txt', 'file contents'))
self.mock.ReplayAll()
item.delete_file()
self.mock.VerifyAll()
def test_delete_file(self):
self.mock.StubOutWithMock(FieldFile, 'delete')
FieldFile.delete()
item = PodcastItem(
file=SimpleUploadedFile('/ep123.txt', 'file contents'))
self.mock.ReplayAll()
item.delete_file()
self.mock.VerifyAll()
def test_generic(self):
field = FileField(name='test', upload_to=self.upload_to)
fake_model = type(str('fake'), (object,), {field.name: None})
_file = FieldFile(field=field, instance=fake_model(), name='')
_file._committed = False
with open(self.archive, 'rb') as f:
_file.save('test.zip', ContentFile(f.read()), save=False)
self.assertEqual(_file.name, 'archive/da39a3ee5e6b4b0d3255bfef95601890afd80709.zip')
_file.delete(save=False)
def test_generic(self):
field = FileField(name='test', upload_to=self.upload_to)
fake_model = type(str('fake'), (object, ), {field.name: None})
_file = FieldFile(field=field, instance=fake_model(), name='')
_file._committed = False
with open(self.archive, 'rb') as f:
_file.save('test.zip', ContentFile(f.read()), save=False)
self.assertEqual(
_file.name, 'archive/da39a3ee5e6b4b0d3255bfef95601890afd80709.zip')
_file.delete(save=False)
def get_encrypted_value(self, value: FieldFile, encryption_key: str):
file_name = value.name
value.delete(save=False)
file = self.get_replacement_file(file_name)
if func_supports_parameter(value.storage.save, 'max_length'):
value.name = value.storage.save(file_name,
file,
max_length=value.field.max_length)
else:
# Backwards compatibility removed in Django 1.10
value.name = value.storage.save(file_name, file)
setattr(value.instance, value.field.name, value.name)
value._size = file.size # Django 1.8 + 1.9
value._committed = True
file.close()
return value
