def configure(self):
self.consumer = MagicMock()
self.ctx.Message.return_value = self.message = NonCallableMagicMock()
self.agent = ConsumerAgent(self.consumer, sentinel.broker,
sentinel.bindings)
self.agent.acknowledge = MagicMock()
self.agent._process = MagicMock()
def configure(self):
self.binding = {
'queue': sentinel.queue,
'exchange': sentinel.exchange,
'routing_key': sentinel.routing_key,
}
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.declare_exchange = MagicMock()
def configure(self):
self.kwargs = {
'queue': sentinel.queue,
'exchange': sentinel.exchange,
'routing_key': sentinel.routing_key,
'method_frame': sentinel.method_frame
}
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.channel = MagicMock()
def configure(self):
self.consumer = MagicMock()
self.consumer.process.side_effect = self.exc
self.message = NonCallableMagicMock()
self.agent = ConsumerAgent(self.consumer, sentinel.broker,
sentinel.bindings)
self.agent.reject = MagicMock()
self.agent.stop = MagicMock()
self.agent._record_exception = MagicMock()
self.agent.processing_failure = MagicMock()
self.agent.reconnect = MagicMock()
def configure(self):
self.bindings = (
{
'queue': sentinel.queue1,
'exchange': sentinel.exchange1,
'routing_key': sentinel.routing_key1,
},
{
'queue': sentinel.queue2,
'exchange': sentinel.exchange2,
'routing_key': sentinel.routing_key2,
},
)
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
self.bindings)
self.agent.create_binding = MagicMock()
def configure(self):
self.consumer = MagicMock()
self.message = NonCallableMagicMock()
self.agent = ConsumerAgent(self.consumer, sentinel.broker,
sentinel.bindings)
def configure(self):
self.broker = MagicMock()
self.broker.connect.return_value = sentinel.connection
self.agent = ConsumerAgent(sentinel.consumer, self.broker,
sentinel.bindings)
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.add_on_cancel_callback = MagicMock()
self.agent.channel = MagicMock()
self.agent.channel.basic_consume.return_value = sentinel.consumer_tag
def execute(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings, False, sentinel.config)
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.add_on_connection_close_callback = MagicMock()
self.agent.open_channel = MagicMock()
def should_have_default_config(self):
agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.assertEqual(agent.config, {})
def configure(self):
self.options = {'durable': True, 'arguments': {'x-ha-policy': 'all'}}
config = {'queues': {sentinel.queue: self.options}}
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings, False, config)
self.agent.channel = MagicMock()
def configure(self):
self.options = {'exchange_type': 'topic', 'durable': True}
config = {'exchanges': {sentinel.exchange: self.options}}
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings, False, config)
self.agent.channel = MagicMock()
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.add_on_channel_close_callback = MagicMock()
self.agent.create_bindings = MagicMock()
def configure(self):
self.message = MagicMock()
self.message.delivery_tag = sentinel.delivery_tag
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.channel = MagicMock()
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent._consumer_tags = {'your.queue': sentinel.consumer_tag}
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.connect = MagicMock()
self.agent.connection = MagicMock()
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.is_consuming_from = MagicMock(return_value=True)
self.agent.start_consuming = MagicMock()
def consumer_agent_from_config(config,
name,
broker='default',
section='rabbitmq'):
"""
Create a :class:`pikachewie.agent.ConsumerAgent` from the given `config`.
The `config` dict should have a structure similar to the following
example::
{
'rabbitmq': {
'brokers': {
'default': {
'nodes': {
'rabbit1': {
'host': 'rabbit1.example.com',
'port': 5672,
},
'rabbit2': {
'host': 'rabbit2.example.com',
'port': 5672,
},
},
'virtual_host': '/integration',
'heartbeat_interval': 60,
},
},
'consumers': {
'message_logger': {
'class': 'my.consumers.LoggingConsumer',
'arguments': {
'level': 'debug',
},
'bindings': [
{
'exchange': 'message',
'queue': 'text',
'routing_key': 'example.text.#',
},
],
},
},
'exchanges': {
'message': {
'exchange_type': 'topic',
'durable': True,
'auto_delete': False,
},
},
'queues': {
'text': {
'durable': True,
'exclusive': False,
'arguments': {
'x-dead-letter-exchange': 'dead.letters',
'x-dead-letter-routing-key': 'omg.such.rejection',
'x-ha-policy': 'all',
'x-message-ttl': 1800000
},
},
},
},
}
"""
consumer_config = config[section]['consumers'][name]
consumer = consumer_from_config(consumer_config)
broker = broker_from_config(config[section]['brokers'][broker])
no_ack = consumer_config.get('no_ack', False)
return ConsumerAgent(consumer, broker, consumer_config['bindings'], no_ack,
config['rabbitmq'])
def execute(self):
self.agent = ConsumerAgent(self.consumer, self.broker, self.bindings)
