def test_celery_failure(self):
silo = factories.Silo(owner=self.user, public=False)
read_type = factories.ReadType(read_type="CSV")
upload_file = open('silo/tests/sample_data/test_broken.csv', 'rb')
read = factories.Read(owner=self.user, type=read_type, file_data=SimpleUploadedFile(upload_file.name, upload_file.read()))
task = factories.CeleryTask(content_object=read)
process_silo(silo.id, read.id)
ctask = CeleryTask.objects.get(
object_id=read.id,
content_type=ContentType.objects.get_for_model(Read)
)
self.assertEqual(ctask.task_id, task.task_id)
self.assertEqual(ctask.task_status, CeleryTask.TASK_FAILED)
def test_celery_success(self):
"""
Test if the celery task process_silo actually imports data
:return:
"""
silo = factories.Silo(owner=self.user, public=False)
read_type = factories.ReadType(read_type="CSV")
upload_file = open('silo/tests/sample_data/test.csv', 'rb')
read = factories.Read(owner=self.user, type=read_type, file_data=SimpleUploadedFile(upload_file.name, upload_file.read()))
task = factories.CeleryTask(task_status=CeleryTask.TASK_CREATED, content_object=read)
process_done = process_silo(silo.id, read.id)
self.assertEqual(getSiloColumnNames(silo.id), ['First_Name', 'Last_Name', 'E-mail'])
self.assertTrue(process_done)
def test_celery_wrong_silo(self, process_silo_retry):
process_silo_retry.side_effect = Retry()
silo = factories.Silo(owner=self.user, public=False)
with self.assertRaises(ObjectDoesNotExist):
process_silo(silo.id, -1)
