def test_get_deployments_strings_default_case_with_smartstack(): with contextlib.nested( mock.patch( 'paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True), mock.patch('service_configuration_lib.read_service_configuration', autospec=True), ) as ( mock_get_actual_deployments, mock_read_service_configuration, ): mock_get_actual_deployments.return_value = [ 'clusterA.main', 'clusterB.main' ] mock_read_service_configuration.return_value = { 'smartstack': { 'main': { 'proxy_port': 9001 } } } actual = info.get_deployments_strings('fake_service') assert ' - clusterA (%s)' % PaastaColors.cyan( 'http://fake_service.paasta-clusterA.yelp/') in actual assert ' - clusterB (%s)' % PaastaColors.cyan( 'http://fake_service.paasta-clusterB.yelp/') in actual
def test_get_deployments_strings_no_deployments(): with mock.patch( 'paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True ) as mock_get_actual_deployments: mock_get_actual_deployments.side_effect = NoDeploymentsAvailable actual = info.get_deployments_strings('unused') assert 'N/A: Not deployed' in actual[0]
def test_get_deployments_strings_non_listening_service(): with contextlib.nested( mock.patch('paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True), mock.patch('paasta_tools.paasta_cli.cmds.info.load_service_namespace_config', autospec=True), ) as (mock_get_actual_deployments, mock_load_service_namespace_config): mock_get_actual_deployments.return_value = ['clusterA.main', 'clusterB.main'] mock_load_service_namespace_config.return_value = ServiceNamespaceConfig() actual = info.get_deployments_strings('unused') assert ' - clusterA (N/A)' in actual assert ' - clusterB (N/A)' in actual
def test_get_deployments_strings_protocol_tcp_case(): with contextlib.nested( mock.patch('paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True), mock.patch('paasta_tools.paasta_cli.cmds.info.load_service_namespace_config', autospec=True), ) as (mock_get_actual_deployments, mock_load_service_namespace_config): mock_get_actual_deployments.return_value = ['clusterA.main', 'clusterB.main'] mock_load_service_namespace_config.return_value = ServiceNamespaceConfig({'mode': 'tcp', 'proxy_port': 8080}) actual = info.get_deployments_strings('unused') assert ' - clusterA (%s)' % PaastaColors.cyan('tcp://paasta-clusterA.yelp:8080/') in actual assert ' - clusterB (%s)' % PaastaColors.cyan('tcp://paasta-clusterB.yelp:8080/') in actual
def test_get_deployments_strings_non_listening_service(): with contextlib.nested( mock.patch( 'paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True), mock.patch( 'paasta_tools.paasta_cli.cmds.info.load_service_namespace_config', autospec=True), ) as (mock_get_actual_deployments, mock_load_service_namespace_config): mock_get_actual_deployments.return_value = [ 'clusterA.main', 'clusterB.main' ] mock_load_service_namespace_config.return_value = ServiceNamespaceConfig( ) actual = info.get_deployments_strings('unused') assert ' - clusterA (N/A)' in actual assert ' - clusterB (N/A)' in actual
def test_get_deployments_strings_default_case_with_smartstack(): with contextlib.nested( mock.patch('paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True), mock.patch('service_configuration_lib.read_service_configuration', autospec=True), ) as ( mock_get_actual_deployments, mock_read_service_configuration, ): mock_get_actual_deployments.return_value = ['clusterA.main', 'clusterB.main'] mock_read_service_configuration.return_value = { 'smartstack': { 'main': { 'proxy_port': 9001 } } } actual = info.get_deployments_strings('fake_service') assert ' - clusterA (%s)' % PaastaColors.cyan('http://fake_service.paasta-clusterA.yelp/') in actual assert ' - clusterB (%s)' % PaastaColors.cyan('http://fake_service.paasta-clusterB.yelp/') in actual
def test_get_deployments_strings_protocol_tcp_case(): with contextlib.nested( mock.patch( 'paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True), mock.patch( 'paasta_tools.paasta_cli.cmds.info.load_service_namespace_config', autospec=True), ) as (mock_get_actual_deployments, mock_load_service_namespace_config): mock_get_actual_deployments.return_value = [ 'clusterA.main', 'clusterB.main' ] mock_load_service_namespace_config.return_value = ServiceNamespaceConfig( { 'mode': 'tcp', 'proxy_port': 8080 }) actual = info.get_deployments_strings('unused') assert ' - clusterA (%s)' % PaastaColors.cyan( 'tcp://paasta-clusterA.yelp:8080/') in actual assert ' - clusterB (%s)' % PaastaColors.cyan( 'tcp://paasta-clusterB.yelp:8080/') in actual
def test_get_deployments_strings_no_deployments(): with mock.patch('paasta_tools.paasta_cli.cmds.info.get_actual_deployments', autospec=True) as mock_get_actual_deployments: mock_get_actual_deployments.side_effect = NoDeploymentsAvailable actual = info.get_deployments_strings('unused') assert 'N/A: Not deployed' in actual[0]