class DescribeOnConnectionOpen(_BaseTestCase): def configure(self): self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker, sentinel.bindings) self.agent.add_on_connection_close_callback = MagicMock() self.agent.open_channel = MagicMock() def execute(self): self.agent.on_connection_open(sentinel.connection) def should_add_on_connection_close_callback(self): self.agent.add_on_connection_close_callback.assert_called_once_with() def should_open_channel(self): self.agent.open_channel.assert_called_once_with()