def test_subscribe_to_single_queue(self, mock_stomp_subscribe,
mock_stomp_set_listener):
"""
Test: subscribe_queues handles a single queue (non-list)
and calls stomp.subscribe_queues for it
When: subscribe_queues is called a single queue passed as queue_list
"""
client = QueueClient()
client.connect()
client.subscribe_queues('single-queue', 'consumer', None)
mock_stomp_set_listener.assert_called_once_with('consumer', None)
mock_stomp_subscribe.assert_called_once()
def test_subscribe_to_single_queue(self, mock_stomp_subscribe, mock_stomp_set_listener):
"""
Test: subscribe_queues handles a single queue (non-list)
and calls stomp.subscribe_queues for it
When: subscribe_queues is called a single queue passed as queue_list
"""
client = QueueClient()
client.connect()
client.subscribe_queues('single-queue', 'consumer', None, 'auto')
mock_stomp_set_listener.assert_called_once_with('consumer', None)
test_expected_args = {'destination': 'single-queue',
'id': '1',
'ack': 'auto',
'header': {'activemq.prefetchSize': '1'}}
mock_stomp_subscribe.assert_called_once_with(**test_expected_args)
def test_subscribe_to_queue_list(self, mock_stomp_subscribe, mock_stomp_set_listener):
"""
Test: subscribe_queues calls stomp.subscribe_queues twice, once for each queue given
When: subscribe_queues is called with a queue_list length of 2
"""
client = QueueClient()
client.connect()
client.subscribe_queues(['test', 'queues'], 'consumer', None, 'auto')
mock_stomp_set_listener.assert_called_once_with('consumer', None)
test_expected_args = {'destination': 'test',
'id': '1',
'ack': 'auto',
'header': {'activemq.prefetchSize': '1'}}
queue_expected_args = {'destination': 'queues',
'id': '1',
'ack': 'auto',
'header': {'activemq.prefetchSize': '1'}}
mock_stomp_subscribe.assert_has_calls([call(**test_expected_args),
call(**queue_expected_args)])