def test_remove(self):
pool = ZerobootPool(
name="test", data={"zerobootHosts": ["host1", "host2", "host3"]})
pool._validate_host = MagicMock()
# remove host
pool.remove("host3")
assert len(pool.data.get("zerobootHosts")) == 2
# empty pool
pool.remove("host1")
assert len(pool.data.get("zerobootHosts")) == 1
pool.remove("host2")
assert len(pool.data.get("zerobootHosts")) == 0
def test_remove_nonexisting_host(self):
pool = ZerobootPool(
name="test", data={"zerobootHosts": ["host1", "host2", "host3"]})
pool._validate_host = MagicMock()
# remove non existing host
pool.remove("ghost_host")
assert len(pool.data.get("zerobootHosts")) == 3
# remove a host and try get a ghost again
pool.remove("host3")
pool.remove("ghost_host")
assert len(pool.data.get("zerobootHosts")) == 2
# add a host and try get a ghost again
pool.add("host3")
pool.remove("ghost_host")
assert len(pool.data.get("zerobootHosts")) == 3
# empty list and try get a ghost again
pool.remove("host1")
pool.remove("host2")
pool.remove("host3")
pool.remove("ghost_host")
assert len(pool.data.get("zerobootHosts")) == 0