def setup_conn(self):
self.conn = MagicMock(name='connection')
self.conn.is_closing = False
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
def __init__(self):
self.response = None
self.connection = Connection()
self.channel = Channel(self.connection)
(self.queue, _, _) = self.channel.queue_declare(exclusive=True)
self.channel.queue_bind(self.queue, exchange='django_amqp_example')
class RpcClient(object):
def __init__(self):
self.response = None
self.connection = Connection()
self.channel = Channel(self.connection)
(self.queue, _, _) = self.channel.queue_declare(exclusive=True)
self.channel.queue_bind(self.queue, exchange='django_amqp_example')
def request(self, body):
message = Message(
body=json.dumps(body),
reply_to=self.queue,
content_type='application/json')
self.channel.basic_publish(
message,
exchange='django_amqp_example',
routing_key='task_queue')
print "Task submitted:", json.dumps(body)
def callback(msg):
self.response = json.loads(msg.body)
self.channel.basic_consume(
callback=callback,
queue=self.queue,
no_ack=True)
while True:
self.connection.drain_events(timeout=60)
if self.response is not None:
break
self.connection.close()
return self.response
class test_Channel:
@pytest.fixture(autouse=True)
def setup_conn(self):
self.conn = MagicMock(name='connection')
self.conn.is_closing = False
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
assert c.basic_publish == c.basic_publish_confirm
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
assert c.basic_publish == c._basic_publish
def test_init_auto_channel(self):
c = Channel(self.conn, None)
self.conn._get_free_channel_id.assert_called_with()
assert c.channel_id is self.conn._get_free_channel_id()
def test_init_explicit_channel(self):
Channel(self.conn, 3)
self.conn._claim_channel_id.assert_called_with(3)
def test_then(self):
self.c.on_open = Mock(name='on_open')
on_success = Mock(name='on_success')
on_error = Mock(name='on_error')
self.c.then(on_success, on_error)
self.c.on_open.then.assert_called_with(on_success, on_error)
def test_collect(self):
self.c.callbacks[(50, 61)] = Mock()
self.c.cancel_callbacks['foo'] = Mock()
self.c.events['bar'].add(Mock())
self.c.no_ack_consumers.add('foo')
self.c.collect()
assert not self.c.callbacks
assert not self.c.cancel_callbacks
assert not self.c.events
assert not self.c.no_ack_consumers
assert not self.c.is_open
self.c.collect()
def test_do_revive(self):
self.c.open = Mock(name='open')
self.c.is_open = True
self.c._do_revive()
assert not self.c.is_open
self.c.open.assert_called_with()
def test_close__not_open(self):
self.c.is_open = False
self.c.close()
def test_close__no_connection(self):
self.c.connection = None
self.c.close()
def test_close(self):
self.c.is_open = True
self.c.close(30, 'text', spec.Queue.Declare)
self.c.send_method.assert_called_with(
spec.Channel.Close,
'BsBB',
(30, 'text', spec.Queue.Declare[0], spec.Queue.Declare[1]),
wait=spec.Channel.CloseOk,
)
assert self.c.is_closing is False
assert self.c.connection is None
def test_on_close(self):
self.c._do_revive = Mock(name='_do_revive')
with pytest.raises(NotFound):
self.c._on_close(404, 'text', 50, 61)
self.c.send_method.assert_called_with(spec.Channel.CloseOk)
self.c._do_revive.assert_called_with()
def test_on_close_ok(self):
self.c.collect = Mock(name='collect')
self.c._on_close_ok()
self.c.collect.assert_called_with()
def test_flow(self):
self.c.flow(0)
self.c.send_method.assert_called_with(
spec.Channel.Flow,
'b',
(0, ),
wait=spec.Channel.FlowOk,
)
def test_on_flow(self):
self.c._x_flow_ok = Mock(name='_x_flow_ok')
self.c._on_flow(0)
assert not self.c.active
self.c._x_flow_ok.assert_called_with(0)
def test_x_flow_ok(self):
self.c._x_flow_ok(1)
self.c.send_method.assert_called_with(spec.Channel.FlowOk, 'b', (1, ))
def test_open(self):
self.c.is_open = True
self.c.open()
self.c.is_open = False
self.c.open()
self.c.send_method.assert_called_with(
spec.Channel.Open,
's',
('', ),
wait=spec.Channel.OpenOk,
)
def test_on_open_ok(self):
self.c.on_open = Mock(name='on_open')
self.c.is_open = False
self.c._on_open_ok()
assert self.c.is_open
self.c.on_open.assert_called_with(self.c)
def test_exchange_declare(self):
self.c.exchange_declare(
'foo',
'direct',
False,
True,
auto_delete=False,
nowait=False,
arguments={'x': 1},
)
self.c.send_method.assert_called_with(
spec.Exchange.Declare,
'BssbbbbbF',
(0, 'foo', 'direct', False, True, False, False, False, {
'x': 1
}),
wait=spec.Exchange.DeclareOk,
)
def test_exchange_declare__auto_delete(self):
self.c.exchange_declare(
'foo',
'direct',
False,
True,
auto_delete=True,
nowait=False,
arguments={'x': 1},
)
def test_exchange_delete(self):
self.c.exchange_delete('foo')
self.c.send_method.assert_called_with(
spec.Exchange.Delete,
'Bsbb',
(0, 'foo', False, False),
wait=spec.Exchange.DeleteOk,
)
def test_exchange_bind(self):
self.c.exchange_bind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Bind,
'BsssbF',
(0, 'dest', 'source', 'rkey', False, {
'x': 1
}),
wait=spec.Exchange.BindOk,
)
def test_exchange_unbind(self):
self.c.exchange_unbind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Unbind,
'BsssbF',
(0, 'dest', 'source', 'rkey', False, {
'x': 1
}),
wait=spec.Exchange.UnbindOk,
)
def test_queue_bind(self):
self.c.queue_bind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Bind,
'BsssbF',
(0, 'q', 'ex', 'rkey', False, {
'x': 1
}),
wait=spec.Queue.BindOk,
)
def test_queue_unbind(self):
self.c.queue_unbind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Unbind,
'BsssF',
(0, 'q', 'ex', 'rkey', {
'x': 1
}),
wait=spec.Queue.UnbindOk,
)
def test_queue_declare(self):
self.c.queue_declare('q', False, True, False, False, True, {'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Declare,
'BsbbbbbF',
(0, 'q', False, True, False, False, True, {
'x': 1
}),
)
def test_queue_declare__sync(self):
self.c.wait = Mock(name='wait')
self.c.wait.return_value = ('name', 123, 45)
ret = self.c.queue_declare(
'q',
False,
True,
False,
False,
False,
{'x': 1},
)
self.c.send_method.assert_called_with(
spec.Queue.Declare,
'BsbbbbbF',
(0, 'q', False, True, False, False, False, {
'x': 1
}),
)
assert ret.queue == 'name'
assert ret.message_count == 123
assert ret.consumer_count == 45
self.c.wait.assert_called_with(spec.Queue.DeclareOk,
returns_tuple=True)
def test_queue_delete(self):
self.c.queue_delete('q')
self.c.send_method.assert_called_with(
spec.Queue.Delete,
'Bsbbb',
(0, 'q', False, False, False),
wait=spec.Queue.DeleteOk,
)
def test_queue_purge(self):
self.c.queue_purge('q')
self.c.send_method.assert_called_with(
spec.Queue.Purge,
'Bsb',
(0, 'q', False),
wait=spec.Queue.PurgeOk,
)
def test_basic_ack(self):
self.c.basic_ack(123, multiple=1)
self.c.send_method.assert_called_with(
spec.Basic.Ack,
'Lb',
(123, 1),
)
def test_basic_cancel(self):
self.c.basic_cancel(123)
self.c.send_method.assert_called_with(
spec.Basic.Cancel,
'sb',
(123, False),
wait=spec.Basic.CancelOk,
)
self.c.connection = None
self.c.basic_cancel(123)
def test_on_basic_cancel(self):
self.c._remove_tag = Mock(name='_remove_tag')
self.c._on_basic_cancel(123)
self.c._remove_tag.return_value.assert_called_with(123)
self.c._remove_tag.return_value = None
with pytest.raises(ConsumerCancelled):
self.c._on_basic_cancel(123)
def test_on_basic_cancel_ok(self):
self.c._remove_tag = Mock(name='remove_tag')
self.c._on_basic_cancel_ok(123)
self.c._remove_tag.assert_called_with(123)
def test_remove_tag(self):
self.c.callbacks[123] = Mock()
p = self.c.cancel_callbacks[123] = Mock()
assert self.c._remove_tag(123) is p
assert 123 not in self.c.callbacks
assert 123 not in self.c.cancel_callbacks
def test_basic_consume(self):
callback = Mock()
on_cancel = Mock()
self.c.send_method.return_value = (123, )
self.c.basic_consume(
'q',
123,
arguments={'x': 1},
callback=callback,
on_cancel=on_cancel,
)
self.c.send_method.assert_called_with(
spec.Basic.Consume,
'BssbbbbF', (0, 'q', 123, False, False, False, False, {
'x': 1
}),
wait=spec.Basic.ConsumeOk,
returns_tuple=True)
assert self.c.callbacks[123] is callback
assert self.c.cancel_callbacks[123] is on_cancel
def test_basic_consume__no_ack(self):
self.c.send_method.return_value = (123, )
self.c.basic_consume(
'q',
123,
arguments={'x': 1},
no_ack=True,
)
assert 123 in self.c.no_ack_consumers
def test_basic_consume_no_consumer_tag(self):
callback = Mock()
self.c.send_method.return_value = (123, )
ret = self.c.basic_consume(
'q',
arguments={'x': 1},
callback=callback,
)
self.c.send_method.assert_called_with(
spec.Basic.Consume,
'BssbbbbF', (0, 'q', '', False, False, False, False, {
'x': 1
}),
wait=spec.Basic.ConsumeOk,
returns_tuple=True)
assert self.c.callbacks[123] is callback
assert ret == 123
def test_basic_consume_no_wait(self):
callback = Mock()
ret_promise = promise()
self.c.send_method.return_value = ret_promise
ret = self.c.basic_consume('q',
123,
arguments={'x': 1},
callback=callback,
nowait=True)
self.c.send_method.assert_called_with(
spec.Basic.Consume,
'BssbbbbF', (0, 'q', 123, False, False, False, True, {
'x': 1
}),
wait=None,
returns_tuple=True)
assert self.c.callbacks[123] is callback
assert ret == ret_promise
def test_basic_consume_no_wait_no_consumer_tag(self):
callback = Mock()
with pytest.raises(ValueError):
self.c.basic_consume('q',
arguments={'x': 1},
callback=callback,
nowait=True)
assert 123 not in self.c.callbacks
def test_on_basic_deliver(self):
msg = Message()
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback = self.c.callbacks[123] = Mock(name='cb')
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback.assert_called_with(msg)
assert msg.channel == self.c
assert msg.delivery_info == {
'consumer_tag': 123,
'delivery_tag': '321',
'redelivered': False,
'exchange': 'ex',
'routing_key': 'rkey',
}
def test_basic_get(self):
self.c._on_get_empty = Mock()
self.c._on_get_ok = Mock()
self.c.send_method.return_value = ('cluster_id', )
self.c.basic_get('q')
self.c.send_method.assert_called_with(
spec.Basic.Get,
'Bsb',
(0, 'q', False),
wait=[spec.Basic.GetOk, spec.Basic.GetEmpty],
returns_tuple=True,
)
self.c._on_get_empty.assert_called_with('cluster_id')
self.c.send_method.return_value = (
'dtag',
'redelivered',
'ex',
'rkey',
'mcount',
'msg',
)
self.c.basic_get('q')
self.c._on_get_ok.assert_called_with(
'dtag',
'redelivered',
'ex',
'rkey',
'mcount',
'msg',
)
def test_on_get_empty(self):
self.c._on_get_empty(1)
def test_on_get_ok(self):
msg = Message()
m = self.c._on_get_ok(
'dtag',
'redelivered',
'ex',
'rkey',
'mcount',
msg,
)
assert m is msg
assert m.channel == self.c
assert m.delivery_info == {
'delivery_tag': 'dtag',
'redelivered': 'redelivered',
'exchange': 'ex',
'routing_key': 'rkey',
'message_count': 'mcount',
}
def test_basic_publish(self):
self.c.connection.transport.having_timeout = ContextMock()
self.c._basic_publish('msg', 'ex', 'rkey')
self.c.send_method.assert_called_with(
spec.Basic.Publish,
'Bssbb',
(0, 'ex', 'rkey', False, False),
'msg',
)
def test_basic_publish_confirm(self):
self.c._confirm_selected = False
self.c.confirm_select = Mock(name='confirm_select')
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait')
ret = self.c.basic_publish_confirm(1, 2, arg=1)
self.c.confirm_select.assert_called_with()
assert self.c._confirm_selected
self.c._basic_publish.assert_called_with(1, 2, arg=1)
assert ret is self.c._basic_publish()
self.c.wait.assert_called_with([spec.Basic.Ack, spec.Basic.Nack],
callback=ANY,
timeout=None)
self.c.basic_publish_confirm(1, 2, arg=1)
def test_basic_publish_confirm_nack(self):
# test checking whether library is handling correctly Nack confirms
# sent from RabbitMQ. Library must raise MessageNacked when server
# sent Nack message.
# Nack frame construction
args = dumps('Lb', (1, False))
frame = (b''.join([pack('>HH', *spec.Basic.Nack), args]))
def wait(method, *args, **kwargs):
# Simple mock simulating registering callbacks of real wait method
for m in method:
self.c._pending[m] = kwargs['callback']
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait', side_effect=wait)
self.c.basic_publish_confirm(1, 2, arg=1)
with pytest.raises(MessageNacked):
# Inject Nack to message handler
self.c.dispatch_method(spec.Basic.Nack, frame, None)
def test_basic_publish_connection_blocked(self):
# Basic test checking that drain_events() is called
# before publishing message and send_method() is called
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_called_once_with(timeout=0)
self.c.send_method.assert_called_once_with(
spec.Basic.Publish,
'Bssbb',
(0, 'ex', 'rkey', False, False),
'msg',
)
self.c.send_method.reset_mock()
# Basic test checking that socket.timeout exception
# is ignored and send_method() is called.
self.conn.drain_events.side_effect = socket.timeout
self.c._basic_publish('msg', 'ex', 'rkey')
self.c.send_method.assert_called_once_with(
spec.Basic.Publish,
'Bssbb',
(0, 'ex', 'rkey', False, False),
'msg',
)
def test_basic_publish_connection_blocked_not_supported(self):
# Test veryfying that when server does not have
# connection.blocked capability, drain_events() are not called
self.conn.client_properties = {
'capabilities': {
'connection.blocked': False
}
}
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_not_called()
self.c.send_method.assert_called_once_with(
spec.Basic.Publish,
'Bssbb',
(0, 'ex', 'rkey', False, False),
'msg',
)
def test_basic_publish_connection_blocked_not_supported_missing(self):
# Test veryfying that when server does not have
# connection.blocked capability, drain_events() are not called
self.conn.client_properties = {'capabilities': {}}
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_not_called()
self.c.send_method.assert_called_once_with(
spec.Basic.Publish,
'Bssbb',
(0, 'ex', 'rkey', False, False),
'msg',
)
def test_basic_publish_connection_blocked_no_capabilities(self):
# Test veryfying that when server does not have
# support of capabilities, drain_events() are not called
self.conn.client_properties = {}
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_not_called()
self.c.send_method.assert_called_once_with(
spec.Basic.Publish,
'Bssbb',
(0, 'ex', 'rkey', False, False),
'msg',
)
def test_basic_publish_confirm_callback(self):
def wait_nack(method, *args, **kwargs):
kwargs['callback'](spec.Basic.Nack)
def wait_ack(method, *args, **kwargs):
kwargs['callback'](spec.Basic.Ack)
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait_nack', side_effect=wait_nack)
with pytest.raises(MessageNacked):
# when callback is called with spec.Basic.Nack it must raise
# MessageNacked exception
self.c.basic_publish_confirm(1, 2, arg=1)
self.c.wait = Mock(name='wait_ack', side_effect=wait_ack)
# when callback is called with spec.Basic.Ack
# it must nost raise exception
self.c.basic_publish_confirm(1, 2, arg=1)
def test_basic_publish_connection_closed(self):
self.c.collect()
with pytest.raises(RecoverableConnectionError) as excinfo:
self.c._basic_publish('msg', 'ex', 'rkey')
assert 'basic_publish: connection closed' in str(excinfo.value)
self.c.send_method.assert_not_called()
def test_basic_qos(self):
self.c.basic_qos(0, 123, False)
self.c.send_method.assert_called_with(
spec.Basic.Qos,
'lBb',
(0, 123, False),
wait=spec.Basic.QosOk,
)
def test_basic_recover(self):
self.c.basic_recover(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Recover,
'b',
(True, ),
)
def test_basic_recover_async(self):
self.c.basic_recover_async(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.RecoverAsync,
'b',
(True, ),
)
def test_basic_reject(self):
self.c.basic_reject(123, requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Reject,
'Lb',
(123, True),
)
def test_on_basic_return(self):
with pytest.raises(NotFound):
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
def test_on_basic_return__handled(self):
with patch('amqp.channel.error_for_code') as error_for_code:
callback = Mock(name='callback')
self.c.events['basic_return'].add(callback)
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
callback.assert_called_with(
error_for_code(),
'ex',
'rkey',
'msg',
)
def test_tx_commit(self):
self.c.tx_commit()
self.c.send_method.assert_called_with(
spec.Tx.Commit,
wait=spec.Tx.CommitOk,
)
def test_tx_rollback(self):
self.c.tx_rollback()
self.c.send_method.assert_called_with(
spec.Tx.Rollback,
wait=spec.Tx.RollbackOk,
)
def test_tx_select(self):
self.c.tx_select()
self.c.send_method.assert_called_with(
spec.Tx.Select,
wait=spec.Tx.SelectOk,
)
def test_confirm_select(self):
self.c.confirm_select()
self.c.send_method.assert_called_with(
spec.Confirm.Select,
'b',
(False, ),
wait=spec.Confirm.SelectOk,
)
def test_on_basic_ack(self):
callback = Mock(name='callback')
self.c.events['basic_ack'].add(callback)
self.c._on_basic_ack(123, True)
callback.assert_called_with(123, True)
def test_on_basic_nack(self):
callback = Mock(name='callback')
self.c.events['basic_nack'].add(callback)
self.c._on_basic_nack(123, True)
callback.assert_called_with(123, True)
def test_init_explicit_channel(self):
Channel(self.conn, 3)
self.conn._claim_channel_id.assert_called_with(3)
def test_init_auto_channel(self):
c = Channel(self.conn, None)
self.conn._get_free_channel_id.assert_called_with()
assert c.channel_id is self.conn._get_free_channel_id()
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
assert c.basic_publish == c._basic_publish
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
assert c.basic_publish == c.basic_publish_confirm
class test_Channel(Case):
def setup(self):
self.conn = Mock(name='connection')
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c.basic_publish_confirm)
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c._basic_publish)
def test_init_auto_channel(self):
c = Channel(self.conn, None)
self.conn._get_free_channel_id.assert_called_with()
self.assertIs(c.channel_id, self.conn._get_free_channel_id())
def test_init_explicit_channel(self):
Channel(self.conn, 3)
self.conn._claim_channel_id.assert_called_with(3)
def test_then(self):
self.c.on_open = Mock(name='on_open')
on_success = Mock(name='on_success')
on_error = Mock(name='on_error')
self.c.then(on_success, on_error)
self.c.on_open.then.assert_called_with(on_success, on_error)
def test_collect(self):
self.c.callbacks[(50, 61)] = Mock()
self.c.cancel_callbacks['foo'] = Mock()
self.c.events['bar'].add(Mock())
self.c.no_ack_consumers.add('foo')
self.c.collect()
self.assertFalse(self.c.callbacks)
self.assertFalse(self.c.cancel_callbacks)
self.assertFalse(self.c.events)
self.assertFalse(self.c.no_ack_consumers)
self.assertFalse(self.c.is_open)
self.c.collect()
def test_do_revive(self):
self.c.open = Mock(name='open')
self.c.is_open = True
self.c._do_revive()
self.assertFalse(self.c.is_open)
self.c.open.assert_called_with()
def test_close__not_open(self):
self.c.is_open = False
self.c.close()
def test_close__no_connection(self):
self.c.connection = None
self.c.close()
def test_close(self):
self.c.is_open = True
self.c.close(30, 'text', spec.Queue.Declare)
self.c.send_method.assert_called_with(
spec.Channel.Close,
'BsBB',
(30, 'text', spec.Queue.Declare[0], spec.Queue.Declare[1]),
wait=spec.Channel.CloseOk,
)
self.assertIsNone(self.c.connection)
def test_on_close(self):
self.c._do_revive = Mock(name='_do_revive')
with self.assertRaises(NotFound):
self.c._on_close(404, 'text', 50, 61)
self.c.send_method.assert_called_with(spec.Channel.CloseOk)
self.c._do_revive.assert_called_with()
def test_on_close_ok(self):
self.c.collect = Mock(name='collect')
self.c._on_close_ok()
self.c.collect.assert_called_with()
def test_flow(self):
self.c.flow(0)
self.c.send_method.assert_called_with(
spec.Channel.Flow,
'b',
(0, ),
wait=spec.Channel.FlowOk,
)
def test_on_flow(self):
self.c._x_flow_ok = Mock(name='_x_flow_ok')
self.c._on_flow(0)
self.assertFalse(self.c.active)
self.c._x_flow_ok.assert_called_with(0)
def test_x_flow_ok(self):
self.c._x_flow_ok(1)
self.c.send_method.assert_called_with(spec.Channel.FlowOk, 'b', (1, ))
def test_open(self):
self.c.is_open = True
self.c.open()
self.c.is_open = False
self.c.open()
self.c.send_method.assert_called_with(
spec.Channel.Open,
's',
('', ),
wait=spec.Channel.OpenOk,
)
def test_on_open_ok(self):
self.c.on_open = Mock(name='on_open')
self.c.is_open = False
self.c._on_open_ok()
self.assertTrue(self.c.is_open)
self.c.on_open.assert_called_with(self.c)
def test_exchange_declare(self):
self.c.exchange_declare(
'foo',
'direct',
False,
True,
auto_delete=False,
nowait=False,
arguments={'x': 1},
)
self.c.send_method.assert_called_with(
spec.Exchange.Declare,
'BssbbbbbF',
(0, 'foo', 'direct', False, True, False, False, False, {
'x': 1
}),
wait=spec.Exchange.DeclareOk,
)
@patch('amqp.channel.warn')
def test_exchange_declare__auto_delete(self, warn):
self.c.exchange_declare(
'foo',
'direct',
False,
True,
auto_delete=True,
nowait=False,
arguments={'x': 1},
)
warn.assert_called()
def test_exchange_delete(self):
self.c.exchange_delete('foo')
self.c.send_method.assert_called_with(
spec.Exchange.Delete,
'Bsbb',
(0, 'foo', False, False),
wait=spec.Exchange.DeleteOk,
)
def test_exchange_bind(self):
self.c.exchange_bind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Bind,
'BsssbF',
(0, 'dest', 'source', 'rkey', False, {
'x': 1
}),
wait=spec.Exchange.BindOk,
)
def test_exchange_unbind(self):
self.c.exchange_unbind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Unbind,
'BsssbF',
(0, 'dest', 'source', 'rkey', False, {
'x': 1
}),
wait=spec.Exchange.UnbindOk,
)
def test_queue_bind(self):
self.c.queue_bind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Bind,
'BsssbF',
(0, 'q', 'ex', 'rkey', False, {
'x': 1
}),
wait=spec.Queue.BindOk,
)
def test_queue_unbind(self):
self.c.queue_unbind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Unbind,
'BsssF',
(0, 'q', 'ex', 'rkey', {
'x': 1
}),
wait=spec.Queue.UnbindOk,
)
def test_queue_declare(self):
self.c.queue_declare('q', False, True, False, False, True, {'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Declare,
'BsbbbbbF',
(0, 'q', False, True, False, False, True, {
'x': 1
}),
)
def test_queue_declare__sync(self):
self.c.wait = Mock(name='wait')
self.c.wait.return_value = ('name', 123, 45)
ret = self.c.queue_declare(
'q',
False,
True,
False,
False,
False,
{'x': 1},
)
self.c.send_method.assert_called_with(
spec.Queue.Declare,
'BsbbbbbF',
(0, 'q', False, True, False, False, False, {
'x': 1
}),
)
self.assertEqual(ret.queue, 'name')
self.assertEqual(ret.message_count, 123)
self.assertEqual(ret.consumer_count, 45)
self.c.wait.assert_called_with(spec.Queue.DeclareOk,
returns_tuple=True)
def test_queue_delete(self):
self.c.queue_delete('q')
self.c.send_method.assert_called_with(
spec.Queue.Delete,
'Bsbbb',
(0, 'q', False, False, False),
wait=spec.Queue.DeleteOk,
)
def test_queue_purge(self):
self.c.queue_purge('q')
self.c.send_method.assert_called_with(
spec.Queue.Purge,
'Bsb',
(0, 'q', False),
wait=spec.Queue.PurgeOk,
)
def test_basic_ack(self):
self.c.basic_ack(123, multiple=1)
self.c.send_method.assert_called_with(
spec.Basic.Ack,
'Lb',
(123, 1),
)
def test_basic_cancel(self):
self.c.basic_cancel(123)
self.c.send_method.assert_called_with(
spec.Basic.Cancel,
'sb',
(123, False),
wait=spec.Basic.CancelOk,
)
self.c.connection = None
self.c.basic_cancel(123)
def test_on_basic_cancel(self):
self.c._remove_tag = Mock(name='_remove_tag')
self.c._on_basic_cancel(123)
self.c._remove_tag.return_value.assert_called_with(123)
self.c._remove_tag.return_value = None
with self.assertRaises(ConsumerCancelled):
self.c._on_basic_cancel(123)
def test_on_basic_cancel_ok(self):
self.c._remove_tag = Mock(name='remove_tag')
self.c._on_basic_cancel_ok(123)
self.c._remove_tag.assert_called_with(123)
def test_remove_tag(self):
self.c.callbacks[123] = Mock()
p = self.c.cancel_callbacks[123] = Mock()
self.assertIs(self.c._remove_tag(123), p)
self.assertNotIn(123, self.c.callbacks)
self.assertNotIn(123, self.c.cancel_callbacks)
def test_basic_consume(self):
callback = Mock()
on_cancel = Mock()
self.c.basic_consume(
'q',
123,
arguments={'x': 1},
callback=callback,
on_cancel=on_cancel,
)
self.c.send_method.assert_called_with(
spec.Basic.Consume,
'BssbbbbF',
(0, 'q', 123, False, False, False, False, {
'x': 1
}),
wait=spec.Basic.ConsumeOk,
)
self.assertIs(self.c.callbacks[123], callback)
self.assertIs(self.c.cancel_callbacks[123], on_cancel)
def test_basic_consume__no_ack(self):
self.c.basic_consume(
'q',
123,
arguments={'x': 1},
no_ack=True,
)
self.assertIn(123, self.c.no_ack_consumers)
def test_on_basic_deliver(self):
msg = Mock()
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback = self.c.callbacks[123] = Mock(name='cb')
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback.assert_called_with(msg)
def test_basic_get(self):
self.c._on_get_empty = Mock()
self.c._on_get_ok = Mock()
self.c.send_method.return_value = ('cluster_id', )
self.c.basic_get('q')
self.c.send_method.assert_called_with(
spec.Basic.Get,
'Bsb',
(0, 'q', False),
wait=[spec.Basic.GetOk, spec.Basic.GetEmpty],
returns_tuple=True,
)
self.c._on_get_empty.assert_called_with('cluster_id')
self.c.send_method.return_value = (
'dtag',
'redelivered',
'ex',
'rkey',
'mcount',
'msg',
)
self.c.basic_get('q')
self.c._on_get_ok.assert_called_with(
'dtag',
'redelivered',
'ex',
'rkey',
'mcount',
'msg',
)
def test_on_get_empty(self):
self.c._on_get_empty(1)
def test_on_get_ok(self):
msg = Mock()
m = self.c._on_get_ok(
'dtag',
'redelivered',
'ex',
'rkey',
'mcount',
msg,
)
self.assertIs(m, msg)
def test_basic_publish(self):
self.c.connection.transport.having_timeout = ContextMock()
self.c._basic_publish('msg', 'ex', 'rkey')
self.c.send_method.assert_called_with(
spec.Basic.Publish,
'Bssbb',
(0, 'ex', 'rkey', False, False),
'msg',
)
def test_basic_publish_confirm(self):
self.c._confirm_selected = False
self.c.confirm_select = Mock(name='confirm_select')
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait')
ret = self.c.basic_publish_confirm(1, 2, arg=1)
self.c.confirm_select.assert_called_with()
self.assertTrue(self.c._confirm_selected)
self.c._basic_publish.assert_called_with(1, 2, arg=1)
self.assertIs(ret, self.c._basic_publish())
self.c.wait.assert_called_with(spec.Basic.Ack)
self.c.basic_publish_confirm(1, 2, arg=1)
def test_basic_qos(self):
self.c.basic_qos(0, 123, False)
self.c.send_method.assert_called_with(
spec.Basic.Qos,
'lBb',
(0, 123, False),
wait=spec.Basic.QosOk,
)
def test_basic_recover(self):
self.c.basic_recover(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Recover,
'b',
(True, ),
)
def test_basic_recover_async(self):
self.c.basic_recover_async(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.RecoverAsync,
'b',
(True, ),
)
def test_basic_reject(self):
self.c.basic_reject(123, requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Reject,
'Lb',
(123, True),
)
def test_on_basic_return(self):
with self.assertRaises(NotFound):
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
@patch('amqp.channel.error_for_code')
def test_on_basic_return__handled(self, error_for_code):
callback = Mock(name='callback')
self.c.events['basic_return'].add(callback)
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
callback.assert_called_with(
error_for_code(),
'ex',
'rkey',
'msg',
)
def test_tx_commit(self):
self.c.tx_commit()
self.c.send_method.assert_called_with(
spec.Tx.Commit,
wait=spec.Tx.CommitOk,
)
def test_tx_rollback(self):
self.c.tx_rollback()
self.c.send_method.assert_called_with(
spec.Tx.Rollback,
wait=spec.Tx.RollbackOk,
)
def test_tx_select(self):
self.c.tx_select()
self.c.send_method.assert_called_with(
spec.Tx.Select,
wait=spec.Tx.SelectOk,
)
def test_confirm_select(self):
self.c.confirm_select()
self.c.send_method.assert_called_with(
spec.Confirm.Select,
'b',
(False, ),
wait=spec.Confirm.SelectOk,
)
def test_on_basic_ack(self):
callback = Mock(name='callback')
self.c.events['basic_ack'].add(callback)
self.c._on_basic_ack(123, True)
callback.assert_called_with(123, True)
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c._basic_publish)
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c.basic_publish_confirm)
def setup(self):
self.conn = Mock(name='connection')
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
class test_Channel(Case):
def setup(self):
self.conn = Mock(name="connection")
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name="send_method")
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c.basic_publish_confirm)
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c._basic_publish)
def test_init_auto_channel(self):
c = Channel(self.conn, None)
self.conn._get_free_channel_id.assert_called_with()
self.assertIs(c.channel_id, self.conn._get_free_channel_id())
def test_init_explicit_channel(self):
Channel(self.conn, 3)
self.conn._claim_channel_id.assert_called_with(3)
def test_then(self):
self.c.on_open = Mock(name="on_open")
on_success = Mock(name="on_success")
on_error = Mock(name="on_error")
self.c.then(on_success, on_error)
self.c.on_open.then.assert_called_with(on_success, on_error)
def test_collect(self):
self.c.callbacks[(50, 61)] = Mock()
self.c.cancel_callbacks["foo"] = Mock()
self.c.events["bar"].add(Mock())
self.c.no_ack_consumers.add("foo")
self.c.collect()
self.assertFalse(self.c.callbacks)
self.assertFalse(self.c.cancel_callbacks)
self.assertFalse(self.c.events)
self.assertFalse(self.c.no_ack_consumers)
self.assertFalse(self.c.is_open)
self.c.collect()
def test_do_revive(self):
self.c.open = Mock(name="open")
self.c.is_open = True
self.c._do_revive()
self.assertFalse(self.c.is_open)
self.c.open.assert_called_with()
def test_close__not_open(self):
self.c.is_open = False
self.c.close()
def test_close__no_connection(self):
self.c.connection = None
self.c.close()
def test_close(self):
self.c.is_open = True
self.c.close(30, "text", spec.Queue.Declare)
self.c.send_method.assert_called_with(
spec.Channel.Close,
"BsBB",
(30, "text", spec.Queue.Declare[0], spec.Queue.Declare[1]),
wait=spec.Channel.CloseOk,
)
self.assertIsNone(self.c.connection)
def test_on_close(self):
self.c._do_revive = Mock(name="_do_revive")
with self.assertRaises(NotFound):
self.c._on_close(404, "text", 50, 61)
self.c.send_method.assert_called_with(spec.Channel.CloseOk)
self.c._do_revive.assert_called_with()
def test_on_close_ok(self):
self.c.collect = Mock(name="collect")
self.c._on_close_ok()
self.c.collect.assert_called_with()
def test_flow(self):
self.c.flow(0)
self.c.send_method.assert_called_with(spec.Channel.Flow, "b", (0,), wait=spec.Channel.FlowOk)
def test_on_flow(self):
self.c._x_flow_ok = Mock(name="_x_flow_ok")
self.c._on_flow(0)
self.assertFalse(self.c.active)
self.c._x_flow_ok.assert_called_with(0)
def test_x_flow_ok(self):
self.c._x_flow_ok(1)
self.c.send_method.assert_called_with(spec.Channel.FlowOk, "b", (1,))
def test_open(self):
self.c.is_open = True
self.c.open()
self.c.is_open = False
self.c.open()
self.c.send_method.assert_called_with(spec.Channel.Open, "s", ("",), wait=spec.Channel.OpenOk)
def test_on_open_ok(self):
self.c.on_open = Mock(name="on_open")
self.c.is_open = False
self.c._on_open_ok()
self.assertTrue(self.c.is_open)
self.c.on_open.assert_called_with(self.c)
def test_exchange_declare(self):
self.c.exchange_declare("foo", "direct", False, True, auto_delete=False, nowait=False, arguments={"x": 1})
self.c.send_method.assert_called_with(
spec.Exchange.Declare,
"BssbbbbbF",
(0, "foo", "direct", False, True, False, False, False, {"x": 1}),
wait=spec.Exchange.DeclareOk,
)
@patch("amqp.channel.warn")
def test_exchange_declare__auto_delete(self, warn):
self.c.exchange_declare("foo", "direct", False, True, auto_delete=True, nowait=False, arguments={"x": 1})
self.assertTrue(warn.called)
def test_exchange_delete(self):
self.c.exchange_delete("foo")
self.c.send_method.assert_called_with(
spec.Exchange.Delete, "Bsbb", (0, "foo", False, False), wait=spec.Exchange.DeleteOk
)
def test_exchange_bind(self):
self.c.exchange_bind("dest", "source", "rkey", arguments={"x": 1})
self.c.send_method.assert_called_with(
spec.Exchange.Bind, "BsssbF", (0, "dest", "source", "rkey", False, {"x": 1}), wait=spec.Exchange.BindOk
)
def test_exchange_unbind(self):
self.c.exchange_unbind("dest", "source", "rkey", arguments={"x": 1})
self.c.send_method.assert_called_with(
spec.Exchange.Unbind, "BsssbF", (0, "dest", "source", "rkey", False, {"x": 1}), wait=spec.Exchange.UnbindOk
)
def test_queue_bind(self):
self.c.queue_bind("q", "ex", "rkey", arguments={"x": 1})
self.c.send_method.assert_called_with(
spec.Queue.Bind, "BsssbF", (0, "q", "ex", "rkey", False, {"x": 1}), wait=spec.Queue.BindOk
)
def test_queue_unbind(self):
self.c.queue_unbind("q", "ex", "rkey", arguments={"x": 1})
self.c.send_method.assert_called_with(
spec.Queue.Unbind, "BsssF", (0, "q", "ex", "rkey", {"x": 1}), wait=spec.Queue.UnbindOk
)
def test_queue_declare(self):
self.c.queue_declare("q", False, True, False, False, True, {"x": 1})
self.c.send_method.assert_called_with(
spec.Queue.Declare, "BsbbbbbF", (0, "q", False, True, False, False, True, {"x": 1})
)
def test_queue_declare__sync(self):
self.c.wait = Mock(name="wait")
self.c.wait.return_value = ("name", 123, 45)
ret = self.c.queue_declare("q", False, True, False, False, False, {"x": 1})
self.c.send_method.assert_called_with(
spec.Queue.Declare, "BsbbbbbF", (0, "q", False, True, False, False, False, {"x": 1})
)
self.assertEqual(ret.queue, "name")
self.assertEqual(ret.message_count, 123)
self.assertEqual(ret.consumer_count, 45)
self.c.wait.assert_called_with(spec.Queue.DeclareOk, returns_tuple=True)
def test_queue_delete(self):
self.c.queue_delete("q")
self.c.send_method.assert_called_with(
spec.Queue.Delete, "Bsbbb", (0, "q", False, False, False), wait=spec.Queue.DeleteOk
)
def test_queue_purge(self):
self.c.queue_purge("q")
self.c.send_method.assert_called_with(spec.Queue.Purge, "Bsb", (0, "q", False), wait=spec.Queue.PurgeOk)
def test_basic_ack(self):
self.c.basic_ack(123, multiple=1)
self.c.send_method.assert_called_with(spec.Basic.Ack, "Lb", (123, 1))
def test_basic_cancel(self):
self.c.basic_cancel(123)
self.c.send_method.assert_called_with(spec.Basic.Cancel, "sb", (123, False), wait=spec.Basic.CancelOk)
self.c.connection = None
self.c.basic_cancel(123)
def test_on_basic_cancel(self):
self.c._remove_tag = Mock(name="_remove_tag")
self.c._on_basic_cancel(123)
self.c._remove_tag.return_value.assert_called_with(123)
self.c._remove_tag.return_value = None
with self.assertRaises(ConsumerCancelled):
self.c._on_basic_cancel(123)
def test_on_basic_cancel_ok(self):
self.c._remove_tag = Mock(name="remove_tag")
self.c._on_basic_cancel_ok(123)
self.c._remove_tag.assert_called_with(123)
def test_remove_tag(self):
self.c.callbacks[123] = Mock()
p = self.c.cancel_callbacks[123] = Mock()
self.assertIs(self.c._remove_tag(123), p)
self.assertNotIn(123, self.c.callbacks)
self.assertNotIn(123, self.c.cancel_callbacks)
def test_basic_consume(self):
callback = Mock()
on_cancel = Mock()
self.c.basic_consume("q", 123, arguments={"x": 1}, callback=callback, on_cancel=on_cancel)
self.c.send_method.assert_called_with(
spec.Basic.Consume,
"BssbbbbF",
(0, "q", 123, False, False, False, False, {"x": 1}),
wait=spec.Basic.ConsumeOk,
)
self.assertIs(self.c.callbacks[123], callback)
self.assertIs(self.c.cancel_callbacks[123], on_cancel)
def test_basic_consume__no_ack(self):
self.c.basic_consume("q", 123, arguments={"x": 1}, no_ack=True)
self.assertIn(123, self.c.no_ack_consumers)
def test_on_basic_deliver(self):
msg = Mock()
self.c._on_basic_deliver(123, "321", False, "ex", "rkey", msg)
callback = self.c.callbacks[123] = Mock(name="cb")
self.c._on_basic_deliver(123, "321", False, "ex", "rkey", msg)
callback.assert_called_with(msg)
def test_basic_get(self):
self.c._on_get_empty = Mock()
self.c._on_get_ok = Mock()
self.c.send_method.return_value = ("cluster_id",)
self.c.basic_get("q")
self.c.send_method.assert_called_with(
spec.Basic.Get, "Bsb", (0, "q", False), wait=[spec.Basic.GetOk, spec.Basic.GetEmpty], returns_tuple=True
)
self.c._on_get_empty.assert_called_with("cluster_id")
self.c.send_method.return_value = ("dtag", "redelivered", "ex", "rkey", "mcount", "msg")
self.c.basic_get("q")
self.c._on_get_ok.assert_called_with("dtag", "redelivered", "ex", "rkey", "mcount", "msg")
def test_on_get_empty(self):
self.c._on_get_empty(1)
def test_on_get_ok(self):
msg = Mock()
m = self.c._on_get_ok("dtag", "redelivered", "ex", "rkey", "mcount", msg)
self.assertIs(m, msg)
def test_basic_publish(self):
self.c._basic_publish("msg", "ex", "rkey")
self.c.send_method.assert_called_with(spec.Basic.Publish, "Bssbb", (0, "ex", "rkey", False, False), "msg")
def test_basic_publish_confirm(self):
self.c._confirm_selected = False
self.c.confirm_select = Mock(name="confirm_select")
self.c._basic_publish = Mock(name="_basic_publish")
self.c.wait = Mock(name="wait")
ret = self.c.basic_publish_confirm(1, 2, arg=1)
self.c.confirm_select.assert_called_with()
self.assertTrue(self.c._confirm_selected)
self.c._basic_publish.assert_called_with(1, 2, arg=1)
self.assertIs(ret, self.c._basic_publish())
self.c.wait.assert_called_with(spec.Basic.Ack)
self.c.basic_publish_confirm(1, 2, arg=1)
def test_basic_qos(self):
self.c.basic_qos(0, 123, False)
self.c.send_method.assert_called_with(spec.Basic.Qos, "lBb", (0, 123, False), wait=spec.Basic.QosOk)
def test_basic_recover(self):
self.c.basic_recover(requeue=True)
self.c.send_method.assert_called_with(spec.Basic.Recover, "b", (True,))
def test_basic_recover_async(self):
self.c.basic_recover_async(requeue=True)
self.c.send_method.assert_called_with(spec.Basic.RecoverAsync, "b", (True,))
def test_basic_reject(self):
self.c.basic_reject(123, requeue=True)
self.c.send_method.assert_called_with(spec.Basic.Reject, "Lb", (123, True))
def test_on_basic_return(self):
with self.assertRaises(NotFound):
self.c._on_basic_return(404, "text", "ex", "rkey", "msg")
@patch("amqp.channel.error_for_code")
def test_on_basic_return__handled(self, error_for_code):
callback = Mock(name="callback")
self.c.events["basic_return"].add(callback)
self.c._on_basic_return(404, "text", "ex", "rkey", "msg")
callback.assert_called_with(error_for_code(), "ex", "rkey", "msg")
def test_tx_commit(self):
self.c.tx_commit()
self.c.send_method.assert_called_with(spec.Tx.Commit, wait=spec.Tx.CommitOk)
def test_tx_rollback(self):
self.c.tx_rollback()
self.c.send_method.assert_called_with(spec.Tx.Rollback, wait=spec.Tx.RollbackOk)
def test_tx_select(self):
self.c.tx_select()
self.c.send_method.assert_called_with(spec.Tx.Select, wait=spec.Tx.SelectOk)
def test_confirm_select(self):
self.c.confirm_select()
self.c.send_method.assert_called_with(spec.Confirm.Select, "b", (False,), wait=spec.Confirm.SelectOk)
def test_on_basic_ack(self):
callback = Mock(name="callback")
self.c.events["basic_ack"].add(callback)
self.c._on_basic_ack(123, True)
callback.assert_called_with(123, True)
#!/usr/bin/env python
from amqp.connection import Connection
from amqp.channel import Channel
from amqp.basic_message import Message
import json
connection = Connection()
channel = Channel(connection)
channel.exchange_declare('django_amqp_example', 'topic', auto_delete=False)
channel.queue_declare(queue='task_queue', durable=True)
channel.queue_bind('task_queue', 'django_amqp_example', 'task_queue')
def callback(msg):
print "Received request:", msg.body
content = json.loads(msg.body)['content']
response = {
'rot13': content.encode('rot13')
}
response_msg = Message(
body=json.dumps(response),
exchange='django_amqp_example')
print "Sending response:", json.dumps(response)
channel.basic_publish(
response_msg,
routing_key=msg.reply_to)
channel.basic_consume(callback=callback, queue='task_queue')
def setup(self):
self.conn = Mock(name='connection')
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
class test_Channel(Case):
def setup(self):
self.conn = Mock(name='connection')
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c.basic_publish_confirm)
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c._basic_publish)
def test_init_auto_channel(self):
c = Channel(self.conn, None)
self.conn._get_free_channel_id.assert_called_with()
self.assertIs(c.channel_id, self.conn._get_free_channel_id())
def test_init_explicit_channel(self):
Channel(self.conn, 3)
self.conn._claim_channel_id.assert_called_with(3)
def test_then(self):
self.c.on_open = Mock(name='on_open')
on_success = Mock(name='on_success')
on_error = Mock(name='on_error')
self.c.then(on_success, on_error)
self.c.on_open.then.assert_called_with(on_success, on_error)
def test_collect(self):
self.c.callbacks[(50, 61)] = Mock()
self.c.cancel_callbacks['foo'] = Mock()
self.c.events['bar'].add(Mock())
self.c.no_ack_consumers.add('foo')
self.c.collect()
self.assertFalse(self.c.callbacks)
self.assertFalse(self.c.cancel_callbacks)
self.assertFalse(self.c.events)
self.assertFalse(self.c.no_ack_consumers)
self.assertFalse(self.c.is_open)
self.c.collect()
def test_do_revive(self):
self.c.open = Mock(name='open')
self.c.is_open = True
self.c._do_revive()
self.assertFalse(self.c.is_open)
self.c.open.assert_called_with()
def test_close__not_open(self):
self.c.is_open = False
self.c.close()
def test_close__no_connection(self):
self.c.connection = None
self.c.close()
def test_close(self):
self.c.is_open = True
self.c.close(30, 'text', spec.Queue.Declare)
self.c.send_method.assert_called_with(
spec.Channel.Close, 'BsBB',
(30, 'text', spec.Queue.Declare[0], spec.Queue.Declare[1]),
wait=spec.Channel.CloseOk,
)
self.assertIsNone(self.c.connection)
def test_on_close(self):
self.c._do_revive = Mock(name='_do_revive')
with self.assertRaises(NotFound):
self.c._on_close(404, 'text', 50, 61)
self.c.send_method.assert_called_with(spec.Channel.CloseOk)
self.c._do_revive.assert_called_with()
def test_on_close_ok(self):
self.c.collect = Mock(name='collect')
self.c._on_close_ok()
self.c.collect.assert_called_with()
def test_flow(self):
self.c.flow(0)
self.c.send_method.assert_called_with(
spec.Channel.Flow, 'b', (0,), wait=spec.Channel.FlowOk,
)
def test_on_flow(self):
self.c._x_flow_ok = Mock(name='_x_flow_ok')
self.c._on_flow(0)
self.assertFalse(self.c.active)
self.c._x_flow_ok.assert_called_with(0)
def test_x_flow_ok(self):
self.c._x_flow_ok(1)
self.c.send_method.assert_called_with(spec.Channel.FlowOk, 'b', (1,))
def test_open(self):
self.c.is_open = True
self.c.open()
self.c.is_open = False
self.c.open()
self.c.send_method.assert_called_with(
spec.Channel.Open, 's', ('',), wait=spec.Channel.OpenOk,
)
def test_on_open_ok(self):
self.c.on_open = Mock(name='on_open')
self.c.is_open = False
self.c._on_open_ok()
self.assertTrue(self.c.is_open)
self.c.on_open.assert_called_with(self.c)
def test_exchange_declare(self):
self.c.exchange_declare(
'foo', 'direct', False, True,
auto_delete=False, nowait=False, arguments={'x': 1},
)
self.c.send_method.assert_called_with(
spec.Exchange.Declare, 'BssbbbbbF',
(0, 'foo', 'direct', False, True, False,
False, False, {'x': 1}),
wait=spec.Exchange.DeclareOk,
)
@patch('amqp.channel.warn')
def test_exchange_declare__auto_delete(self, warn):
self.c.exchange_declare(
'foo', 'direct', False, True,
auto_delete=True, nowait=False, arguments={'x': 1},
)
self.assertTrue(warn.called)
def test_exchange_delete(self):
self.c.exchange_delete('foo')
self.c.send_method.assert_called_with(
spec.Exchange.Delete, 'Bsbb',
(0, 'foo', False, False),
wait=spec.Exchange.DeleteOk,
)
def test_exchange_bind(self):
self.c.exchange_bind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Bind, 'BsssbF',
(0, 'dest', 'source', 'rkey', False, {'x': 1}),
wait=spec.Exchange.BindOk,
)
def test_exchange_unbind(self):
self.c.exchange_unbind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Unbind, 'BsssbF',
(0, 'dest', 'source', 'rkey', False, {'x': 1}),
wait=spec.Exchange.UnbindOk,
)
def test_queue_bind(self):
self.c.queue_bind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Bind, 'BsssbF',
(0, 'q', 'ex', 'rkey', False, {'x': 1}),
wait=spec.Queue.BindOk,
)
def test_queue_unbind(self):
self.c.queue_unbind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Unbind, 'BsssF',
(0, 'q', 'ex', 'rkey', {'x': 1}),
wait=spec.Queue.UnbindOk,
)
def test_queue_declare(self):
self.c.queue_declare('q', False, True, False, False, True, {'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Declare, 'BsbbbbbF',
(0, 'q', False, True, False, False, True, {'x': 1}),
)
def test_queue_declare__sync(self):
self.c.wait = Mock(name='wait')
self.c.wait.return_value = ('name', 123, 45)
ret = self.c.queue_declare(
'q', False, True, False, False, False, {'x': 1},
)
self.c.send_method.assert_called_with(
spec.Queue.Declare, 'BsbbbbbF',
(0, 'q', False, True, False, False, False, {'x': 1}),
)
self.assertEqual(ret.queue, 'name')
self.assertEqual(ret.message_count, 123)
self.assertEqual(ret.consumer_count, 45)
self.c.wait.assert_called_with(
spec.Queue.DeclareOk, returns_tuple=True)
def test_queue_delete(self):
self.c.queue_delete('q')
self.c.send_method.assert_called_with(
spec.Queue.Delete, 'Bsbbb',
(0, 'q', False, False, False),
wait=spec.Queue.DeleteOk,
)
def test_queue_purge(self):
self.c.queue_purge('q')
self.c.send_method.assert_called_with(
spec.Queue.Purge, 'Bsb', (0, 'q', False),
wait=spec.Queue.PurgeOk,
)
def test_basic_ack(self):
self.c.basic_ack(123, multiple=1)
self.c.send_method.assert_called_with(
spec.Basic.Ack, 'Lb', (123, 1),
)
def test_basic_cancel(self):
self.c.basic_cancel(123)
self.c.send_method.assert_called_with(
spec.Basic.Cancel, 'sb', (123, False),
wait=spec.Basic.CancelOk,
)
self.c.connection = None
self.c.basic_cancel(123)
def test_on_basic_cancel(self):
self.c._remove_tag = Mock(name='_remove_tag')
self.c._on_basic_cancel(123)
self.c._remove_tag.return_value.assert_called_with(123)
self.c._remove_tag.return_value = None
with self.assertRaises(ConsumerCancelled):
self.c._on_basic_cancel(123)
def test_on_basic_cancel_ok(self):
self.c._remove_tag = Mock(name='remove_tag')
self.c._on_basic_cancel_ok(123)
self.c._remove_tag.assert_called_with(123)
def test_remove_tag(self):
self.c.callbacks[123] = Mock()
p = self.c.cancel_callbacks[123] = Mock()
self.assertIs(self.c._remove_tag(123), p)
self.assertNotIn(123, self.c.callbacks)
self.assertNotIn(123, self.c.cancel_callbacks)
def test_basic_consume(self):
callback = Mock()
on_cancel = Mock()
self.c.basic_consume(
'q', 123, arguments={'x': 1},
callback=callback,
on_cancel=on_cancel,
)
self.c.send_method.assert_called_with(
spec.Basic.Consume, 'BssbbbbF',
(0, 'q', 123, False, False, False, False, {'x': 1}),
wait=spec.Basic.ConsumeOk,
)
self.assertIs(self.c.callbacks[123], callback)
self.assertIs(self.c.cancel_callbacks[123], on_cancel)
def test_basic_consume__no_ack(self):
self.c.basic_consume(
'q', 123, arguments={'x': 1}, no_ack=True,
)
self.assertIn(123, self.c.no_ack_consumers)
def test_on_basic_deliver(self):
msg = Mock()
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback = self.c.callbacks[123] = Mock(name='cb')
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback.assert_called_with(msg)
def test_basic_get(self):
self.c._on_get_empty = Mock()
self.c._on_get_ok = Mock()
self.c.send_method.return_value = ('cluster_id',)
self.c.basic_get('q')
self.c.send_method.assert_called_with(
spec.Basic.Get, 'Bsb', (0, 'q', False),
wait=[spec.Basic.GetOk, spec.Basic.GetEmpty], returns_tuple=True,
)
self.c._on_get_empty.assert_called_with('cluster_id')
self.c.send_method.return_value = (
'dtag', 'redelivered', 'ex', 'rkey', 'mcount', 'msg',
)
self.c.basic_get('q')
self.c._on_get_ok.assert_called_with(
'dtag', 'redelivered', 'ex', 'rkey', 'mcount', 'msg',
)
def test_on_get_empty(self):
self.c._on_get_empty(1)
def test_on_get_ok(self):
msg = Mock()
m = self.c._on_get_ok(
'dtag', 'redelivered', 'ex', 'rkey', 'mcount', msg,
)
self.assertIs(m, msg)
def test_basic_publish(self):
self.c.connection.transport.having_timeout = ContextMock()
self.c._basic_publish('msg', 'ex', 'rkey')
self.c.send_method.assert_called_with(
spec.Basic.Publish, 'Bssbb',
(0, 'ex', 'rkey', False, False), 'msg',
)
def test_basic_publish_confirm(self):
self.c._confirm_selected = False
self.c.confirm_select = Mock(name='confirm_select')
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait')
ret = self.c.basic_publish_confirm(1, 2, arg=1)
self.c.confirm_select.assert_called_with()
self.assertTrue(self.c._confirm_selected)
self.c._basic_publish.assert_called_with(1, 2, arg=1)
self.assertIs(ret, self.c._basic_publish())
self.c.wait.assert_called_with(spec.Basic.Ack)
self.c.basic_publish_confirm(1, 2, arg=1)
def test_basic_qos(self):
self.c.basic_qos(0, 123, False)
self.c.send_method.assert_called_with(
spec.Basic.Qos, 'lBb', (0, 123, False),
wait=spec.Basic.QosOk,
)
def test_basic_recover(self):
self.c.basic_recover(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Recover, 'b', (True,),
)
def test_basic_recover_async(self):
self.c.basic_recover_async(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.RecoverAsync, 'b', (True,),
)
def test_basic_reject(self):
self.c.basic_reject(123, requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Reject, 'Lb', (123, True),
)
def test_on_basic_return(self):
with self.assertRaises(NotFound):
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
@patch('amqp.channel.error_for_code')
def test_on_basic_return__handled(self, error_for_code):
callback = Mock(name='callback')
self.c.events['basic_return'].add(callback)
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
callback.assert_called_with(
error_for_code(), 'ex', 'rkey', 'msg',
)
def test_tx_commit(self):
self.c.tx_commit()
self.c.send_method.assert_called_with(
spec.Tx.Commit, wait=spec.Tx.CommitOk,
)
def test_tx_rollback(self):
self.c.tx_rollback()
self.c.send_method.assert_called_with(
spec.Tx.Rollback, wait=spec.Tx.RollbackOk,
)
def test_tx_select(self):
self.c.tx_select()
self.c.send_method.assert_called_with(
spec.Tx.Select, wait=spec.Tx.SelectOk,
)
def test_confirm_select(self):
self.c.confirm_select()
self.c.send_method.assert_called_with(
spec.Confirm.Select, 'b', (False,),
wait=spec.Confirm.SelectOk,
)
def test_on_basic_ack(self):
callback = Mock(name='callback')
self.c.events['basic_ack'].add(callback)
self.c._on_basic_ack(123, True)
callback.assert_called_with(123, True)
class test_Channel:
@pytest.fixture(autouse=True)
def setup_conn(self):
self.conn = MagicMock(name='connection')
self.conn.is_closing = False
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
assert c.basic_publish == c.basic_publish_confirm
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
assert c.basic_publish == c._basic_publish
def test_init_auto_channel(self):
c = Channel(self.conn, None)
self.conn._get_free_channel_id.assert_called_with()
assert c.channel_id is self.conn._get_free_channel_id()
def test_init_explicit_channel(self):
Channel(self.conn, 3)
self.conn._claim_channel_id.assert_called_with(3)
def test_then(self):
self.c.on_open = Mock(name='on_open')
on_success = Mock(name='on_success')
on_error = Mock(name='on_error')
self.c.then(on_success, on_error)
self.c.on_open.then.assert_called_with(on_success, on_error)
def test_collect(self):
self.c.callbacks[(50, 61)] = Mock()
self.c.cancel_callbacks['foo'] = Mock()
self.c.events['bar'].add(Mock())
self.c.no_ack_consumers.add('foo')
self.c.collect()
assert not self.c.callbacks
assert not self.c.cancel_callbacks
assert not self.c.events
assert not self.c.no_ack_consumers
assert not self.c.is_open
self.c.collect()
def test_do_revive(self):
self.c.open = Mock(name='open')
self.c.is_open = True
self.c._do_revive()
assert not self.c.is_open
self.c.open.assert_called_with()
def test_close__not_open(self):
self.c.is_open = False
self.c.close()
def test_close__no_connection(self):
self.c.connection = None
self.c.close()
def test_close(self):
self.c.is_open = True
self.c.close(30, 'text', spec.Queue.Declare)
self.c.send_method.assert_called_with(
spec.Channel.Close, 'BsBB',
(30, 'text', spec.Queue.Declare[0], spec.Queue.Declare[1]),
wait=spec.Channel.CloseOk,
)
assert self.c.connection is None
def test_on_close(self):
self.c._do_revive = Mock(name='_do_revive')
with pytest.raises(NotFound):
self.c._on_close(404, 'text', 50, 61)
self.c.send_method.assert_called_with(spec.Channel.CloseOk)
self.c._do_revive.assert_called_with()
def test_on_close_ok(self):
self.c.collect = Mock(name='collect')
self.c._on_close_ok()
self.c.collect.assert_called_with()
def test_flow(self):
self.c.flow(0)
self.c.send_method.assert_called_with(
spec.Channel.Flow, 'b', (0,), wait=spec.Channel.FlowOk,
)
def test_on_flow(self):
self.c._x_flow_ok = Mock(name='_x_flow_ok')
self.c._on_flow(0)
assert not self.c.active
self.c._x_flow_ok.assert_called_with(0)
def test_x_flow_ok(self):
self.c._x_flow_ok(1)
self.c.send_method.assert_called_with(spec.Channel.FlowOk, 'b', (1,))
def test_open(self):
self.c.is_open = True
self.c.open()
self.c.is_open = False
self.c.open()
self.c.send_method.assert_called_with(
spec.Channel.Open, 's', ('',), wait=spec.Channel.OpenOk,
)
def test_on_open_ok(self):
self.c.on_open = Mock(name='on_open')
self.c.is_open = False
self.c._on_open_ok()
assert self.c.is_open
self.c.on_open.assert_called_with(self.c)
def test_exchange_declare(self):
self.c.exchange_declare(
'foo', 'direct', False, True,
auto_delete=False, nowait=False, arguments={'x': 1},
)
self.c.send_method.assert_called_with(
spec.Exchange.Declare, 'BssbbbbbF',
(0, 'foo', 'direct', False, True, False,
False, False, {'x': 1}),
wait=spec.Exchange.DeclareOk,
)
def test_exchange_declare__auto_delete(self):
with patch('amqp.channel.warn') as warn:
self.c.exchange_declare(
'foo', 'direct', False, True,
auto_delete=True, nowait=False, arguments={'x': 1},
)
warn.assert_called()
def test_exchange_delete(self):
self.c.exchange_delete('foo')
self.c.send_method.assert_called_with(
spec.Exchange.Delete, 'Bsbb',
(0, 'foo', False, False),
wait=spec.Exchange.DeleteOk,
)
def test_exchange_bind(self):
self.c.exchange_bind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Bind, 'BsssbF',
(0, 'dest', 'source', 'rkey', False, {'x': 1}),
wait=spec.Exchange.BindOk,
)
def test_exchange_unbind(self):
self.c.exchange_unbind('dest', 'source', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Exchange.Unbind, 'BsssbF',
(0, 'dest', 'source', 'rkey', False, {'x': 1}),
wait=spec.Exchange.UnbindOk,
)
def test_queue_bind(self):
self.c.queue_bind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Bind, 'BsssbF',
(0, 'q', 'ex', 'rkey', False, {'x': 1}),
wait=spec.Queue.BindOk,
)
def test_queue_unbind(self):
self.c.queue_unbind('q', 'ex', 'rkey', arguments={'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Unbind, 'BsssF',
(0, 'q', 'ex', 'rkey', {'x': 1}),
wait=spec.Queue.UnbindOk,
)
def test_queue_declare(self):
self.c.queue_declare('q', False, True, False, False, True, {'x': 1})
self.c.send_method.assert_called_with(
spec.Queue.Declare, 'BsbbbbbF',
(0, 'q', False, True, False, False, True, {'x': 1}),
)
def test_queue_declare__sync(self):
self.c.wait = Mock(name='wait')
self.c.wait.return_value = ('name', 123, 45)
ret = self.c.queue_declare(
'q', False, True, False, False, False, {'x': 1},
)
self.c.send_method.assert_called_with(
spec.Queue.Declare, 'BsbbbbbF',
(0, 'q', False, True, False, False, False, {'x': 1}),
)
assert ret.queue == 'name'
assert ret.message_count == 123
assert ret.consumer_count == 45
self.c.wait.assert_called_with(
spec.Queue.DeclareOk, returns_tuple=True)
def test_queue_delete(self):
self.c.queue_delete('q')
self.c.send_method.assert_called_with(
spec.Queue.Delete, 'Bsbbb',
(0, 'q', False, False, False),
wait=spec.Queue.DeleteOk,
)
def test_queue_purge(self):
self.c.queue_purge('q')
self.c.send_method.assert_called_with(
spec.Queue.Purge, 'Bsb', (0, 'q', False),
wait=spec.Queue.PurgeOk,
)
def test_basic_ack(self):
self.c.basic_ack(123, multiple=1)
self.c.send_method.assert_called_with(
spec.Basic.Ack, 'Lb', (123, 1),
)
def test_basic_cancel(self):
self.c.basic_cancel(123)
self.c.send_method.assert_called_with(
spec.Basic.Cancel, 'sb', (123, False),
wait=spec.Basic.CancelOk,
)
self.c.connection = None
self.c.basic_cancel(123)
def test_on_basic_cancel(self):
self.c._remove_tag = Mock(name='_remove_tag')
self.c._on_basic_cancel(123)
self.c._remove_tag.return_value.assert_called_with(123)
self.c._remove_tag.return_value = None
with pytest.raises(ConsumerCancelled):
self.c._on_basic_cancel(123)
def test_on_basic_cancel_ok(self):
self.c._remove_tag = Mock(name='remove_tag')
self.c._on_basic_cancel_ok(123)
self.c._remove_tag.assert_called_with(123)
def test_remove_tag(self):
self.c.callbacks[123] = Mock()
p = self.c.cancel_callbacks[123] = Mock()
assert self.c._remove_tag(123) is p
assert 123 not in self.c.callbacks
assert 123 not in self.c.cancel_callbacks
def test_basic_consume(self):
callback = Mock()
on_cancel = Mock()
self.c.send_method.return_value = (123, )
self.c.basic_consume(
'q', 123, arguments={'x': 1},
callback=callback,
on_cancel=on_cancel,
)
self.c.send_method.assert_called_with(
spec.Basic.Consume, 'BssbbbbF',
(0, 'q', 123, False, False, False, False, {'x': 1}),
wait=spec.Basic.ConsumeOk,
returns_tuple=True
)
assert self.c.callbacks[123] is callback
assert self.c.cancel_callbacks[123] is on_cancel
def test_basic_consume__no_ack(self):
self.c.send_method.return_value = (123,)
self.c.basic_consume(
'q', 123, arguments={'x': 1}, no_ack=True,
)
assert 123 in self.c.no_ack_consumers
def test_basic_consume_no_consumer_tag(self):
callback = Mock()
self.c.send_method.return_value = (123,)
ret = self.c.basic_consume(
'q', arguments={'x': 1},
callback=callback,
)
self.c.send_method.assert_called_with(
spec.Basic.Consume, 'BssbbbbF',
(0, 'q', '', False, False, False, False, {'x': 1}),
wait=spec.Basic.ConsumeOk,
returns_tuple=True
)
assert self.c.callbacks[123] is callback
assert ret == 123
def test_basic_consume_no_wait(self):
callback = Mock()
ret_promise = promise()
self.c.send_method.return_value = ret_promise
ret = self.c.basic_consume(
'q', 123, arguments={'x': 1},
callback=callback, nowait=True
)
self.c.send_method.assert_called_with(
spec.Basic.Consume, 'BssbbbbF',
(0, 'q', 123, False, False, False, True, {'x': 1}),
wait=None,
returns_tuple=True
)
assert self.c.callbacks[123] is callback
assert ret == ret_promise
def test_basic_consume_no_wait_no_consumer_tag(self):
callback = Mock()
with pytest.raises(ValueError):
self.c.basic_consume(
'q', arguments={'x': 1},
callback=callback, nowait=True
)
assert 123 not in self.c.callbacks
def test_on_basic_deliver(self):
msg = Message()
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback = self.c.callbacks[123] = Mock(name='cb')
self.c._on_basic_deliver(123, '321', False, 'ex', 'rkey', msg)
callback.assert_called_with(msg)
assert msg.channel == self.c
assert msg.delivery_info == {
'consumer_tag': 123,
'delivery_tag': '321',
'redelivered': False,
'exchange': 'ex',
'routing_key': 'rkey',
}
def test_basic_get(self):
self.c._on_get_empty = Mock()
self.c._on_get_ok = Mock()
self.c.send_method.return_value = ('cluster_id',)
self.c.basic_get('q')
self.c.send_method.assert_called_with(
spec.Basic.Get, 'Bsb', (0, 'q', False),
wait=[spec.Basic.GetOk, spec.Basic.GetEmpty], returns_tuple=True,
)
self.c._on_get_empty.assert_called_with('cluster_id')
self.c.send_method.return_value = (
'dtag', 'redelivered', 'ex', 'rkey', 'mcount', 'msg',
)
self.c.basic_get('q')
self.c._on_get_ok.assert_called_with(
'dtag', 'redelivered', 'ex', 'rkey', 'mcount', 'msg',
)
def test_on_get_empty(self):
self.c._on_get_empty(1)
def test_on_get_ok(self):
msg = Message()
m = self.c._on_get_ok(
'dtag', 'redelivered', 'ex', 'rkey', 'mcount', msg,
)
assert m is msg
assert m.channel == self.c
assert m.delivery_info == {
'delivery_tag': 'dtag',
'redelivered': 'redelivered',
'exchange': 'ex',
'routing_key': 'rkey',
'message_count': 'mcount',
}
def test_basic_publish(self):
self.c.connection.transport.having_timeout = ContextMock()
self.c._basic_publish('msg', 'ex', 'rkey')
self.c.send_method.assert_called_with(
spec.Basic.Publish, 'Bssbb',
(0, 'ex', 'rkey', False, False), 'msg',
)
def test_basic_publish_confirm(self):
self.c._confirm_selected = False
self.c.confirm_select = Mock(name='confirm_select')
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait')
ret = self.c.basic_publish_confirm(1, 2, arg=1)
self.c.confirm_select.assert_called_with()
assert self.c._confirm_selected
self.c._basic_publish.assert_called_with(1, 2, arg=1)
assert ret is self.c._basic_publish()
self.c.wait.assert_called_with(
[spec.Basic.Ack, spec.Basic.Nack], callback=ANY
)
self.c.basic_publish_confirm(1, 2, arg=1)
def test_basic_publish_confirm_nack(self):
# test checking whether library is handling correctly Nack confirms
# sent from RabbitMQ. Library must raise MessageNacked when server
# sent Nack message.
# Nack frame construction
args = dumps('Lb', (1, False))
frame = (b''.join([pack('>HH', *spec.Basic.Nack), args]))
def wait(method, *args, **kwargs):
# Simple mock simulating registering callbacks of real wait method
for m in method:
self.c._pending[m] = kwargs['callback']
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait', side_effect=wait)
self.c.basic_publish_confirm(1, 2, arg=1)
with pytest.raises(MessageNacked):
# Inject Nack to message handler
self.c.dispatch_method(
spec.Basic.Nack, frame, None
)
def test_basic_publish_connection_blocked(self):
# Basic test checking that drain_events() is called
# before publishing message and send_method() is called
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_called_once_with(timeout=0)
self.c.send_method.assert_called_once_with(
spec.Basic.Publish, 'Bssbb',
(0, 'ex', 'rkey', False, False), 'msg',
)
self.c.send_method.reset_mock()
# Basic test checking that socket.timeout exception
# is ignored and send_method() is called.
self.conn.drain_events.side_effect = socket.timeout
self.c._basic_publish('msg', 'ex', 'rkey')
self.c.send_method.assert_called_once_with(
spec.Basic.Publish, 'Bssbb',
(0, 'ex', 'rkey', False, False), 'msg',
)
def test_basic_publish_connection_blocked_not_supported(self):
# Test veryfying that when server does not have
# connection.blocked capability, drain_events() are not called
self.conn.client_properties = {
'capabilities': {
'connection.blocked': False
}
}
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_not_called()
self.c.send_method.assert_called_once_with(
spec.Basic.Publish, 'Bssbb',
(0, 'ex', 'rkey', False, False), 'msg',
)
def test_basic_publish_connection_blocked_not_supported_missing(self):
# Test veryfying that when server does not have
# connection.blocked capability, drain_events() are not called
self.conn.client_properties = {
'capabilities': {}
}
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_not_called()
self.c.send_method.assert_called_once_with(
spec.Basic.Publish, 'Bssbb',
(0, 'ex', 'rkey', False, False), 'msg',
)
def test_basic_publish_connection_blocked_no_capabilities(self):
# Test veryfying that when server does not have
# support of capabilities, drain_events() are not called
self.conn.client_properties = {
}
self.c._basic_publish('msg', 'ex', 'rkey')
self.conn.drain_events.assert_not_called()
self.c.send_method.assert_called_once_with(
spec.Basic.Publish, 'Bssbb',
(0, 'ex', 'rkey', False, False), 'msg',
)
def test_basic_publish_confirm_callback(self):
def wait_nack(method, *args, **kwargs):
kwargs['callback'](spec.Basic.Nack)
def wait_ack(method, *args, **kwargs):
kwargs['callback'](spec.Basic.Ack)
self.c._basic_publish = Mock(name='_basic_publish')
self.c.wait = Mock(name='wait_nack', side_effect=wait_nack)
with pytest.raises(MessageNacked):
# when callback is called with spec.Basic.Nack it must raise
# MessageNacked exception
self.c.basic_publish_confirm(1, 2, arg=1)
self.c.wait = Mock(name='wait_ack', side_effect=wait_ack)
# when callback is called with spec.Basic.Ack
# it must nost raise exception
self.c.basic_publish_confirm(1, 2, arg=1)
def test_basic_publish_connection_closed(self):
self.c.collect()
with pytest.raises(RecoverableConnectionError) as excinfo:
self.c._basic_publish('msg', 'ex', 'rkey')
assert 'basic_publish: connection closed' in str(excinfo.value)
self.c.send_method.assert_not_called()
def test_basic_qos(self):
self.c.basic_qos(0, 123, False)
self.c.send_method.assert_called_with(
spec.Basic.Qos, 'lBb', (0, 123, False),
wait=spec.Basic.QosOk,
)
def test_basic_recover(self):
self.c.basic_recover(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Recover, 'b', (True,),
)
def test_basic_recover_async(self):
self.c.basic_recover_async(requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.RecoverAsync, 'b', (True,),
)
def test_basic_reject(self):
self.c.basic_reject(123, requeue=True)
self.c.send_method.assert_called_with(
spec.Basic.Reject, 'Lb', (123, True),
)
def test_on_basic_return(self):
with pytest.raises(NotFound):
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
def test_on_basic_return__handled(self):
with patch('amqp.channel.error_for_code') as error_for_code:
callback = Mock(name='callback')
self.c.events['basic_return'].add(callback)
self.c._on_basic_return(404, 'text', 'ex', 'rkey', 'msg')
callback.assert_called_with(
error_for_code(), 'ex', 'rkey', 'msg',
)
def test_tx_commit(self):
self.c.tx_commit()
self.c.send_method.assert_called_with(
spec.Tx.Commit, wait=spec.Tx.CommitOk,
)
def test_tx_rollback(self):
self.c.tx_rollback()
self.c.send_method.assert_called_with(
spec.Tx.Rollback, wait=spec.Tx.RollbackOk,
)
def test_tx_select(self):
self.c.tx_select()
self.c.send_method.assert_called_with(
spec.Tx.Select, wait=spec.Tx.SelectOk,
)
def test_confirm_select(self):
self.c.confirm_select()
self.c.send_method.assert_called_with(
spec.Confirm.Select, 'b', (False,),
wait=spec.Confirm.SelectOk,
)
def test_on_basic_ack(self):
callback = Mock(name='callback')
self.c.events['basic_ack'].add(callback)
self.c._on_basic_ack(123, True)
callback.assert_called_with(123, True)
def test_on_basic_nack(self):
callback = Mock(name='callback')
self.c.events['basic_nack'].add(callback)
self.c._on_basic_nack(123, True)
callback.assert_called_with(123, True)