explicitly passing in a loop for connection removal

Signed-off-by: Jesse Jaggars <jjaggars@redhat.com>

Author: jhjaggars

.gitignore

@@ -111,3 +111,4 @@ venv.bak/
 
 # mypy
 .mypy_cache/
+graph_*.dot

receptor/receptor.py

@@ -112,10 +112,13 @@ def update_connections(self, protocol_obj, id_=None):
     def add_connection(self, protocol_obj):
         self.update_connections(protocol_obj)
 
-    def remove_connection(self, protocol_obj, id_=None):
+    def remove_connection(self, protocol_obj, id_=None, loop=None):
         if id_ is None:
             id_ = protocol_obj.id
 
+        if loop is None:
+            loop = protocol_obj.loop
+
         notify_connections = []
         for connection_node in self.connections:
             if protocol_obj in self.connections[connection_node]:
@@ -126,7 +129,7 @@ def remove_connection(self, protocol_obj, id_=None):
                 self.router.debug_router()
                 self.update_connection_manifest(connection_node)
             notify_connections += self.connections[connection_node]
-        protocol_obj.loop.create_task(self.send_route_advertisement(self.router.get_edges()))
+        loop.create_task(self.send_route_advertisement(self.router.get_edges()))
 
     async def shutdown_handler(self):
         while True:

receptor/ws.py

@@ -53,7 +53,7 @@ def register(self, ws):
         self.receptor.update_connections(ws, id_=self.remote_id)
 
     def unregister(self, ws):
-        self.receptor.remove_connection(ws, id_=self.remote_id)
+        self.receptor.remove_connection(ws, id_=self.remote_id, loop=self.loop)
         self._cancel(self.read_task)
         self._cancel(self.handle_task)
         self._cancel(self.write_task)