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_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_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_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_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 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)
