Ejemplos de get_hosts_past_maintenance_start en Python

Ejemplo n.º 1

def test_get_hosts_past_maintenance_start_grace(
    mock_datetime_to_nanoseconds,
    mock_get_maintenance_schedule,
):
    mock_schedule = {
        "windows": [{
            "machine_ids": [{
                "hostname": "host3"
            }],
            "unavailability": {
                "start": {
                    "nanoseconds": 10
                }
            }
        }, {
            "machine_ids": [{
                "hostname": "host2"
            }],
            "unavailability": {
                "start": {
                    "nanoseconds": 5
                }
            }
        }]
    }
    mock_maintenance_dict = mock.Mock(return_value=mock_schedule)
    mock_get_maintenance_schedule.return_value = mock.Mock(
        json=mock_maintenance_dict)
    mock_datetime_to_nanoseconds.return_value = 7
    ret = get_hosts_past_maintenance_start(grace=1)
    assert ret == ['host2']

    ret = get_hosts_past_maintenance_start(grace=5)
    assert ret == []

Ejemplo n.º 2

Archivo: test_mesos_maintenance.py Proyecto: kkellyy/paasta

def test_get_hosts_past_maintenance_start(
    mock_datetime_to_nanoseconds,
    mock_get_maintenance_schedule,
):
    mock_schedule = {
        "type": "GET_MAINTENANCE_SCHEDULE",
        "get_maintenance_schedule": {
            "schedule": {
                "windows": [
                    {
                        "machine_ids": [
                            {
                                "hostname": "host3",
                            },
                        ],
                        "unavailability": {
                            "start": {
                                "nanoseconds": 10,
                            },
                        },
                    },
                    {
                        "machine_ids": [
                            {
                                "hostname": "host2",
                            },
                        ],
                        "unavailability": {
                            "start": {
                                "nanoseconds": 5,
                            },
                        },
                    },
                ]
            }
        },
    }
    mock_maintenance_dict = mock.Mock(return_value=mock_schedule)
    mock_get_maintenance_schedule.return_value = mock.Mock(
        json=mock_maintenance_dict)
    mock_datetime_to_nanoseconds.return_value = 7
    ret = get_hosts_past_maintenance_start()
    assert ret == ['host2']

    mock_schedule = {
        "type": "GET_MAINTENANCE_SCHEDULE",
        "get_maintenance_schedule": {
            "schedule": {}
        },
    }
    mock_maintenance_dict = mock.Mock(return_value=mock_schedule)
    mock_get_maintenance_schedule.return_value = mock.Mock(
        json=mock_maintenance_dict)
    ret = get_hosts_past_maintenance_start()
    assert ret == []

Ejemplo n.º 3

Archivo: test_mesos_maintenance.py Proyecto: oktopuz/paasta

def test_get_hosts_past_maintenance_start(
    mock_datetime_to_nanoseconds,
    mock_get_maintenance_schedule,
):
    mock_schedule = {
        "windows": [
            {
                "machine_ids": [
                    {
                        "hostname": "host3"
                    }
                ],
                "unavailability": {
                    "start": {
                        "nanoseconds": 10
                    }
                }
            },
            {
                "machine_ids": [
                    {
                        "hostname": "host2"
                    }
                ],
                "unavailability": {
                    "start": {
                        "nanoseconds": 5
                    }
                }
            }
        ]
    }
    mock_maintenance_dict = mock.Mock(return_value=mock_schedule)
    mock_get_maintenance_schedule.return_value = mock.Mock(json=mock_maintenance_dict)
    mock_datetime_to_nanoseconds.return_value = 7
    ret = get_hosts_past_maintenance_start()
    assert ret == ['host2']

    mock_schedule = {}
    mock_maintenance_dict = mock.Mock(return_value=mock_schedule)
    mock_get_maintenance_schedule.return_value = mock.Mock(json=mock_maintenance_dict)
    ret = get_hosts_past_maintenance_start()
    assert ret == []