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