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