def test_send_event(self):
mytask = self.mytask._get_current_object()
mytask.app.events = Mock(name='events')
mytask.app.events.attach_mock(ContextMock(), 'default_dispatcher')
mytask.request.id = 'fb'
mytask.send_event('task-foo', id=3122)
mytask.app.events.default_dispatcher().send.assert_called_with(
'task-foo', uuid='fb', id=3122)
def setup(self):
self.maybe_signature = self.patch('celery.canvas.maybe_signature')
self.maybe_signature.side_effect = pass1
self.app.producer_or_acquire = Mock()
self.app.producer_or_acquire.attach_mock(ContextMock(), 'return_value')
self.app.conf.task_always_eager = True
self.task = builtins.add_group_task(self.app)
super(test_group, self).setup()
def assert_sent_with_ids(self, task, rid, pid, **options):
self.app.amqp.send_task_message = Mock(name='send_task_message')
self.app.backend = Mock()
self.app.producer_or_acquire = ContextMock()
task.apply_async(**options)
self.assertTrue(self.app.amqp.send_task_message.called)
message = self.app.amqp.send_task_message.call_args[0][2]
self.assertEqual(message.headers['parent_id'], pid)
self.assertEqual(message.headers['root_id'], rid)
def test_send_task__connection_provided(self):
connection = Mock(name='connection')
router = Mock(name='router')
router.route.return_value = {}
self.app.amqp = Mock(name='amqp')
self.app.amqp.Producer.attach_mock(ContextMock(), 'return_value')
self.app.send_task('foo', (1, 2), connection=connection, router=router)
self.app.amqp.Producer.assert_called_with(connection)
self.app.amqp.send_task_message.assert_called_with(
self.app.amqp.Producer(), 'foo',
self.app.amqp.create_task_message())
def test_on_chord_part_return__other_error(self):
with self.chord_context(1) as (_, request, callback):
self.b.client.pipeline = ContextMock()
raise_on_second_call(self.b.client.pipeline, RuntimeError())
self.b.client.pipeline.return_value.rpush().llen().get().expire(
).expire().execute.return_value = (1, 1, 0, 4, 5)
task = self.app._tasks['add'] = Mock(name='add_task')
self.b.on_chord_part_return(request, states.SUCCESS, 10)
task.backend.fail_from_current_stack.assert_called_with(
callback.id, exc=ANY,
)
def _amqp_connection():
connection = ContextMock(name='Connection')
connection.return_value = ContextMock(name='connection')
connection.return_value.transport.driver_type = 'amqp'
return connection