def test_queue_declare(self):
# Test verifying declaring queue
frame_writer_cls_mock = Mock()
conn = Connection(frame_writer=frame_writer_cls_mock)
with patch.object(conn, 'Transport') as transport_mock:
handshake(conn, transport_mock)
ch = create_channel(1, conn, transport_mock)
transport_mock().read_frame.return_value = build_frame_type_1(
spec.Queue.DeclareOk,
channel=1,
arg_format='sll',
args=('foo', 1, 2)
)
frame_writer_mock = frame_writer_cls_mock()
frame_writer_mock.reset_mock()
ret = ch.queue_declare('foo')
assert ret == queue_declare_ok_t(
queue='foo', message_count=1, consumer_count=2
)
frame_writer_mock.assert_called_once_with(
1, 1, spec.Queue.Declare,
dumps(
'BsbbbbbF',
(
0,
# queue, passive, durable, exclusive,
'foo', False, False, False,
# auto_delete, nowait, arguments
True, False, None
)
),
None
)
def queue_declare(
self, queue="", passive=False, durable=False, exclusive=False, auto_delete=False, arguments=None, nowait=False
):
return queue_declare_ok_t(
*self.connection._queue_declare(
self.channel_id, queue, passive, durable, exclusive, auto_delete, arguments or {}
)
)
def queue_declare(self, queue='', passive=False, durable=False,
exclusive=False, auto_delete=False, arguments=None,
nowait=False):
return queue_declare_ok_t(
*self.connection._queue_declare(
self.channel_id, queue, passive, durable,
exclusive, auto_delete, arguments or {},
)
)
def queue_declare(self, queue=None, passive=False, **kwargs):
"""Declare queue."""
queue = queue or 'amq.gen-%s' % uuid()
if passive and not self._has_queue(queue, **kwargs):
raise ChannelError(
'NOT_FOUND - no queue {0!r} in vhost {1!r}'.format(
queue, self.connection.client.virtual_host or '/'),
(50, 10), 'Channel.queue_declare', '404',
)
else:
self._new_queue(queue, **kwargs)
return queue_declare_ok_t(queue, self._size(queue), 0)
def queue_declare(self, queue=None, passive=False, **kwargs):
"""Declare queue."""
queue = queue or 'amq.gen-%s' % uuid()
if passive and not self._has_queue(queue, **kwargs):
raise ChannelError(
'NOT_FOUND - no queue {0!r} in vhost {1!r}'.format(
queue, self.connection.client.virtual_host or '/'),
(50, 10), 'Channel.queue_declare', '404',
)
else:
self._new_queue(queue, **kwargs)
return queue_declare_ok_t(queue, self._size(queue), 0)
