def test_run_task_not_found(manager):
row = postgres.TaskRow(id=16,
args={
"a": 9,
"b": 3
},
targeted_object="sherlock",
task_type="job")
worker = task_worker.Worker(manager, "yay")
with pytest.raises(exceptions.TaskNotFound):
worker.run_task(row)
def test_run_task_error(manager):
def job(task_run, a, b): # pylint: disable=unused-argument
raise ValueError("nope")
task = tasks.Task(manager=manager, queue="yay", name="job")
task.func = job
manager.tasks = {"job": task}
row = postgres.TaskRow(id=16,
args={
"a": 9,
"b": 3
},
targeted_object="sherlock",
task_type="job")
worker = task_worker.Worker(manager, "yay")
with pytest.raises(exceptions.TaskError):
worker.run_task(row)
def test_process_tasks(mocker, manager):
row1, row2, row3 = mocker.Mock(id=42), mocker.Mock(id=43), mocker.Mock(
id=44)
mocker.patch("cabbage.postgres.get_tasks", return_value=[row1, row2, row3])
worker = task_worker.Worker(manager, "queue")
i = 0
def side_effect(task_row):
nonlocal i
i += 1
if i == 1:
# First time the task runs
return None
elif i == 2:
# Then the task fails
raise exceptions.TaskError()
else:
# While the third task runs, a stop signal is received
worker.stop(None, None)
call_task = mocker.patch("cabbage.task_worker.Worker.run_task",
side_effect=side_effect)
finish_task = mocker.patch("cabbage.postgres.finish_task")
worker.process_tasks()
assert call_task.call_count == 3
assert finish_task.call_count == 3
assert call_task.call_args_list == [
mocker.call(task_row=row1),
mocker.call(task_row=row2),
mocker.call(task_row=row3),
]
assert finish_task.call_args_list == [
mocker.call(manager.connection, 42, "done"),
mocker.call(manager.connection, 43, "error"),
mocker.call(manager.connection, 44, "done"),
]
def test_run_task(manager):
result = []
def job(a, b): # pylint: disable=unused-argument
result.append(a + b)
task = tasks.Task(job, manager=manager, queue="yay", name="job")
manager.tasks = {"job": task}
row = postgres.TaskRow(id=16,
args={
"a": 9,
"b": 3
},
targeted_object="sherlock",
task_type="job")
worker = task_worker.Worker(manager, "yay")
worker.run_task(row)
assert result == [12]