def test_task_get_no_task(self, queue_mock, priority_queue_mock):
queue_mock.fetch_job.side_effect = JobNotFound()
priority_queue_mock.fetch_job.side_effect = JobNotFound()
response = self.client.get(
reverse("kolibri:core:task-detail", kwargs={"pk": "1"}),
{"task_id": "1"},
format="json",
)
self.assertEqual(response.status_code, 404)
def test_task_cancel_no_task(self, queue_mock):
queue_mock.cancel.side_effect = JobNotFound()
response = self.client.post(reverse("kolibri:core:task-canceltask"),
{"task_id": "1"},
format="json")
self.assertEqual(response.status_code, 200)
def get_job(self, job_id):
with self.session_scope() as session:
scheduled_job = session.query(ScheduledJob).get(job_id)
if scheduled_job is None:
raise JobNotFound()
return scheduled_job.obj
def _get_job_and_orm_job(self, job_id, session):
orm_job = session.query(ORMJob).filter_by(id=job_id).one_or_none()
if orm_job is None:
raise JobNotFound()
job = self._add_storage(orm_job.obj)
return job, orm_job
