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)
def test_close_failed(self): url = 'test-url' c = Connection(url) impl = Mock() impl.close.side_effect = ValueError c._impl = impl c.close()
def __init__(self, url): """ :param url: The broker url. :type url: str """ BaseSender.__init__(self, url) self.connection = Connection(url) self.session = None
def test_close(self): url = 'test-url' c = Connection(url) impl = Mock() c._impl = impl c.close() impl.close.assert_called_once_with() self.assertEqual(c._impl, None)
def __init__(self, url, name, arguments): """ :param url: The broker url. :type url: str :param name: The method name. :type name: str :param arguments: The method arguments. :type arguments: dict """ super(Method, self).__init__(url) self.name = name self.arguments = arguments self.connection = Connection(url) self.session = None self.sender = None self.receiver = None self.reply_to = REPLY_TO % uuid4()
def test_init(self): url = TEST_URL c = Connection(url) self.assertTrue(isinstance(c, BaseConnection)) self.assertEqual(c.url, url)
def test_session(self): url = TEST_URL c = Connection(url) c._impl = Mock() session = c.session() self.assertEqual(session, c._impl.session.return_value)
def test_open_already(self): url = TEST_URL c = Connection(url) c._impl = Mock() c.open() self.assertFalse(c._impl.open.called)