def test_add_periodic_task(): signature = pretend.stub() task_obj = pretend.stub(s=lambda: signature) celery_app = pretend.stub( add_periodic_task=pretend.call_recorder(lambda *a, **k: None)) actions = [] config = pretend.stub( action=pretend.call_recorder(lambda d, f, order: actions.append(f)), registry={"celery.app": celery_app}, task=pretend.call_recorder(lambda t: task_obj), ) schedule = pretend.stub() func = pretend.stub() tasks._add_periodic_task(config, schedule, func) for action in actions: action() assert config.action.calls == [pretend.call(None, mock.ANY, order=100)] assert config.task.calls == [pretend.call(func)] assert celery_app.add_periodic_task.calls == [ pretend.call(schedule, signature, args=(), kwargs=(), name=None) ]
def test_add_periodic_task(): signature = pretend.stub() task_obj = pretend.stub(s=lambda: signature) celery_app = pretend.stub( add_periodic_task=pretend.call_recorder(lambda *a, **k: None), ) actions = [] config = pretend.stub( action=pretend.call_recorder(lambda d, f, order: actions.append(f)), registry={"celery.app": celery_app}, task=pretend.call_recorder(lambda t: task_obj), ) schedule = pretend.stub() func = pretend.stub() tasks._add_periodic_task(config, schedule, func) for action in actions: action() assert config.action.calls == [pretend.call(None, mock.ANY, order=100)] assert config.task.calls == [pretend.call(func)] assert celery_app.add_periodic_task.calls == [ pretend.call(schedule, signature, args=(), kwargs=(), name=None), ]