def test_register_task(self): @taskgroup('testgroup') def func1(): pass def func2(): pass app = WorkerApplication() app.declare_taskgroup('testgroup', 'testexchange', 'testroutingkey') app.tasks = MagicMock() app.register_task(func1, 'funcone') app.tasks._set.assert_called_with(func1, 'funcone', exchange='testexchange', routing_key='testroutingkey') app.register_task(func2, taskgroup='testgroup') app.tasks._set.assert_called_with(func2, 'func2', exchange='testexchange', routing_key='testroutingkey')
def test_register_module(self): app = WorkerApplication() app.declare_task = MagicMock() app.register_task = MagicMock() mod = MagicMock() mod.__all__ = ['func1', 'func2'] mod.__declare_tasks__ = [('testgroup', 'testtask')] mod.func1 = MagicMock(__name__='func1') mod.func2 = MagicMock(__name__='func2') app.register_module(mod, 't_') app.register_task.assert_any_call(mod.func1, 't_func1', None) app.register_task.assert_any_call(mod.func2, 't_func2', None) app.declare_task.assert_called_with('testgroup', 'testtask')