def smartstack_check(service, service_path):
"""Check whether smartstack.yaml exists in service directory, and the proxy
ports are declared. Print appropriate message depending on outcome.
:param service: name of service currently being examined
:param service_path: path to loction of smartstack.yaml file"""
if is_file_in_dir("smartstack.yaml", service_path):
print PaastaCheckMessages.SMARTSTACK_YAML_FOUND
instances = get_all_namespaces_for_service(service)
if len(instances) > 0:
for namespace, config in get_all_namespaces_for_service(service, full_name=False):
if "proxy_port" in config:
print PaastaCheckMessages.smartstack_port_found(namespace, config.get("proxy_port"))
else:
print PaastaCheckMessages.SMARTSTACK_PORT_MISSING
else:
print PaastaCheckMessages.SMARTSTACK_PORT_MISSING
def smartstack_check(service, service_path):
"""Check whether smartstack.yaml exists in service directory, and the proxy
ports are declared. Print appropriate message depending on outcome.
:param service: name of service currently being examined
:param service_path: path to loction of smartstack.yaml file"""
if is_file_in_dir('smartstack.yaml', service_path):
print PaastaCheckMessages.SMARTSTACK_YAML_FOUND
instances = get_all_namespaces_for_service(service)
if len(instances) > 0:
for namespace, config in get_all_namespaces_for_service(service, full_name=False):
if 'proxy_port' in config:
print PaastaCheckMessages.smartstack_port_found(
namespace, config.get('proxy_port'))
else:
print PaastaCheckMessages.SMARTSTACK_PORT_MISSING
else:
print PaastaCheckMessages.SMARTSTACK_PORT_MISSING
def test_check_smartstack_check_pass(mock_stdout, mock_is_file_in_dir,
mock_read_service_info):
# smartstack.yaml exists and port is found
mock_is_file_in_dir.return_value = True
port = 80
instance = 'main'
smartstack_dict = {'smartstack': {instance: {'proxy_port': port}}}
mock_read_service_info.return_value = smartstack_dict
expected_output = "%s\n%s\n" \
% (PaastaCheckMessages.SMARTSTACK_YAML_FOUND,
PaastaCheckMessages.smartstack_port_found(
instance, port))
smartstack_check('fake_service', 'path')
output = mock_stdout.getvalue()
assert output == expected_output
def test_check_smartstack_check_pass(mock_stdout, mock_is_file_in_dir,
mock_read_service_info):
# smartstack.yaml exists and port is found
mock_is_file_in_dir.return_value = True
port = 80
instance = 'main'
smartstack_dict = {
'smartstack': {
instance: {
'proxy_port': port
}
}
}
mock_read_service_info.return_value = smartstack_dict
expected_output = "%s\n%s\n" \
% (PaastaCheckMessages.SMARTSTACK_YAML_FOUND,
PaastaCheckMessages.smartstack_port_found(
instance, port))
smartstack_check('fake_service', 'path')
output = mock_stdout.getvalue()
assert output == expected_output
