def test_report_cluster_status(mock_stdout, mock_load_system_paasta_config):
    cluster = 'fake_cluster'

    fake_system_paasta_config = SystemPaastaConfig(
        {
            'dashboard_links': {
                'fake_cluster': {
                    'URL': 'http://paasta-fake_cluster.yelp:5050',
                },
            },
        }, 'fake_directory')

    mock_load_system_paasta_config.return_value = fake_system_paasta_config

    thing_to_patch = 'paasta_tools.cli.cmds.metastatus.execute_paasta_metastatus_on_remote_master'
    with mock.patch(
            thing_to_patch) as mock_execute_paasta_metastatus_on_remote_master:
        mock_execute_paasta_metastatus_on_remote_master.return_value = 'mock_status'
        metastatus.print_cluster_status(cluster, fake_system_paasta_config)
        mock_execute_paasta_metastatus_on_remote_master.assert_called_once_with(
            cluster,
            fake_system_paasta_config,
            verbose=False,
        )
        actual = mock_stdout.getvalue()
        assert 'Cluster: %s' % cluster in actual
        assert 'mock_status' in actual
示例#2
0
def test_report_cluster_status(mock_stdout, mock_load_system_paasta_config):
    cluster = 'fake_cluster'

    fake_system_paasta_config = SystemPaastaConfig({
        'dashboard_links': {
            'fake_cluster': {
                'URL': 'http://paasta-fake_cluster.yelp:5050',
            },
        },
    }, 'fake_directory')

    mock_load_system_paasta_config.return_value = fake_system_paasta_config

    thing_to_patch = 'paasta_tools.cli.cmds.metastatus.execute_paasta_metastatus_on_remote_master'
    with mock.patch(thing_to_patch) as mock_execute_paasta_metastatus_on_remote_master:
        mock_execute_paasta_metastatus_on_remote_master.return_value = 'mock_status'
        metastatus.print_cluster_status(
            cluster,
            fake_system_paasta_config,
            False,
            [],
            verbose=0
        )
        mock_execute_paasta_metastatus_on_remote_master.assert_called_once_with(
            cluster=cluster,
            system_paasta_config=fake_system_paasta_config,
            humanize=False,
            groupings=[],
            verbose=0,
        )
        actual = mock_stdout.getvalue()
        assert 'Cluster: %s' % cluster in actual
        assert 'mock_status' in actual
示例#3
0
def test_report_cluster_status(mock_stdout):
    cluster = 'fake_cluster'
    thing_to_patch = 'paasta_tools.cli.cmds.metastatus.execute_paasta_metastatus_on_remote_master'
    with mock.patch(thing_to_patch) as mock_execute_paasta_metastatus_on_remote_master:
        mock_execute_paasta_metastatus_on_remote_master.return_value = 'mock_status'
        metastatus.print_cluster_status(cluster)
        mock_execute_paasta_metastatus_on_remote_master.assert_called_once_with(
            cluster, False
        )
        actual = mock_stdout.getvalue()
        assert 'Cluster: %s' % cluster in actual
        assert 'mock_status' in actual
示例#4
0
def test_report_cluster_status(mock_load_system_paasta_config, capfd):
    cluster = 'fake_cluster'

    fake_system_paasta_config = SystemPaastaConfig(
        {
            'dashboard_links': {
                'fake_cluster': {
                    'URL': 'http://paasta-fake_cluster.yelp:5050',
                },
            },
        }, 'fake_directory')

    mock_load_system_paasta_config.return_value = fake_system_paasta_config

    thing_to_patch = 'paasta_tools.cli.cmds.metastatus.execute_paasta_metastatus_on_remote_master'
    with mock.patch(
            thing_to_patch,
            autospec=True) as mock_execute_paasta_metastatus_on_remote_master:
        mock_execute_paasta_metastatus_on_remote_master.return_value = mock.sentinel.return_value, 'mock_status'
        return_code = metastatus.print_cluster_status(
            cluster, fake_system_paasta_config, False, [], verbose=0)
        mock_execute_paasta_metastatus_on_remote_master.assert_called_once_with(
            cluster=cluster,
            system_paasta_config=fake_system_paasta_config,
            humanize=False,
            groupings=[],
            verbose=0,
            autoscaling_info=False)
        actual, _ = capfd.readouterr()
        assert 'Cluster: %s' % cluster in actual
        assert 'mock_status' in actual
        assert return_code == mock.sentinel.return_value
示例#5
0
def test_report_cluster_status(mock_load_system_paasta_config, capfd):
    cluster = "fake_cluster"

    fake_system_paasta_config = SystemPaastaConfig(
        {
            "dashboard_links": {
                "fake_cluster": {"URL": "http://paasta-fake_cluster.yelp:5050"}
            }
        },
        "fake_directory",
    )

    mock_load_system_paasta_config.return_value = fake_system_paasta_config

    thing_to_patch = (
        "paasta_tools.cli.cmds.metastatus.execute_paasta_metastatus_on_remote_master"
    )
    with mock.patch(
        thing_to_patch, autospec=True
    ) as mock_execute_paasta_metastatus_on_remote_master:
        mock_execute_paasta_metastatus_on_remote_master.return_value = (
            mock.sentinel.return_value,
            "mock_status",
        )
        return_code = metastatus.print_cluster_status(
            cluster, fake_system_paasta_config, [], verbose=0
        )
        mock_execute_paasta_metastatus_on_remote_master.assert_called_once_with(
            cluster=cluster,
            system_paasta_config=fake_system_paasta_config,
            groupings=[],
            verbose=0,
            autoscaling_info=False,
            use_mesos_cache=False,
        )
        actual, _ = capfd.readouterr()
        assert "Cluster: %s" % cluster in actual
        assert "mock_status" in actual
        assert return_code == mock.sentinel.return_value