def _update_hosts(server, remove, add, modify):
"""Update the hosts using the OMAPI."""
omapi_client = OmapiClient(server.omapi_key, server.ipv6)
try:
for host in remove:
omapi_client.del_host(host["mac"])
except OmapiError as e:
raise CannotRemoveHostMap(str(e))
try:
for host in add:
omapi_client.add_host(host["mac"], host["ip"])
except OmapiError as e:
raise CannotCreateHostMap(str(e))
try:
for host in modify:
omapi_client.update_host(host["mac"], host["ip"])
except OmapiError as e:
raise CannotModifyHostMap(str(e))
def test_remove_host(self):
cli = OmapiClient("shared-key")
cli.del_host("aa:bb:cc:dd:ee:ff")
self.mock_omapi_cli.del_host("aa:bb:cc:dd:ee:ff")