def test_png_container(self):
doc = uploaded_new_document(self.upload)
doc.title = 'File uploaded'
doc.author = self.u1
doc.save()
self.upload.name = 'fake.png'
derived = uploaded_new_derived_document(self.upload)
derived.derived_from = doc._blob
derived.index = 0
derived.save()
def test_document_upload(self):
"""
Tests that we can upload a file and get it back
"""
doc = uploaded_new_document(self.upload)
doc.title = 'File uploaded'
doc.author = self.u1
doc.file.read()
doc.save()
doc = Document.objects.get(title='File uploaded')
self.upload.seek(0)
self.assertEqual(doc.file.read(), self.upload.read())
def test_files_are_deleted(self):
"""Make sure file is deleted when blob is"""
doc = uploaded_new_document(self.upload)
doc.title = 'File uploaded'
doc.author = self.u1
doc.save()
name = doc._blob.file.name
from django.core.files.storage import get_storage_class
storage = get_storage_class()()
doc.delete()
self.assertFalse(storage.exists(name))
def test_deleted_derived(self):
"""
Make sure that derived documents are deleted when parent document and
blob are deleted.
"""
doc = uploaded_new_document(self.upload)
doc.title = 'File uploaded'
doc.author = self.u1
doc.save()
self.upload.name = 'fake.png'
derived = uploaded_new_derived_document(self.upload)
derived.derived_from = doc._blob
derived.index = 0
derived.save()
Document.objects.all().delete()
self.assertEqual(DerivedBlob.objects.count(), 0)
def test_orphaned_blobs(self):
"""Make sure that deleting a Document does not result in orphanded blobs"""
doc = uploaded_new_document(self.upload)
doc.title = 'File uploaded'
doc.author = self.u1
doc.save()
self.upload.name = 'fake.png'
derived = uploaded_new_derived_document(self.upload)
derived.derived_from = doc._blob
derived.index = 0
derived.save()
DerivedDocument.objects.all().delete()
self.assertEqual(DerivedBlob.objects.count(), 0)
Document.objects.all().delete()
self.assertEqual(ParentBlob.objects.count(), 0)
def test_derived_document_upload(self):
"""
Test derived file upload
"""
doc = uploaded_new_document(self.upload)
doc.title = 'File uploaded'
doc.author = self.u1
doc.save()
derived = uploaded_new_derived_document(self.upload2)
derived.derived_from = doc._blob
derived.index = 0
derived.save()
derived2 = doc.get_derived_documents_of_type('pdf')[0]
self.assertEqual(derived, derived2)
self.upload2.seek(0)
self.assertEqual(derived2.file.read(), self.upload2.read())
