diff options
| -rw-r--r-- | t/unit/test_pidbox.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/t/unit/test_pidbox.py b/t/unit/test_pidbox.py index 5b14ffff..dc2467cb 100644 --- a/t/unit/test_pidbox.py +++ b/t/unit/test_pidbox.py @@ -95,6 +95,47 @@ class test_Mailbox: de.side_effect = socket.timeout mailbox._collect(ticket, limit=1, channel=channel) + def test_reply__collect_uses_default_channel(self): + class ConsumerCalled(Exception): + pass + + def fake_Consumer(channel, *args, **kwargs): + raise ConsumerCalled(channel) + + ticket = uuid() + with patch('kombu.pidbox.Consumer') as Consumer: + mailbox = pidbox.Mailbox('test_reply__collect')(self.connection) + assert mailbox.connection.default_channel is not None + Consumer.side_effect = fake_Consumer + try: + mailbox._collect(ticket, limit=1) + except ConsumerCalled as c: + assert c.args[0] is not None + except Exception: + raise + else: + assert False, "Consumer not called" + + def test__publish_uses_default_channel(self): + class QueueCalled(Exception): + pass + + def queue__call__side(channel, *args, **kwargs): + raise QueueCalled(channel) + + ticket = uuid() + with patch.object(pidbox.Queue, '__call__') as queue__call__: + mailbox = pidbox.Mailbox('test_reply__collect')(self.connection) + queue__call__.side_effect = queue__call__side + try: + mailbox._publish(ticket, {}, reply_ticket=ticket) + except QueueCalled as c: + assert c.args[0] is not None + except Exception: + raise + else: + assert False, "Queue not called" + def test_constructor(self): assert self.mailbox.connection is None assert self.mailbox.exchange.name |