示例#1
0
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
示例#2
0
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
示例#3
0
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
示例#4
0
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