Python ConsumerAgent.on_connection_open Beispiele

Beispiel #1

Datei: test_agent.py Projekt: nooperpudd/PikaChewie

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()

Beispiel #2

Datei: test_agent.py Projekt: bdeeney/PikaChewie

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()