class DescribeAddOnCancelCallback(_BaseTestCase):
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.channel = MagicMock()
def execute(self):
self.agent.add_on_cancel_callback()
def should_add_on_connection_close_callback(self):
self.agent.channel.add_on_cancel_callback.assert_called_once_with(
self.agent.on_consumer_cancel)
class DescribeAddOnCancelCallback(_BaseTestCase):
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.channel = MagicMock()
def execute(self):
self.agent.add_on_cancel_callback()
def should_add_on_connection_close_callback(self):
self.agent.channel.add_on_cancel_callback.assert_called_once_with(
self.agent.on_consumer_cancel)
