diff options
author | Chris Lambacher <clambacher@heroku.com> | 2020-10-07 11:52:18 -0400 |
---|---|---|
committer | Asif Saif Uddin <auvipy@gmail.com> | 2020-10-13 10:38:43 +0600 |
commit | b79146d8a1f822c486f6ab0df8bf36b5c53cf70d (patch) | |
tree | bd508ba3cfba106dab7fc4163c408920e8800de4 /t | |
parent | 3704a1c61e802957ed64f61e25106347d5bba856 (diff) | |
download | kombu-b79146d8a1f822c486f6ab0df8bf36b5c53cf70d.tar.gz |
Test default channel gets used
Diffstat (limited to 't')
-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 |