def test_task_result_must_be_ready_but_isnt(self): self.request.matchdict['taskid'] = 'b3c84d96-4dc7-4532-a864-3573202f202a' views = Views(self.request) mresult = Mock(AsyncResult) is_ready = False mresult.ready.return_value = is_ready views.celery.AsyncResult.return_value = mresult with self.assertRaises(TaskNotReady): views.task_result(True)
def test_task_result(self): self.request.matchdict['taskid'] = 'b3c84d96-4dc7-4532-a864-3573202f202a' views = Views(self.request) mresult = Mock(AsyncResult) is_ready = True mresult.ready.return_value = is_ready views.celery.AsyncResult.return_value = mresult result = views.task_result() self.assertEqual(result, mresult)
def test_result_template(self): self.request.matchdict['script'] = 'plot' self.request.matchdict['taskid'] = 'mytaskid' views = Views(self.request) task_result = Mock(AsyncResult) views.task_result = Mock(return_value=task_result) task = PythonTask() task.render_result = Mock(return_value='mytemplate') views.task = Mock(return_value=task) files = {'result.csv': '/plot/mytaskid/result.csv', } views.result_files = Mock(return_value=files) result = views.result() eresult = {'result': task_result, 'files': files, 'task': task, 'result_html': 'mytemplate', } self.assertEqual(result, eresult)