Python get_hosts_with_state Examples

Example #1

File: test_mesos_maintenance.py Project: oktopuz/paasta

def test_get_hosts_with_state_none(
    mock_get_maintenance_status,
):
    fake_status = {}
    mock_get_maintenance_status.return_value = mock.Mock()
    mock_get_maintenance_status.return_value.json.return_value = fake_status
    assert get_hosts_with_state(state='fake_state') == []

Example #2

File: test_mesos_maintenance.py Project: Yelp/paasta

def test_get_hosts_with_state_none(mock_get_maintenance_status,
                                   mock_get_mesos_config_path):
    mock_get_mesos_config_path.return_value = "/dev/null"
    fake_status = {"get_maintenance_status": {"status": {}}}
    mock_get_maintenance_status.return_value = mock.Mock()
    mock_get_maintenance_status.return_value.json.return_value = fake_status
    assert get_hosts_with_state(state="fake_state") == []

Example #3

def test_get_hosts_with_state_none(
    mock_get_maintenance_status,
):
    fake_status = {'get_maintenance_status': {'status': {}}}
    mock_get_maintenance_status.return_value = mock.Mock()
    mock_get_maintenance_status.return_value.json.return_value = fake_status
    assert get_hosts_with_state(state='fake_state') == []

Example #4

def test_get_hosts_with_state_down(mock_get_maintenance_status, ):
    fake_status = {
        "down_machines": [{
            "hostname": "fake-host1.fakesite.something",
            "ip": "0.0.0.0"
        }, {
            "hostname": "fake-host2.fakesite.something",
            "ip": "0.0.0.1"
        }]
    }
    mock_get_maintenance_status.return_value = mock.Mock()
    mock_get_maintenance_status.return_value.json.return_value = fake_status
    expected = sorted(
        ['fake-host1.fakesite.something', 'fake-host2.fakesite.something'])
    assert sorted(get_hosts_with_state(state='down_machines')) == expected

Example #5

File: test_mesos_maintenance.py Project: oktopuz/paasta

def test_get_hosts_with_state_down(
    mock_get_maintenance_status,
):
    fake_status = {
        "down_machines": [
            {
                "hostname": "fake-host1.fakesite.something",
                "ip": "0.0.0.0"
            },
            {
                "hostname": "fake-host2.fakesite.something",
                "ip": "0.0.0.1"
            }
        ]
    }
    mock_get_maintenance_status.return_value = mock.Mock()
    mock_get_maintenance_status.return_value.json.return_value = fake_status
    expected = sorted(['fake-host1.fakesite.something', 'fake-host2.fakesite.something'])
    assert sorted(get_hosts_with_state(state='down_machines')) == expected

Example #6

def test_get_hosts_with_state_draining(
    mock_get_maintenance_status,
):
    fake_status = {
        "type": "GET_MAINTENANCE_STATUS",
        "get_maintenance_status": {"status": {"draining_machines": [
            {
                "hostname": "fake-host1.fakesite.something",
                "ip": "0.0.0.0",
            },
            {
                "hostname": "fake-host2.fakesite.something",
                "ip": "0.0.0.1",
            },
        ]}},
    }
    mock_get_maintenance_status.return_value = mock.Mock()
    mock_get_maintenance_status.return_value.json.return_value = fake_status
    expected = sorted(['fake-host1.fakesite.something', 'fake-host2.fakesite.something'])
    assert sorted(get_hosts_with_state(state='draining_machines')) == expected

Example #7

File: test_mesos_maintenance.py Project: Yelp/paasta

def test_get_hosts_with_state_down(mock_get_maintenance_status,
                                   mock_get_mesos_config_path):
    fake_status = {
        "type": "GET_MAINTENANCE_STATUS",
        "get_maintenance_status": {
            "status": {
                "down_machines": [
                    {
                        "hostname": "fake-host1.fakesite.something",
                        "ip": "0.0.0.0"
                    },
                    {
                        "hostname": "fake-host2.fakesite.something",
                        "ip": "0.0.0.1"
                    },
                ]
            }
        },
    }
    mock_get_maintenance_status.return_value = mock.Mock()
    mock_get_maintenance_status.return_value.json.return_value = fake_status
    expected = sorted(
        ["fake-host1.fakesite.something", "fake-host2.fakesite.something"])
    assert sorted(get_hosts_with_state(state="down_machines")) == expected