def test_passes_connection_to_consumer(self, fake_realtime, fake_consumer, queue): messages.process_messages({}, 'foobar', queue, raise_error=False) fake_consumer.assert_called_once_with(connection=fake_realtime.get_connection.return_value, routing_key=mock.ANY, handler=mock.ANY, sentry_client=mock.ANY)
def test_passes_routing_key_to_consumer(self, fake_consumer, queue): messages.process_messages({}, 'foobar', queue, raise_error=False) fake_consumer.assert_called_once_with(connection=mock.ANY, routing_key='foobar', handler=mock.ANY, sentry_client=mock.ANY)
def test_passes_connection_to_consumer(self, fake_realtime, fake_consumer, queue): messages.process_messages({}, "foobar", queue, raise_error=False) fake_consumer.assert_called_once_with( connection=fake_realtime.get_connection.return_value, routing_key=Any(), handler=Any(), )
def test_message_handler_puts_message_on_queue(self, fake_consumer, queue): messages.process_messages({}, 'foobar', queue, raise_error=False) message_handler = fake_consumer.call_args[1]['handler'] message_handler({'foo': 'bar'}) result = queue.get_nowait() assert result.topic == 'foobar' assert result.payload == {'foo': 'bar'}
def test_message_handler_puts_message_on_queue(self, fake_consumer, queue): messages.process_messages({}, "foobar", queue, raise_error=False) message_handler = fake_consumer.call_args[1]["handler"] message_handler({"foo": "bar"}) result = queue.get_nowait() assert result.topic == "foobar" assert result.payload == {"foo": "bar"}
def test_passes_stats_client_to_consumer(self, fake_stats, fake_consumer, queue): messages.process_messages({}, "foobar", queue, raise_error=False) fake_consumer.assert_called_once_with( connection=mock.ANY, routing_key=mock.ANY, handler=mock.ANY, statsd_client=fake_stats.get_client.return_value, )
def test_passes_routing_key_to_consumer(self, fake_consumer, queue): messages.process_messages({}, "foobar", queue, raise_error=False) fake_consumer.assert_called_once_with( connection=Any(), routing_key="foobar", handler=Any(), statsd_client=Any(), )
def test_it_creates_and_runs_a_consumer(self, Consumer, realtime, work_queue): settings = {} messages.process_messages(settings, "routing_key", work_queue, raise_error=False) realtime.get_connection.assert_called_once_with(settings) Consumer.assert_called_once_with( connection=realtime.get_connection.return_value, routing_key="routing_key", handler=Any(), ) consumer = Consumer.return_value consumer.run.assert_called_once_with()
def test_runs_consumer(self, fake_consumer, queue): messages.process_messages({}, "foobar", queue, raise_error=False) consumer = fake_consumer.return_value consumer.run.assert_called_once_with()
def test_initializes_new_connection(self, fake_realtime, fake_consumer, queue): settings = {} messages.process_messages(settings, "foobar", queue, raise_error=False) fake_realtime.get_connection.assert_called_once_with(settings)
def test_runs_consumer(self, fake_consumer, queue): messages.process_messages({}, 'foobar', queue, raise_error=False) consumer = fake_consumer.return_value consumer.run.assert_called_once_with()
def test_initializes_new_connection(self, fake_realtime, fake_consumer, queue): settings = {} messages.process_messages(settings, 'foobar', queue, raise_error=False) fake_realtime.get_connection.assert_called_once_with(settings)
def test_creates_statsd_client(self, fake_stats, fake_consumer, queue): settings = {} messages.process_messages(settings, 'foobar', queue, raise_error=False) fake_stats.get_client.assert_called_once_with(settings)
def _handler(self, Consumer, work_queue): messages.process_messages({}, "routing_key", work_queue, raise_error=False) return Consumer.call_args[1]["handler"]
def test_it_raises_if_the_consumer_exits(self, work_queue): with pytest.raises(RuntimeError): messages.process_messages({}, "routing_key", work_queue)