def test_paasta_metastatus_returns_one_on_error(): args = mock.Mock(soa_dir=mock.sentinel.soa_dir, clusters="cluster1,cluster2,cluster3") with mock.patch( "paasta_tools.cli.cmds.metastatus.list_clusters", autospec=True) as mock_list_clusters, mock.patch( "paasta_tools.cli.cmds.metastatus.print_cluster_status", autospec=True ) as mock_print_cluster_status, mock.patch( "paasta_tools.cli.cmds.metastatus.load_system_paasta_config", autospec=True): mock_list_clusters.return_value = ["cluster1", "cluster2", "cluster3"] mock_print_cluster_status.side_effect = [0, 0, 255] return_code = metastatus.paasta_metastatus(args) assert return_code == 1 assert mock_print_cluster_status.call_count == 3
def test_paasta_metastatus_returns_zero_all_clusters_ok(): args = mock.Mock( soa_dir=mock.sentinel.soa_dir, clusters='cluster1,cluster2,cluster3', ) with mock.patch( 'paasta_tools.cli.cmds.metastatus.list_clusters', autospec=True, ) as mock_list_clusters, mock.patch( 'paasta_tools.cli.cmds.metastatus.print_cluster_status', autospec=True, ) as mock_print_cluster_status, mock.patch( 'paasta_tools.cli.cmds.metastatus.load_system_paasta_config', autospec=True, ): mock_list_clusters.return_value = ['cluster1', 'cluster2', 'cluster3'] mock_print_cluster_status.side_effect = [0, 0, 0] return_code = metastatus.paasta_metastatus(args) assert return_code == 0 assert mock_print_cluster_status.call_count == 3
def test_paasta_metastatus_returns_one_on_error(): args = mock.Mock( soa_dir=mock.sentinel.soa_dir, clusters='cluster1,cluster2,cluster3', ) with contextlib.nested( mock.patch('paasta_tools.cli.cmds.metastatus.list_clusters', autospec=True), mock.patch('paasta_tools.cli.cmds.metastatus.print_cluster_status', autospec=True), mock.patch('paasta_tools.cli.cmds.metastatus.load_system_paasta_config', autospec=True), ) as ( mock_list_clusters, mock_print_cluster_status, _ ): mock_list_clusters.return_value = ['cluster1', 'cluster2', 'cluster3'] mock_print_cluster_status.side_effect = [0, 0, 255] return_code = metastatus.paasta_metastatus(args) assert return_code == 1 assert mock_print_cluster_status.call_count == 3
def test_paasta_metastatus_returns_one_on_error(): args = mock.Mock( soa_dir=mock.sentinel.soa_dir, clusters='cluster1,cluster2,cluster3', ) with contextlib.nested( mock.patch('paasta_tools.cli.cmds.metastatus.list_clusters', autospec=True), mock.patch('paasta_tools.cli.cmds.metastatus.print_cluster_status', autospec=True), mock.patch( 'paasta_tools.cli.cmds.metastatus.load_system_paasta_config', autospec=True), ) as (mock_list_clusters, mock_print_cluster_status, _): mock_list_clusters.return_value = ['cluster1', 'cluster2', 'cluster3'] mock_print_cluster_status.side_effect = [0, 0, 255] return_code = metastatus.paasta_metastatus(args) assert return_code == 1 assert mock_print_cluster_status.call_count == 3