def assertStatusForIs(self, status, outcome): uuid = gen_unique_id() result = gen_unique_id() default_backend.store_result(uuid, result, status) json = self.client.get(task_is_successful(task_id=uuid)) self.assertJSONEqual(json, {"task": {"id": uuid, "executed": outcome}})
def assertStatusForIs(self, status, res, traceback=None): uuid = gen_unique_id() default_backend.store_result(uuid, res, status, traceback=traceback) json = self.client.get(task_status(task_id=uuid)) expect = dict(id=uuid, status=status, result=res) if status in default_backend.EXCEPTION_STATES: instore = default_backend.get_result(uuid) self.assertEqual(str(instore.args), str(res.args)) expect["result"] = str(res.args[0]) expect["exc"] = get_full_cls_name(res.__class__) expect["traceback"] = traceback self.assertJSONEqual(json, dict(task=expect))
def assertStatusForIs(self, status, res, traceback=None): uuid = gen_unique_id() default_backend.store_result(uuid, res, status, traceback=traceback) json = self.client.get(task_status(task_id=uuid)) expect = dict(id=uuid, status=status, result=res) if status in default_backend.EXCEPTION_STATES: instore = default_backend.get_result(uuid) self.assertEqual(str(instore.args), str(res.args)) expect["result"] = repr(res) expect["exc"] = get_full_cls_name(res.__class__) expect["traceback"] = traceback self.assertJSONEqual(json, dict(task=expect))
def get_task_meta_error(exception): """ Take an exception and turn it in to a Celery result tombstone that mimics what would happen if that error were thrown during a Task run. This is copy/pasted from pstat.core.utils """ # Need to return an object that has a uuid attribute, and need to store the # result in the cache fake_result = TaskMeta() task_id = str(uuid4()) fake_result.task_id = task_id default_backend.store_result(task_id, exception, status=states.FAILURE) return fake_result
def assertStatusForIs(self, status, outcome): uuid = gen_unique_id() result = gen_unique_id() default_backend.store_result(uuid, result, status) json = self.client.get(task_is_successful(task_id=uuid)) self.assertJSONEqual(json, {"task": {"id": uuid, "executed": outcome}})