def establish_connection(self):
"""
Create a stomp connection
"""
connection_url = 'wss://{0}:{1}/agent/stomp/v1'.format(
self.config.server_hostname, self.config.secured_url_port)
connection_helper = security.VerifiedHTTPSConnection(
self.config.server_hostname, connection_url, self.config)
self.connection = connection_helper.connect()
def test_Verified_HTTPSConnection_non_secure_connect(
self, wrap_socket_mock, create_connection_mock,
init_security_mock):
connection = security.VerifiedHTTPSConnection(
"example.com", self.config.get('server', 'secured_url_port'),
self.config)
connection._tunnel_host = False
connection.sock = None
connection.connect()
self.assertFalse(init_security_mock.called)
def test_VerifiedHTTPSConnection_connect(self, wrap_socket_mock,
create_connection_mock,
init_security_mock):
init_security_mock.return_value = None
self.config.set('security', 'keysdir', '/dummy-keysdir')
connection = security.VerifiedHTTPSConnection("example.com",
self.config.get('server', 'secured_url_port'), self.config)
connection._tunnel_host = False
connection.sock = None
connection.connect()
self.assertTrue(wrap_socket_mock.called)
def test_Verified_HTTPSConnection_two_way_ssl_connect(self, wrap_socket_mock,
create_connection_mock,
init_security_mock):
wrap_socket_mock.side_effect=ssl.SSLError()
connection = security.VerifiedHTTPSConnection("example.com",
self.config.get('server', 'secured_url_port'), self.config)
connection._tunnel_host = False
connection.sock = None
try:
connection.connect()
except ssl.SSLError:
pass
self.assertTrue(init_security_mock.called)
