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