def stop(args):
"""`conduct stop` command"""
log = logging.getLogger(__name__)
path = "bundles/{}?scale=0".format(args.bundle)
url = conduct_url.url(path, args)
response = conduct_request.put(
args.dcos_mode,
conductr_host(args),
url,
auth=args.conductr_auth,
verify=args.server_verification_file,
timeout=DEFAULT_HTTP_TIMEOUT,
)
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json["bundleId"] if args.long_ids else bundle_utils.short_id(response_json["bundleId"])
log.info("Bundle stop request sent.")
if not args.no_wait:
bundle_scale.wait_for_scale(response_json["bundleId"], 0, args)
if not args.disable_instructions:
log.info("Unload bundle with: {} unload{} {}".format(args.command, args.cli_parameters, bundle_id))
log.info("Print ConductR info with: {} info{}".format(args.command, args.cli_parameters))
return True
def stop(args):
"""`conduct stop` command"""
log = logging.getLogger(__name__)
path = 'bundles/{}?scale=0'.format(args.bundle)
url = conduct_url.url(path, args)
response = conduct_request.put(args.dcos_mode,
conductr_host(args),
url,
auth=args.conductr_auth,
verify=args.server_verification_file,
timeout=DEFAULT_HTTP_TIMEOUT)
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json[
'bundleId'] if args.long_ids else bundle_utils.short_id(
response_json['bundleId'])
log.info('Bundle stop request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'], 0, args)
if not args.disable_instructions:
log.info('Unload bundle with: {} unload{} {}'.format(
args.command, args.cli_parameters, bundle_id))
log.info('Print ConductR info with: {} info{}'.format(
args.command, args.cli_parameters))
return True
def run(args):
"""`conduct run` command"""
log = logging.getLogger(__name__)
if args.affinity is not None and args.api_version == '1':
log.error('Affinity feature is only available for v1.1 onwards of ConductR')
return
elif args.affinity is not None:
path = 'bundles/{}?scale={}&affinity={}'.format(args.bundle, args.scale, args.affinity)
else:
path = 'bundles/{}?scale={}'.format(args.bundle, args.scale)
url = conduct_url.url(path, args)
response = requests.put(url)
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json['bundleId'] if args.long_ids else bundle_utils.short_id(response_json['bundleId'])
log.info('Bundle run request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'], args.scale, args)
log.info('Stop bundle with: conduct stop{} {}'.format(args.cli_parameters, bundle_id))
log.info('Print ConductR info with: conduct info{}'.format(args.cli_parameters))
return True
def test_return_immediately_if_scale_is_met(self):
conductr_host = '10.0.0.1'
conductr_host_mock = MagicMock(return_value=conductr_host)
get_events_mock = MagicMock(return_value=[])
get_scale_mock = MagicMock(side_effect=[3])
stdout = MagicMock()
bundle_id = 'a101449418187d92c789d1adc240b6d6'
dcos_mode = False
args = MagicMock(**{
'dcos_mode': dcos_mode,
'wait_timeout': 10,
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
})
with patch('conductr_cli.bundle_scale.get_scale', get_scale_mock), \
patch('conductr_cli.conduct_url.conductr_host', conductr_host_mock), \
patch('conductr_cli.sse_client.get_events', get_events_mock):
logging_setup.configure_logging(args, stdout)
bundle_scale.wait_for_scale(bundle_id, 3, args)
self.assertEqual(get_scale_mock.call_args_list, [
call(bundle_id, args)
])
conductr_host_mock.assert_not_called()
get_events_mock.assert_not_called()
self.assertEqual(strip_margin("""|Bundle a101449418187d92c789d1adc240b6d6 expected scale 3 is met
|"""), self.output(stdout))
def stop(args):
"""`conduct stop` command"""
log = logging.getLogger(__name__)
path = 'bundles/{}?scale=0'.format(args.bundle)
url = conduct_url.url(path, args)
# At the time when this comment is being written, we need to pass the Host header when making HTTP request due to
# a bug with requests python library not working properly when IPv6 address is supplied:
# https://github.com/kennethreitz/requests/issues/3002
# The workaround for this problem is to explicitly set the Host header when making HTTP request.
# This fix is benign and backward compatible as the library would do this when making HTTP request anyway.
response = requests.put(url, timeout=DEFAULT_HTTP_TIMEOUT, headers=conduct_url.request_headers(args))
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json['bundleId'] if args.long_ids else bundle_utils.short_id(response_json['bundleId'])
log.info('Bundle stop request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'], 0, args)
log.info('Unload bundle with: conduct unload{} {}'.format(args.cli_parameters, bundle_id))
log.info('Print ConductR info with: conduct info{}'.format(args.cli_parameters))
return True
def stop(args):
"""`conduct stop` command"""
log = logging.getLogger(__name__)
path = 'bundles/{}?scale=0'.format(args.bundle)
url = conduct_url.url(path, args)
# At the time when this comment is being written, we need to pass the Host header when making HTTP request due to
# a bug with requests python library not working properly when IPv6 address is supplied:
# https://github.com/kennethreitz/requests/issues/3002
# The workaround for this problem is to explicitly set the Host header when making HTTP request.
# This fix is benign and backward compatible as the library would do this when making HTTP request anyway.
response = requests.put(url,
timeout=DEFAULT_HTTP_TIMEOUT,
headers=conduct_url.request_headers(args))
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json[
'bundleId'] if args.long_ids else bundle_utils.short_id(
response_json['bundleId'])
log.info('Bundle stop request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'], 0, args)
log.info('Unload bundle with: conduct unload{} {}'.format(
args.cli_parameters, bundle_id))
log.info('Print ConductR info with: conduct info{}'.format(
args.cli_parameters))
return True
def test_return_immediately_if_scale_is_met(self):
conductr_host = '10.0.0.1'
conductr_host_mock = MagicMock(return_value=conductr_host)
get_events_mock = MagicMock(return_value=[])
get_scale_mock = MagicMock(side_effect=[3])
stdout = MagicMock()
bundle_id = 'a101449418187d92c789d1adc240b6d6'
dcos_mode = False
args = MagicMock(
**{
'dcos_mode': dcos_mode,
'wait_timeout': 10,
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
})
with patch('conductr_cli.bundle_scale.get_scale', get_scale_mock), \
patch('conductr_cli.conduct_url.conductr_host', conductr_host_mock), \
patch('conductr_cli.sse_client.get_events', get_events_mock):
logging_setup.configure_logging(args, stdout)
bundle_scale.wait_for_scale(bundle_id, 3, args)
self.assertEqual(get_scale_mock.call_args_list,
[call(bundle_id, args)])
conductr_host_mock.assert_not_called()
get_events_mock.assert_not_called()
self.assertEqual(
strip_margin(
"""|Bundle a101449418187d92c789d1adc240b6d6 expected scale 3 is met
|"""), self.output(stdout))
def run(args):
"""`conduct run` command"""
log = logging.getLogger(__name__)
if args.affinity is not None and args.api_version == '1':
log.error(
'Affinity feature is only available for v1.1 onwards of ConductR')
return
elif args.affinity is not None:
path = 'bundles/{}?scale={}&affinity={}'.format(
args.bundle, args.scale, args.affinity)
else:
path = 'bundles/{}?scale={}'.format(args.bundle, args.scale)
url = conduct_url.url(path, args)
response = conduct_request.put(args.dcos_mode,
conductr_host(args),
url,
auth=args.conductr_auth,
verify=args.server_verification_file)
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json[
'bundleId'] if args.long_ids else bundle_utils.short_id(
response_json['bundleId'])
log.info('Bundle run request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'],
args.scale,
wait_for_is_active=True,
args=args)
if not args.disable_instructions:
log.info('Stop bundle with: {} stop{} {}'.format(
args.command, args.cli_parameters, bundle_id))
log.info('Print ConductR info with: {} info{}'.format(
args.command, args.cli_parameters))
log.info('Print bundle info with: {} info{} {}'.format(
args.command, args.cli_parameters, bundle_id))
return True
def run(args):
"""`conduct run` command"""
log = logging.getLogger(__name__)
if args.affinity is not None and args.api_version == '1':
log.error(
'Affinity feature is only available for v1.1 onwards of ConductR')
return
elif args.affinity is not None:
path = 'bundles/{}?scale={}&affinity={}'.format(
args.bundle, args.scale, args.affinity)
else:
path = 'bundles/{}?scale={}'.format(args.bundle, args.scale)
url = conduct_url.url(path, args)
# At the time when this comment is being written, we need to pass the Host header when making HTTP request due to
# a bug with requests python library not working properly when IPv6 address is supplied:
# https://github.com/kennethreitz/requests/issues/3002
# The workaround for this problem is to explicitly set the Host header when making HTTP request.
# This fix is benign and backward compatible as the library would do this when making HTTP request anyway.
response = requests.put(url, headers=conduct_url.request_headers(args))
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json[
'bundleId'] if args.long_ids else bundle_utils.short_id(
response_json['bundleId'])
log.info('Bundle run request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'], args.scale,
args)
log.info('Stop bundle with: conduct stop{} {}'.format(
args.cli_parameters, bundle_id))
log.info('Print ConductR info with: conduct info{}'.format(
args.cli_parameters))
return True
def test_return_immediately_if_scale_is_met(self):
get_scale_mock = MagicMock(side_effect=[3])
stdout = MagicMock()
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = MagicMock(**{
'wait_timeout': 10
})
with patch('conductr_cli.bundle_scale.get_scale', get_scale_mock):
logging_setup.configure_logging(args, stdout)
bundle_scale.wait_for_scale(bundle_id, 3, args)
self.assertEqual(get_scale_mock.call_args_list, [
call(bundle_id, args)
])
self.assertEqual(strip_margin("""|Bundle a101449418187d92c789d1adc240b6d6 expected scale 3 is met
|"""), self.output(stdout))
def run(args):
"""`conduct run` command"""
log = logging.getLogger(__name__)
if args.affinity is not None and args.api_version == '1':
log.error('Affinity feature is only available for v1.1 onwards of ConductR')
return
elif args.affinity is not None:
path = 'bundles/{}?scale={}&affinity={}'.format(args.bundle, args.scale, args.affinity)
else:
path = 'bundles/{}?scale={}'.format(args.bundle, args.scale)
url = conduct_url.url(path, args)
# At the time when this comment is being written, we need to pass the Host header when making HTTP request due to
# a bug with requests python library not working properly when IPv6 address is supplied:
# https://github.com/kennethreitz/requests/issues/3002
# The workaround for this problem is to explicitly set the Host header when making HTTP request.
# This fix is benign and backward compatible as the library would do this when making HTTP request anyway.
response = requests.put(url, headers=conduct_url.request_headers(args))
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json['bundleId'] if args.long_ids else bundle_utils.short_id(response_json['bundleId'])
log.info('Bundle run request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'], args.scale, args)
log.info('Stop bundle with: conduct stop{} {}'.format(args.cli_parameters, bundle_id))
log.info('Print ConductR info with: conduct info{}'.format(args.cli_parameters))
return True
def stop(args):
"""`conduct stop` command"""
log = logging.getLogger(__name__)
path = 'bundles/{}?scale=0'.format(args.bundle)
url = conduct_url.url(path, args)
response = requests.put(url, timeout=DEFAULT_HTTP_TIMEOUT)
validation.raise_for_status_inc_3xx(response)
if log.is_verbose_enabled():
log.verbose(validation.pretty_json(response.text))
response_json = json.loads(response.text)
bundle_id = response_json['bundleId'] if args.long_ids else bundle_utils.short_id(response_json['bundleId'])
log.info('Bundle stop request sent.')
if not args.no_wait:
bundle_scale.wait_for_scale(response_json['bundleId'], 0, args)
log.info('Unload bundle with: conduct unload{} {}'.format(args.cli_parameters, bundle_id))
log.info('Print ConductR info with: conduct info{}'.format(args.cli_parameters))
return True
def test_wait_for_scale(self):
get_scale_mock = MagicMock(side_effect=[0, 1, 2, 3])
url_mock = MagicMock(return_value='/bundle-events/endpoint')
get_events_mock = MagicMock(return_value=[
self.create_test_event(None),
self.create_test_event('bundleExecutionAdded'),
self.create_test_event('bundleExecutionAdded'),
self.create_test_event('bundleExecutionAdded')
])
stdout = MagicMock()
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = MagicMock(**{
'wait_timeout': 10
})
with patch('conductr_cli.conduct_url.url', url_mock), \
patch('conductr_cli.bundle_scale.get_scale', get_scale_mock), \
patch('conductr_cli.sse_client.get_events', get_events_mock):
logging_setup.configure_logging(args, stdout)
bundle_scale.wait_for_scale(bundle_id, 3, args)
self.assertEqual(get_scale_mock.call_args_list, [
call(bundle_id, args),
call(bundle_id, args),
call(bundle_id, args),
call(bundle_id, args)
])
url_mock.assert_called_with('bundles/events', args)
self.assertEqual(strip_margin("""|Bundle a101449418187d92c789d1adc240b6d6 waiting to reach expected scale 3
|Bundle a101449418187d92c789d1adc240b6d6 has scale 1, expected 3
|Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3
|Bundle a101449418187d92c789d1adc240b6d6 expected scale 3 is met
|"""), self.output(stdout))
def test_periodic_check_between_events(self):
get_scale_mock = MagicMock(side_effect=[0, 1, 2, 2, 2, 3])
url_mock = MagicMock(return_value='/bundle-events/endpoint')
conductr_host = '10.0.0.1'
conductr_host_mock = MagicMock(return_value=conductr_host)
get_events_mock = MagicMock(return_value=[
self.create_test_event(None),
self.create_test_event('bundleExecutionAdded'),
self.create_test_event('bundleExecutionAdded'),
self.create_test_event('otherEvent'),
self.create_test_event(None),
self.create_test_event(None),
self.create_test_event(None),
self.create_test_event('bundleExecutionAdded')
])
stdout = MagicMock()
bundle_id = 'a101449418187d92c789d1adc240b6d6'
dcos_mode = False
args = MagicMock(**{
'dcos_mode': dcos_mode,
'wait_timeout': 10,
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
})
with patch('conductr_cli.conduct_url.url', url_mock), \
patch('conductr_cli.conduct_url.conductr_host', conductr_host_mock), \
patch('conductr_cli.bundle_scale.get_scale', get_scale_mock), \
patch('conductr_cli.sse_client.get_events', get_events_mock):
logging_setup.configure_logging(args, stdout)
bundle_scale.wait_for_scale(bundle_id, 3, args)
self.assertEqual(get_scale_mock.call_args_list, [
call(bundle_id, args),
call(bundle_id, args),
call(bundle_id, args),
call(bundle_id, args),
call(bundle_id, args),
call(bundle_id, args)
])
url_mock.assert_called_with('bundles/events', args)
conductr_host_mock.assert_called_with(args)
get_events_mock.assert_called_with(dcos_mode, conductr_host, '/bundle-events/endpoint', auth=self.conductr_auth,
verify=self.server_verification_file)
self.assertEqual(stdout.method_calls, [
call.write('Bundle a101449418187d92c789d1adc240b6d6 waiting to reach expected scale 3'),
call.write('\n'),
call.flush(),
call.write('Bundle a101449418187d92c789d1adc240b6d6 has scale 1, expected 3\r'),
call.write(''),
call.flush(),
call.write('Bundle a101449418187d92c789d1adc240b6d6 has scale 1, expected 3\n'),
call.write(''),
call.flush(),
call.write('Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3\r'),
call.write(''),
call.flush(),
call.write('Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3.\r'),
call.write(''),
call.flush(),
call.write('Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3..\r'),
call.write(''),
call.flush(),
call.write('Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3..\n'),
call.write(''),
call.flush(),
call.write('Bundle a101449418187d92c789d1adc240b6d6 expected scale 3 is met'),
call.write('\n'),
call.flush()
])
def test_periodic_check_between_events(self):
get_scale_mock = MagicMock(side_effect=[0, 1, 2, 2, 2, 3])
url_mock = MagicMock(return_value='/bundle-events/endpoint')
conductr_host = '10.0.0.1'
conductr_host_mock = MagicMock(return_value=conductr_host)
get_events_mock = MagicMock(return_value=[
self.create_test_event(None),
self.create_test_event('bundleExecutionAdded'),
self.create_test_event('bundleExecutionAdded'),
self.create_test_event('otherEvent'),
self.create_test_event(None),
self.create_test_event(None),
self.create_test_event(None),
self.create_test_event('bundleExecutionAdded')
])
stdout = MagicMock()
is_tty_mock = MagicMock(return_value=True)
bundle_id = 'a101449418187d92c789d1adc240b6d6'
dcos_mode = False
args = MagicMock(
**{
'dcos_mode': dcos_mode,
'wait_timeout': 10,
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
})
with patch('conductr_cli.conduct_url.url', url_mock), \
patch('conductr_cli.conduct_url.conductr_host', conductr_host_mock), \
patch('conductr_cli.bundle_scale.get_scale', get_scale_mock), \
patch('conductr_cli.sse_client.get_events', get_events_mock), \
patch('sys.stdout.isatty', is_tty_mock):
logging_setup.configure_logging(args, stdout)
bundle_scale.wait_for_scale(bundle_id,
3,
wait_for_is_active=True,
args=args)
self.assertEqual(get_scale_mock.call_args_list, [
call(bundle_id, True, args),
call(bundle_id, True, args),
call(bundle_id, True, args),
call(bundle_id, True, args),
call(bundle_id, True, args),
call(bundle_id, True, args)
])
url_mock.assert_called_with('bundles/events', args)
conductr_host_mock.assert_called_with(args)
get_events_mock.assert_called_with(
dcos_mode,
conductr_host,
'/bundle-events/endpoint',
auth=self.conductr_auth,
verify=self.server_verification_file)
self.assertEqual(stdout.method_calls, [
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 waiting to reach expected scale 3'
),
call.write('\n'),
call.flush(),
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 has scale 1, expected 3\r'
),
call.write(''),
call.flush(),
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 has scale 1, expected 3\n'
),
call.write(''),
call.flush(),
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3\r'
),
call.write(''),
call.flush(),
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3.\r'
),
call.write(''),
call.flush(),
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3..\r'
),
call.write(''),
call.flush(),
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 has scale 2, expected 3..\n'
),
call.write(''),
call.flush(),
call.write(
'Bundle a101449418187d92c789d1adc240b6d6 expected scale 3 is met'
),
call.write('\n'),
call.flush()
])
