def listen(self, target):
listener = pika_drv_poller.RpcServicePikaPoller(
self._pika_engine,
target,
prefetch_count=self._pika_engine.rpc_listener_prefetch_count)
listener.start()
return listener
def listen(self, target, batch_size, batch_timeout):
listener = pika_drv_poller.RpcServicePikaPoller(
self._pika_engine,
target,
prefetch_count=self._pika_engine.rpc_listener_prefetch_count)
listener.start()
return base.PollStyleListenerAdapter(listener, batch_size,
batch_timeout)
def test_declare_rpc_queue_bindings(self, rpc_pika_incoming_message_mock):
poller = pika_poller.RpcServicePikaPoller(self._pika_engine,
self._target, 1, None,
self._prefetch_count)
poller.start(None)
self.assertTrue(self._pika_engine.create_connection.called)
self.assertTrue(self._poller_connection_mock.channel.called)
declare_queue_binding_by_channel_mock = (
self._pika_engine.declare_queue_binding_by_channel)
self.assertEqual(declare_queue_binding_by_channel_mock.call_count, 6)
declare_queue_binding_by_channel_mock.assert_has_calls(
(mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange",
exchange_type='direct',
queue="topic_None_True_False",
queue_expiration=12345,
routing_key="topic_None_True_False"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange",
exchange_type='direct',
queue="topic_server_True_False",
queue_expiration=12345,
routing_key="topic_server_True_False"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange",
exchange_type='direct',
queue="topic_server_True_True",
queue_expiration=12345,
routing_key="topic_all_workers_True_False"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange",
exchange_type='direct',
queue="topic_None_False_False",
queue_expiration=12345,
routing_key="topic_None_False_False"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange",
exchange_type='direct',
queue="topic_server_False_False",
queue_expiration=12345,
routing_key='topic_server_False_False'),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange",
exchange_type='direct',
queue="topic_server_False_True",
queue_expiration=12345,
routing_key='topic_all_workers_False_False')))
def listen(self, target, batch_size, batch_timeout):
return pika_drv_poller.RpcServicePikaPoller(
self._pika_engine, target, batch_size, batch_timeout,
self._pika_engine.rpc_listener_prefetch_count)
def test_declare_rpc_queue_bindings(self, rpc_pika_incoming_message_mock):
poller = pika_poller.RpcServicePikaPoller(
self._pika_engine,
self._target,
self._prefetch_count,
)
self._poller_connection_mock.process_data_events.side_effect = (
lambda time_limit: poller._on_message_with_ack_callback(
None, None, None, None))
poller.start()
res = poller.poll()
self.assertEqual(len(res), 1)
self.assertEqual(res[0], rpc_pika_incoming_message_mock.return_value)
self.assertTrue(self._pika_engine.create_connection.called)
self.assertTrue(self._poller_connection_mock.channel.called)
declare_queue_binding_by_channel_mock = (
self._pika_engine.declare_queue_binding_by_channel)
self.assertEqual(declare_queue_binding_by_channel_mock.call_count, 6)
declare_queue_binding_by_channel_mock.assert_has_calls((
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange_topic_False_True",
exchange_type='direct',
queue="topic_None_True",
queue_expiration=12345,
routing_key="topic_None_True"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange_topic_False_True",
exchange_type='direct',
queue="topic_server_True",
queue_expiration=12345,
routing_key="topic_server_True"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange_topic_True_True",
exchange_type='fanout',
queue="topic_server_True",
queue_expiration=12345,
routing_key=''),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange_topic_False_False",
exchange_type='direct',
queue="topic_None_False",
queue_expiration=12345,
routing_key="topic_None_False"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange_topic_False_False",
exchange_type='direct',
queue="topic_server_False",
queue_expiration=12345,
routing_key="topic_server_False"),
mock.call(channel=self._poller_channel_mock,
durable=False,
exchange="exchange_topic_True_False",
exchange_type='fanout',
queue="topic_server_False",
queue_expiration=12345,
routing_key=''),
))