def include_router_with_task(zeebe_worker: ZeebeWorker,
router: ZeebeTaskRouter,
task_type: str = None) -> Task:
task_type = task_type or str(uuid4())
@router.task(task_type)
def dummy_function():
return {}
zeebe_worker.include_router(router)
return zeebe_worker.get_task(task_type)
def test_variables_to_fetch_match_function_parameters(
self, zeebe_worker: ZeebeWorker, task_type: str):
expected_variables_to_fetch = ["x"]
@zeebe_worker.task(task_type)
def dummy_function(x):
pass
assert zeebe_worker.get_task(
task_type).config.variables_to_fetch == expected_variables_to_fetch
def test_include_router_adds_task(self, zeebe_worker: ZeebeWorker,
router: ZeebeTaskRouter, task_type: str):
self.include_router_with_task(zeebe_worker, router, task_type)
assert zeebe_worker.get_task(task_type) is not None
def test_add_constructor_after_decorator(
self, aio_grpc_channel: grpc.aio.Channel,
decorator: TaskDecorator):
zeebe_worker = ZeebeWorker(aio_grpc_channel, after=[decorator])
assert len(zeebe_worker._after) == 1
assert decorator in zeebe_worker._after
def test_add_after_decorator(self, zeebe_worker: ZeebeWorker,
decorator: TaskDecorator):
zeebe_worker.after(decorator)
assert len(zeebe_worker._after) == 1
assert decorator in zeebe_worker._after
def test_add_before_decorator(self, zeebe_worker: ZeebeWorker,
decorator: TaskDecorator):
zeebe_worker.before(decorator)
assert len(zeebe_worker._before) == 1
assert decorator in zeebe_worker._before
def test_task_type_saved(self, zeebe_worker: ZeebeWorker, task: Task):
zeebe_worker._add_task(task)
assert zeebe_worker.get_task(task.type).type == task.type
def test_raises_on_duplicate(self, zeebe_worker: ZeebeWorker, task: Task):
zeebe_worker._add_task(task)
with pytest.raises(DuplicateTaskTypeError):
zeebe_worker._add_task(task)
def test_add_task(self, zeebe_worker: ZeebeWorker, task: Task):
zeebe_worker._add_task(task)
assert zeebe_worker.get_task(task.type) == task
def test_add_constructor_after_decorator(self, decorator):
zeebe_worker = ZeebeWorker(after=[decorator])
assert len(zeebe_worker._after) == 1
assert decorator in zeebe_worker._after
def test_add_constructor_before_decorator(self, decorator):
zeebe_worker = ZeebeWorker(before=[decorator])
assert len(zeebe_worker._before) == 1
assert decorator in zeebe_worker._before