def get_job():
job_id = request.args['job_id']
job = Job(job_id)
job.fetch()
if job.data["params"].get("user"):
if not g.user.is_authenticated() or (job.data["params"].get("user") != str(g.user.id)):
return "Unauthorized."
return json.dumps({k: v for k, v in job.data.iteritems() if k in ("status", "result")})
def test_cli_run_nonblocking(worker):
worker.start_deps()
job_id1 = worker.send_task_cli("tests.tasks.general.Add", {"a": 41, "b": 1}, block=False)
job1 = Job(job_id1).fetch()
job1.wait(poll_interval=0.01)
job1.fetch()
assert job1.data["status"] == "success"
assert job1.data["result"] == 42
def get_job():
job_id = request.args['job_id']
job = Job(job_id)
job.fetch()
if job.data["params"].get("user"):
if not g.user.is_authenticated() or (job.data["params"].get("user") !=
str(g.user.id)):
return "Unauthorized."
return json.dumps(
{k: v
for k, v in job.data.iteritems() if k in ("status", "result")})
def test_cli_run_nonblocking(worker):
worker.start()
job_id1 = worker.send_task_cli(
"tests.tasks.general.Add", {"a": 41, "b": 1}, queue="default")
job1 = Job(job_id1).fetch()
job1.wait(poll_interval=0.01)
job1.fetch()
assert job1.data["status"] == "success"
assert job1.data["result"] == 42
def test_cancel_by_worker(worker):
from bson import ObjectId
job_id = worker.send_task("tests.tasks.general.Add", {
"a": 41,
"b": 1
},
queue="default",
block=False)
job = Job(job_id)
job.wait(poll_interval=0.01)
job_data = job.fetch().data
worker.send_task("mrq.basetasks.utils.JobAction", {
"action": "cancel",
"worker": str(job_data["worker"])
},
block=True)
job_data = job.fetch().data
assert job_data["status"] == "cancel"
