def test_consume_basic(self):
"""
Should run the wrapped function when a message arrives with its routing key.
"""
with mock.patch.object(ec, 'REGISTRY', new=dict()) as reg:
f1 = message_handler('my.routing.key1')(
mock.MagicMock(__name__='mock_handler1'))
f2 = message_handler('my.routing.key2')(
mock.MagicMock(__name__='mock_handler2'))
assert len(reg) == 2
self.configure_handlers()
assert len(self.handlers) == len(reg)
h1 = self.get_handlers_for_key('my.routing.key1')[0]
h2 = self.get_handlers_for_key('my.routing.key2')[0]
p1 = self.get_producer(h1)
p2 = self.get_producer(h2)
body1 = self.body()
body2 = self.body()
p1.publish(body1)
p2.publish(body2)
for _ in range(2):
self.connection.drain_events(timeout=0.3)
f1.assert_called_once_with(body1)
f2.assert_called_once_with(body2)
# no retries:
e = None
try:
self.connection.drain_events(timeout=0.3)
except socket.timeout as exc:
e = exc
self.assertIsNotNone(
e, msg="e=None here means task was unexpectedly retried")
# no further calls
f1.call_count = 1
f2.call_count = 1
def test_consume_multiple_routes(self):
"""
Should run the wrapped function when a message arrives with its routing key.
Test that we can connect multiple routing keys on the same queue and the
appropriate handler will be called in each case.
"""
with mock.patch.object(ec, 'REGISTRY', new=dict()) as reg:
decorator = message_handler(
['my.routing.key1', 'my.routing.key2'],
exchange='custom',
)
f1 = decorator(mock.MagicMock())
assert len(reg) == 2
self.configure_handlers()
assert len(self.handlers) == len(reg)
h1 = self.get_handlers_for_key('my.routing.key1')[0]
h2 = self.get_handlers_for_key('my.routing.key2')[0]
p1 = self.get_producer(h1)
p2 = self.get_producer(h2)
body1 = self.body()
body2 = self.body()
p1.publish(body1)
p2.publish(body2)
for _ in range(2):
self.connection.drain_events(timeout=0.3)
f1.assert_has_calls(
[mock.call(body1), mock.call(body2)], any_order=True)
# no retries:
e = None
try:
self.connection.drain_events(timeout=0.3)
except socket.timeout as exc:
e = exc
self.assertIsNotNone(
e, msg="e=None here means task was unexpectedly retried")
# no further calls
f1.call_count = 2
def test_consume_custom_queue_name(self):
"""
Should run the wrapped function when a message arrives with its routing key.
Test that we can connect multiple routing keys on the same queue and the
appropriate handler will be called in each case.
"""
with mock.patch.object(ec, 'REGISTRY', new=dict()) as reg:
# we have to use a named exchange to be able to bind a custom queue name
f1 = message_handler('my.routing.key1',
queue='custom_queue',
exchange='custom')(
mock.MagicMock(__name__='mock_handler1'))
assert len(reg) == 1
self.configure_handlers()
assert len(self.handlers) == len(reg)
h1 = self.get_handlers_for_key('my.routing.key1')[0]
p1 = self.get_producer(h1)
body1 = self.body()
p1.publish(body1)
self.connection.drain_events(timeout=0.3)
f1.assert_called_once_with(body1)
# no retries:
e = None
try:
self.connection.drain_events(timeout=0.3)
except socket.timeout as exc:
e = exc
self.assertIsNotNone(
e, msg="e=None here means task was unexpectedly retried")
# no further calls
f1.call_count = 1