Exemple #1
0
    def test_open(self, connection, find, add_transports, ssl_domain):
        url = TEST_URL
        connector = Connector(url)
        connector.ssl.ca_certificate = 'test-ca'
        connector.ssl.client_key = 'test-key'
        connector.ssl.client_certificate = 'test-crt'
        connector.heartbeat = '4'
        find.return_value = connector

        ssl_properties = {'A': 1, 'B': 2}
        ssl_domain.return_value = ssl_properties

        # test
        c = Connection(url)
        c._ssh = Mock()
        c.open()

        # validation
        add_transports.assert_called_once_with()
        connection.assert_called_once_with(
            host=connector.host,
            port=connector.port,
            tcp_nodelay=True,
            transport=connector.scheme,
            username=connector.userid,
            password=connector.password,
            heartbeat=connector.heartbeat,
            **ssl_properties)

        ssl_domain.assert_called_once_with(connector)
        c._impl.open.assert_called_once_with()
        self.assertEqual(c._impl, connection.return_value)
Exemple #2
0
    def test_open(self, connection, find, add_transports, ssl_domain):
        url = TEST_URL
        connector = Connector(url)
        connector.ssl.ca_certificate = 'test-ca'
        connector.ssl.client_key = 'test-key'
        connector.ssl.client_certificate = 'test-crt'
        connector.heartbeat = '4'
        find.return_value = connector

        ssl_properties = {'A': 1, 'B': 2}
        ssl_domain.return_value = ssl_properties

        # test
        c = Connection(url)
        c._ssh = Mock()
        c.open()

        # validation
        add_transports.assert_called_once_with()
        connection.assert_called_once_with(host=connector.host,
                                           port=connector.port,
                                           tcp_nodelay=True,
                                           transport=connector.scheme,
                                           username=connector.userid,
                                           password=connector.password,
                                           heartbeat=connector.heartbeat,
                                           **ssl_properties)

        ssl_domain.assert_called_once_with(connector)
        c._impl.open.assert_called_once_with()
        self.assertEqual(c._impl, connection.return_value)