class test_RPCBackend:
def setup(self):
self.b = RPCBackend(app=self.app)
def test_oid(self):
oid = self.b.oid
oid2 = self.b.oid
assert oid == oid2
assert oid == self.app.oid
def test_interface(self):
self.b.on_reply_declare('task_id')
def test_destination_for(self):
req = Mock(name='request')
req.reply_to = 'reply_to'
req.correlation_id = 'corid'
assert self.b.destination_for('task_id', req) == ('reply_to', 'corid')
task = Mock()
_task_stack.push(task)
try:
task.request.reply_to = 'reply_to'
task.request.correlation_id = 'corid'
assert self.b.destination_for('task_id', None) == (
'reply_to',
'corid',
)
finally:
_task_stack.pop()
with pytest.raises(RuntimeError):
self.b.destination_for('task_id', None)
def test_rkey(self):
assert self.b.rkey('id1') == 'id1'
def test_binding(self):
queue = self.b.binding
assert queue.name == self.b.oid
assert queue.exchange == self.b.exchange
assert queue.routing_key == self.b.oid
assert not queue.durable
assert queue.auto_delete
def test_create_binding(self):
assert self.b._create_binding('id') == self.b.binding
def test_on_task_call(self):
with patch('celery.backends.rpc.maybe_declare') as md:
with self.app.amqp.producer_pool.acquire() as prod:
self.b.on_task_call(prod, 'task_id'),
md.assert_called_with(
self.b.binding(prod.channel),
retry=True,
)
def test_create_exchange(self):
ex = self.b._create_exchange('name')
assert isinstance(ex, self.b.Exchange)
assert ex.name == ''
class test_RPCBackend:
def setup(self):
self.b = RPCBackend(app=self.app)
def test_oid(self):
oid = self.b.oid
oid2 = self.b.oid
assert oid == oid2
assert oid == self.app.oid
def test_interface(self):
self.b.on_reply_declare('task_id')
def test_destination_for(self):
req = Mock(name='request')
req.reply_to = 'reply_to'
req.correlation_id = 'corid'
assert self.b.destination_for('task_id', req) == ('reply_to', 'corid')
task = Mock()
_task_stack.push(task)
try:
task.request.reply_to = 'reply_to'
task.request.correlation_id = 'corid'
assert self.b.destination_for('task_id', None) == (
'reply_to', 'corid',
)
finally:
_task_stack.pop()
with pytest.raises(RuntimeError):
self.b.destination_for('task_id', None)
def test_rkey(self):
assert self.b.rkey('id1') == 'id1'
def test_binding(self):
queue = self.b.binding
assert queue.name == self.b.oid
assert queue.exchange == self.b.exchange
assert queue.routing_key == self.b.oid
assert not queue.durable
assert queue.auto_delete
def test_create_binding(self):
assert self.b._create_binding('id') == self.b.binding
def test_on_task_call(self):
with patch('celery.backends.rpc.maybe_declare') as md:
with self.app.amqp.producer_pool.acquire() as prod:
self.b.on_task_call(prod, 'task_id'),
md.assert_called_with(
self.b.binding(prod.channel),
retry=True,
)
def test_create_exchange(self):
ex = self.b._create_exchange('name')
assert isinstance(ex, self.b.Exchange)
assert ex.name == ''
class test_RPCBackend(AppCase):
def setup(self):
self.b = RPCBackend(app=self.app)
def test_oid(self):
oid = self.b.oid
oid2 = self.b.oid
self.assertEqual(oid, oid2)
self.assertEqual(oid, self.app.oid)
def test_interface(self):
self.b.on_reply_declare('task_id')
def test_destination_for(self):
req = Mock(name='request')
req.reply_to = 'reply_to'
req.correlation_id = 'corid'
self.assertTupleEqual(
self.b.destination_for('task_id', req),
('reply_to', 'corid'),
)
task = Mock()
_task_stack.push(task)
try:
task.request.reply_to = 'reply_to'
task.request.correlation_id = 'corid'
self.assertTupleEqual(
self.b.destination_for('task_id', None),
('reply_to', 'corid'),
)
finally:
_task_stack.pop()
with self.assertRaises(RuntimeError):
self.b.destination_for('task_id', None)
def test_rkey(self):
self.assertEqual(self.b.rkey('id1'), 'id1')
def test_binding(self):
queue = self.b.binding
self.assertEqual(queue.name, self.b.oid)
self.assertEqual(queue.exchange, self.b.exchange)
self.assertEqual(queue.routing_key, self.b.oid)
self.assertFalse(queue.durable)
self.assertTrue(queue.auto_delete)
def test_many_bindings(self):
self.assertListEqual(
self.b._many_bindings(['a', 'b']),
[self.b.binding],
)
def test_create_binding(self):
self.assertEqual(self.b._create_binding('id'), self.b.binding)
def test_on_task_call(self):
with patch('celery.backends.rpc.maybe_declare') as md:
with self.app.amqp.producer_pool.acquire() as prod:
self.b.on_task_call(prod, 'task_id'),
md.assert_called_with(
self.b.binding(prod.channel),
retry=True,
)
def test_create_exchange(self):
ex = self.b._create_exchange('name')
self.assertIsInstance(ex, self.b.Exchange)
self.assertEqual(ex.name, '')
class test_RPCBackend(AppCase):
def setup(self):
self.b = RPCBackend(app=self.app)
def test_oid(self):
oid = self.b.oid
oid2 = self.b.oid
self.assertEqual(oid, oid2)
self.assertEqual(oid, self.app.oid)
def test_interface(self):
self.b.on_reply_declare('task_id')
def test_destination_for(self):
req = Mock(name='request')
req.reply_to = 'reply_to'
req.correlation_id = 'corid'
self.assertTupleEqual(
self.b.destination_for('task_id', req),
('reply_to', 'corid'),
)
task = Mock()
_task_stack.push(task)
try:
task.request.reply_to = 'reply_to'
task.request.correlation_id = 'corid'
self.assertTupleEqual(
self.b.destination_for('task_id', None),
('reply_to', 'corid'),
)
finally:
_task_stack.pop()
with self.assertRaises(RuntimeError):
self.b.destination_for('task_id', None)
def test_rkey(self):
self.assertEqual(self.b.rkey('id1'), 'id1')
def test_binding(self):
queue = self.b.binding
self.assertEqual(queue.name, self.b.oid)
self.assertEqual(queue.exchange, self.b.exchange)
self.assertEqual(queue.routing_key, self.b.oid)
self.assertFalse(queue.durable)
self.assertTrue(queue.auto_delete)
def test_many_bindings(self):
self.assertListEqual(
self.b._many_bindings(['a', 'b']),
[self.b.binding],
)
def test_create_binding(self):
self.assertEqual(self.b._create_binding('id'), self.b.binding)
def test_on_task_call(self):
with patch('celery.backends.rpc.maybe_declare') as md:
with self.app.amqp.producer_pool.acquire() as prod:
self.b.on_task_call(prod, 'task_id'),
md.assert_called_with(
self.b.binding(prod.channel),
retry=True,
)
def test_create_exchange(self):
ex = self.b._create_exchange('name')
self.assertIsInstance(ex, self.b.Exchange)
self.assertEqual(ex.name, '')
