Exemplos de ConsumerAgent.connect em Python

Exemplo n.º 1

Arquivo: test_agent.py Projeto: nooperpudd/PikaChewie

class WhenConnectingToBroker(_BaseTestCase):
    def configure(self):
        self.broker = MagicMock()
        self.broker.connect.return_value = sentinel.connection
        self.agent = ConsumerAgent(sentinel.consumer, self.broker,
                                   sentinel.bindings)

    def execute(self):
        self.agent.connect()

    def should_create_broker_connection(self):
        self.broker.connect.assert_called_once_with(
            self.agent.on_connection_open,
            on_failure_callback=self.agent.on_connection_failure)

    def should_set_connection(self):
        self.assertIs(self.agent.connection, sentinel.connection)

Exemplo n.º 2

Arquivo: test_agent.py Projeto: bdeeney/PikaChewie

class WhenConnectingToBroker(_BaseTestCase):

    def configure(self):
        self.broker = MagicMock()
        self.broker.connect.return_value = sentinel.connection
        self.agent = ConsumerAgent(sentinel.consumer, self.broker,
                                   sentinel.bindings)

    def execute(self):
        self.agent.connect()

    def should_create_broker_connection(self):
        self.broker.connect.assert_called_once_with(
            self.agent.on_connection_open,
            on_failure_callback=self.agent.on_connection_failure
        )

    def should_set_connection(self):
        self.assertIs(self.agent.connection, sentinel.connection)