class test_BaseBackend_interface:
def setup(self):
self.b = BaseBackend(self.app)
def test__forget(self):
with pytest.raises(NotImplementedError):
self.b._forget('SOMExx-N0Nex1stant-IDxx-')
def test_forget(self):
with pytest.raises(NotImplementedError):
self.b.forget('SOMExx-N0nex1stant-IDxx-')
def test_on_chord_part_return(self):
self.b.on_chord_part_return(None, None, None)
def test_apply_chord(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
self.b.apply_chord(
group(app=self.app),
(),
'dakj221',
None,
result=[self.app.AsyncResult(x) for x in [1, 2, 3]],
)
assert self.app.tasks[unlock].apply_async.call_count
class test_BaseBackend_interface:
def setup(self):
self.b = BaseBackend(self.app)
@self.app.task(shared=False)
def callback(result):
pass
self.callback = callback
def test__forget(self):
with pytest.raises(NotImplementedError):
self.b._forget('SOMExx-N0Nex1stant-IDxx-')
def test_forget(self):
with pytest.raises(NotImplementedError):
self.b.forget('SOMExx-N0nex1stant-IDxx-')
def test_on_chord_part_return(self):
self.b.on_chord_part_return(None, None, None)
def test_apply_chord(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
header_result = self.app.GroupResult(
uuid(),
[self.app.AsyncResult(x) for x in range(3)],
)
self.b.apply_chord(header_result, self.callback.s())
assert self.app.tasks[unlock].apply_async.call_count
def test_chord_unlock_queue(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
header_result = self.app.GroupResult(
uuid(),
[self.app.AsyncResult(x) for x in range(3)],
)
body = self.callback.s()
self.b.apply_chord(header_result, body)
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] is None
self.b.apply_chord(header_result, body.set(queue='test_queue'))
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'test_queue'
@self.app.task(shared=False, queue='test_queue_two')
def callback_queue(result):
pass
self.b.apply_chord(header_result, callback_queue.s())
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'test_queue_two'
class test_BaseBackend_interface:
def setup(self):
self.b = BaseBackend(self.app)
@self.app.task(shared=False)
def callback(result):
pass
self.callback = callback
def test__forget(self):
with pytest.raises(NotImplementedError):
self.b._forget('SOMExx-N0Nex1stant-IDxx-')
def test_forget(self):
with pytest.raises(NotImplementedError):
self.b.forget('SOMExx-N0nex1stant-IDxx-')
def test_on_chord_part_return(self):
self.b.on_chord_part_return(None, None, None)
def test_apply_chord(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
header_result = self.app.GroupResult(
uuid(),
[self.app.AsyncResult(x) for x in range(3)],
)
self.b.apply_chord(header_result, self.callback.s())
assert self.app.tasks[unlock].apply_async.call_count
def test_chord_unlock_queue(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
header_result = self.app.GroupResult(
uuid(),
[self.app.AsyncResult(x) for x in range(3)],
)
body = self.callback.s()
self.b.apply_chord(header_result, body)
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] is None
self.b.apply_chord(header_result, body.set(queue='test_queue'))
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'test_queue'
@self.app.task(shared=False, queue='test_queue_two')
def callback_queue(result):
pass
self.b.apply_chord(header_result, callback_queue.s())
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'test_queue_two'
class test_BaseBackend_interface(AppCase):
def setup(self):
self.b = BaseBackend(self.app)
def test__forget(self):
with self.assertRaises(NotImplementedError):
self.b._forget("SOMExx-N0Nex1stant-IDxx-")
def test_forget(self):
with self.assertRaises(NotImplementedError):
self.b.forget("SOMExx-N0nex1stant-IDxx-")
def test_on_chord_part_return(self):
self.b.on_chord_part_return(None, None, None)
def test_apply_chord(self, unlock="celery.chord_unlock"):
self.app.tasks[unlock] = Mock()
self.b.apply_chord(
group(app=self.app), (), "dakj221", None, result=[self.app.AsyncResult(x) for x in [1, 2, 3]]
)
self.assertTrue(self.app.tasks[unlock].apply_async.call_count)
class test_BaseBackend_interface:
def setup(self):
self.b = BaseBackend(self.app)
def test__forget(self):
with pytest.raises(NotImplementedError):
self.b._forget('SOMExx-N0Nex1stant-IDxx-')
def test_forget(self):
with pytest.raises(NotImplementedError):
self.b.forget('SOMExx-N0nex1stant-IDxx-')
def test_on_chord_part_return(self):
self.b.on_chord_part_return(None, None, None)
def test_apply_chord(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
self.b.apply_chord(
group(app=self.app), (), 'dakj221', None,
result=[self.app.AsyncResult(x) for x in [1, 2, 3]],
)
assert self.app.tasks[unlock].apply_async.call_count
class test_BaseBackend_interface:
def setup(self):
self.b = BaseBackend(self.app)
def test__forget(self):
with pytest.raises(NotImplementedError):
self.b._forget('SOMExx-N0Nex1stant-IDxx-')
def test_forget(self):
with pytest.raises(NotImplementedError):
self.b.forget('SOMExx-N0nex1stant-IDxx-')
def test_on_chord_part_return(self):
self.b.on_chord_part_return(None, None, None)
def test_apply_chord(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
header_result = self.app.GroupResult(
uuid(),
[self.app.AsyncResult(x) for x in range(3)],
)
self.b.apply_chord(header_result, None)
assert self.app.tasks[unlock].apply_async.call_count
class test_BaseBackend_interface:
def setup(self):
self.b = BaseBackend(self.app)
@self.app.task(shared=False)
def callback(result):
pass
self.callback = callback
def test__forget(self):
with pytest.raises(NotImplementedError):
self.b._forget('SOMExx-N0Nex1stant-IDxx-')
def test_forget(self):
with pytest.raises(NotImplementedError):
self.b.forget('SOMExx-N0nex1stant-IDxx-')
def test_on_chord_part_return(self):
self.b.on_chord_part_return(None, None, None)
def test_apply_chord(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
header_result_args = (
uuid(),
[self.app.AsyncResult(x) for x in range(3)],
)
self.b.apply_chord(header_result_args, self.callback.s())
assert self.app.tasks[unlock].apply_async.call_count
def test_chord_unlock_queue(self, unlock='celery.chord_unlock'):
self.app.tasks[unlock] = Mock()
header_result_args = (
uuid(),
[self.app.AsyncResult(x) for x in range(3)],
)
body = self.callback.s()
self.b.apply_chord(header_result_args, body)
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'testcelery'
routing_queue = Mock()
routing_queue.name = "routing_queue"
self.app.amqp.router.route = Mock(
return_value={"queue": routing_queue})
self.b.apply_chord(header_result_args, body)
assert self.app.amqp.router.route.call_args[0][1] == body.name
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs["queue"] == "routing_queue"
self.b.apply_chord(header_result_args, body.set(queue='test_queue'))
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'test_queue'
@self.app.task(shared=False, queue='test_queue_two')
def callback_queue(result):
pass
self.b.apply_chord(header_result_args, callback_queue.s())
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'test_queue_two'
with self.Celery() as app2:
@app2.task(name='callback_different_app', shared=False)
def callback_different_app(result):
pass
callback_different_app_signature = self.app.signature(
'callback_different_app')
self.b.apply_chord(header_result_args,
callback_different_app_signature)
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'routing_queue'
callback_different_app_signature.set(queue='test_queue_three')
self.b.apply_chord(header_result_args,
callback_different_app_signature)
called_kwargs = self.app.tasks[unlock].apply_async.call_args[1]
assert called_kwargs['queue'] == 'test_queue_three'