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