tests: add System.del_address

Add a helper for removing a locally-assigned address.

Signed-off-by: Jeremy Kerr <[email protected]>

Author: jk-ozlabs

tests/conftest.py

@@ -119,6 +119,14 @@ async def add_address(self, address):
         if self.nl:
             await self.nl.notify_newaddr(address)
 
+    async def del_address(self, address):
+        addr = self.lookup_address(address.iface, address.eid)
+        if not addr:
+            raise NetlinkError(errno.ENOENT)
+        self.addresses.remove(addr)
+        if self.nl:
+            await self.nl.notify_deladdr(address)
+
     async def add_neighbour(self, neigh):
         if self.lookup_neighbour(neigh.iface, neigh.eid):
             raise NetlinkError(errno.EEXIST)
@@ -166,6 +174,12 @@ def lookup_neighbour(self, iface, eid):
                 return neighbour
         return None
 
+    def lookup_address(self, iface, eid):
+        for address in self.addresses:
+            if address.iface == iface and address.eid == eid:
+                return address
+        return None
+
     def find_endpoint(self, addr):
         iface = None
         lladdr = None
@@ -725,14 +739,20 @@ async def _handle_getaddr(self, msg):
 
         await self._send_msg(buf)
 
-    async def notify_newaddr(self, addr):
-        msg = self._create_msg(ifaddrmsg_mctp, rtnl.RTM_NEWADDR, 0)
+    async def _notify_addr(self, addr, typ):
+        msg = self._create_msg(ifaddrmsg_mctp, typ, 0)
         self._format_addr(msg, addr)
         buf = bytearray()
         msg.encode()
         buf.extend(msg.data)
         await self._send_msg(buf)
 
+    async def notify_newaddr(self, addr):
+        await self._notify_addr(addr, rtnl.RTM_NEWADDR)
+
+    async def notify_deladdr(self, addr):
+        await self._notify_addr(addr, rtnl.RTM_DELADDR)
+
     def _format_neigh(self, msg, neigh):
         msg['ifindex'] = neigh.iface.ifindex
         msg['attrs'] = [