test

kgiusti · Feb 4, 2025 · c3c82f4 · c3c82f4
1 parent 72ff46d
commit c3c82f4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/system_tests_websockets.py b/tests/system_tests_websockets.py
@@ -51,13 +51,14 @@ def test_stopping_broker_while_websocket_is_connected_does_not_crash(self):
 
         Previously, this was broken in DISPATCH-857, DISPATCH-945, DISPATCH-2098"""
 
+        from websockets.asyncio.client import connect
         async def run():
             uri = f"ws://localhost:{self.http_port}"
-            async with websockets.connect(uri, subprotocols=['amqp']) as ws:
+            async with connect(uri, subprotocols=['amqp']) as ws:
                 self.router.terminate()
                 self.router.wait()
 
-        asyncio.get_event_loop().run_until_complete(run())
+        asyncio.run(run())
 
 
 if __name__ == '__main__':