def test_gets_task(self):
task_func = pretend.stub(__name__="task_func", __module__="tests.foo")
task_obj = pretend.stub()
celery_app = pretend.stub(
gen_task_name=lambda func, module: module + "." + func,
tasks={"tests.foo.task_func": task_obj},
)
assert tasks._get_task(celery_app, task_func) is task_obj
def test_gets_task(self):
task_func = pretend.stub(__name__="task_func", __module__="tests.foo")
task_obj = pretend.stub()
celery_app = pretend.stub(
gen_task_name=lambda func, module: module + "." + func,
tasks={"tests.foo.task_func": task_obj},
)
assert tasks._get_task(celery_app, task_func) is task_obj
