def test_declare_task(self):
app = WorkerApplication()
app.declare_taskgroup('testgroup', 'testexchange', 'testroutingkey')
app.tasks = MagicMock()
app.declare_task('testgroup', 'taskname')
app.tasks._declare.assert_called_with('taskname',
exchange='testexchange',
routing_key='testroutingkey')
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')
