def test_parse_address_for_hostname_and_url(self):
hostname, port = ThriftExecutor._parse_address_for_hostname_and_port(
data.TEST_SERVER_ADDRESS)
hostname2, port2 = ThriftExecutor._parse_address_for_hostname_and_port(
data.TEST_SERVER_ADDRESS2)
hostname3, port3 = ThriftExecutor._parse_address_for_hostname_and_port(
data.TEST_SERVER_ADDRESS3)
expected_hostname, expected_port = data.TEST_SERVER_HOSTNAME, data.TEST_SERVER_PORT
expected_hostname2, expected_port2 = data.TEST_SERVER_HOSTNAME2, data.TEST_SERVER_PORT2
expected_hostname3, expected_port3 = data.TEST_SERVER_HOSTNAME3, data.TEST_SERVER_PORT3
self.assertEqual((hostname, port), (expected_hostname, expected_port))
self.assertEqual((hostname2, port2),
(expected_hostname2, expected_port2))
self.assertEqual((hostname3, port3),
(expected_hostname3, expected_port3))
def test_init(self, mock_load_file, mock_call, mock_import_package,
mock_tsocket, mock_transport_open, mock_finagle_protocol):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
mock_call.return_value = 0
ThriftExecutor(data.TEST_THRIFT_PATH, data.TEST_SERVER_ADDRESS,
data.TEST_THRIFT_SERVICE_REFERENCE,
data.TEST_THRIFT_NAMESPACES)
command = 'thrift -r -I %s --gen py %s' % (data.TEST_THRIFT_DIR,
data.TEST_THRIFT_PATH)
mock_call.assert_called_with(command, shell=True)
mock_import_package.assert_called_with(data.TEST_THRIFT_MODULE_NAME,
data.TEST_THRIFT_PY_NAMESPACE)
mock_tsocket.assert_called_with(data.TEST_SERVER_HOSTNAME,
data.TEST_SERVER_PORT)
self.assertTrue(mock_transport_open.called)
mock_finagle_protocol.assert_called()
def test_init_ssl(self, mock_load_file, mock_call, mock_import_package,
mock_tsocket, mock_transport_open,
mock_finagle_protocol):
mock_load_file.return_value = data.TEST_THRIFT_CONTENT
mock_call.return_value = 0
ThriftExecutor(data.TEST_THRIFT_FILE, data.TEST_SERVER_ADDRESS,
data.TEST_THRIFT_SERVICE_REFERENCE,
data.TEST_THRIFT_NAMESPACES, True, None, 'none')
command = 'thrift -r -I . --gen py %s' % data.TEST_THRIFT_FILE
mock_call.assert_called_with(command, shell=True)
mock_import_package.assert_called_with(data.TEST_THRIFT_MODULE_NAME,
data.TEST_THRIFT_PY_NAMESPACE)
mock_tsocket.assert_called_with(data.TEST_SERVER_HOSTNAME,
data.TEST_SERVER_PORT,
ssl_context=mock.ANY)
self.assertTrue(mock_transport_open.called)
mock_finagle_protocol.assert_called()