def test_conductr_host_env_set(self):
os_getenv_mock = MagicMock(return_value='conductr_host_env')
with patch('os.getenv', os_getenv_mock):
result = host.resolve_host_from_env()
self.assertEqual(result, 'conductr_host_env')
os_getenv_mock.assert_called_with('CONDUCTR_HOST', 'conductr_host_env')
def test_no_env_set(self):
os_getenv_mock = MagicMock(side_effect=[None, None])
with patch('os.getenv', os_getenv_mock):
result = host.resolve_host_from_env()
self.assertEqual(result, None)
os_getenv_mock.assert_has_calls(
[call('CONDUCTR_IP', None),
call('CONDUCTR_HOST', None)])
def run(_args=[], configure_logging=True):
# If we're being invoked via DC/OS then route our http
# calls via its extension to the requests library. In
# addition remove the 'conduct-dcos' and 'conduct' arg so that the conduct
# sub-commands are positioned correctly, along with their
# arguments.
if sys.argv and Path(
sys.argv[0]).name == constants.DCOS_COMMAND_PREFIX + 'conduct':
dcos_mode = True
_args = sys.argv[2:]
else:
dcos_mode = False
if not _args:
# Remove the 'conduct' arg so that we start with the sub command directly
_args = sys.argv[1:]
# Parse arguments
parser = build_parser(dcos_mode)
argcomplete.autocomplete(parser)
args = parser.parse_args(_args)
args.dcos_mode = dcos_mode
if not vars(args).get('func'):
if vars(args).get('dcos_info'):
print(
'Lightbend ConductR sub commands. Type \'dcos conduct\' to see more.'
)
exit(0)
else:
parser.print_help()
else:
# Offline functions are the functions which do not require network to run, e.g. `conduct version` or
# `conduct setup-dcos`.
offline_functions = ['version', 'setup']
# Only setup network related args (i.e. host, bundle resolvers, basic auth, etc) for functions which requires
# connectivity to ConductR.
current_function = vars(args).get('func').__name__
if current_function not in offline_functions:
# Add custom plugin dir to import path
custom_plugins_dir = vars(args).get('custom_plugins_dir')
if custom_plugins_dir:
sys.path.append(custom_plugins_dir)
# DC/OS provides the location of ConductR...
if dcos_mode:
args.command = 'dcos conduct'
dcos_url = urlparse(config.get_config_val('core.dcos_url'))
args.scheme = dcos_url.scheme
args.ip = dcos_url.hostname
default_http_port = 80 if dcos_url.scheme == 'http' else 443
args.port = dcos_url.port if dcos_url.port else default_http_port
dcos_url_path = dcos_url.path if dcos_url.path else '/'
args.base_path = dcos_url_path + 'service/{}/'.format(
DEFAULT_DCOS_SERVICE)
else:
args.command = 'conduct'
# Set ConductR host is --host or --ip argument not set
# Also set the local_connection argument accordingly
host_from_args = conduct_url.conductr_host(args)
if not host_from_args:
host_from_env = host.resolve_host_from_env()
if host_from_env:
args.host = host_from_env
args.local_connection = False
else:
args.host = host.resolve_default_host()
else:
args.local_connection = False
args.cli_parameters = get_cli_parameters(args)
args.custom_settings = custom_settings.load_from_file(args)
args.conductr_auth = custom_settings.load_conductr_credentials(
args)
# Ensure HTTPS is used if authentication is configured
if args.conductr_auth and args.scheme != 'https':
args.scheme = 'https'
args.server_verification_file = custom_settings.load_server_ssl_verification_file(
args)
# Ensure verification file exists if specified
if args.server_verification_file \
and not os.path.exists(args.server_verification_file):
# Configure logging so error message can be logged properly before exiting with failure
logging_setup.configure_logging(args)
log = logging.getLogger(__name__)
log.error(
'Ensure server SSL verification file exists: {}'.format(
args.server_verification_file))
exit(1)
if not args.dcos_mode and args.scheme == 'https':
disable_urllib3_warnings()
if configure_logging:
logging_setup.configure_logging(args)
is_completed_without_error = args.func(args)
if not is_completed_without_error:
exit(1)