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())