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') == []
def test_get_hosts_with_state_down(mock_get_maintenance_status,):
fake_status = {
"down_machines": [
{"id": {"hostname": "fake-host1.fakesite.something", "ip": "0.0.0.0"}},
{"id": {"hostname": "fake-host2.fakesite.something", "ip": "0.0.0.1"}},
]
}
mock_get_maintenance_status.return_value = fake_status
expected = sorted(["fake-host1.fakesite.something", "fake-host2.fakesite.something"])
assert sorted(get_hosts_with_state(state="down_machines")) == expected
def test_get_hosts_with_state_down(mock_get_maintenance_status, ):
fake_status = {
"down_machines": [{
"id": {
"hostname": "fake-host1.fakesite.something",
"ip": "0.0.0.0"
}
}, {
"id": {
"hostname": "fake-host2.fakesite.something",
"ip": "0.0.0.1"
}
}]
}
mock_get_maintenance_status.return_value = fake_status
expected = sorted(
['fake-host1.fakesite.something', 'fake-host2.fakesite.something'])
assert sorted(get_hosts_with_state(state='down_machines')) == expected
def test_get_hosts_with_state_draining(
mock_get_maintenance_status,
):
fake_status = {
"draining_machines": [
{
"id": {
"hostname": "fake-host1.fakesite.something",
"ip": "0.0.0.0"
}
},
{
"id": {
"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
def test_get_hosts_with_state_none(mock_get_maintenance_status,):
mock_get_maintenance_status.return_value = {}
assert get_hosts_with_state(state="fake_state") == []
def test_get_hosts_with_state_none(mock_get_maintenance_status, ):
mock_get_maintenance_status.return_value = {}
assert get_hosts_with_state(state='fake_state') == []
