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 == []
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 == []
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 == []