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