Fix tests

This commit is contained in:
grossmj committed 2024-11-17 15:00:13 +10:00
1 parent 31a2cb998d
commit b37db57bb9
3 files changed
+6 -8

No files matched your search

+3 -4
View File
@@ -32,7 +32,6 @@ class Notification:
self._controller = controller
self._project_listeners = {}
self._controller_listeners = set()
self._loop = asyncio.get_event_loop()
@contextmanager
def project_queue(self, project_id):
@@ -74,7 +73,7 @@ class Notification:
"""
for controller_listener in self._controller_listeners:
self._loop.call_soon_threadsafe(controller_listener.put_nowait, (action, event, {}))
asyncio.get_event_loop().call_soon_threadsafe(controller_listener.put_nowait, (action, event, {}))
def project_has_listeners(self, project_id):
"""
@@ -135,7 +134,7 @@ class Notification:
except KeyError:
return
for listener in project_listeners:
self._loop.call_soon_threadsafe(listener.put_nowait, (action, event, {}))
asyncio.get_event_loop().call_soon_threadsafe(listener.put_nowait, (action, event, {}))
def _send_event_to_all_projects(self, action, event):
"""
@@ -147,4 +146,4 @@ class Notification:
"""
for project_listeners in self._project_listeners.values():
for listener in project_listeners:
self._loop.call_soon_threadsafe(listener.put_nowait, (action, event, {}))
asyncio.get_event_loop().call_soon_threadsafe(listener.put_nowait, (action, event, {}))