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)
def test_task_invalidtaskname_notfound(self): self.request.matchdict = {'script': 'plotblablabla'} views = Views(self.request) with self.assertRaises(HTTPNotFound): views.task()
def test_task(self): self.request.matchdict = {'script': 'plot'} views = Views(self.request) views.celery.tasks = {'plot': 'task1'} self.assertEqual(views.task(), 'task1')