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)