Try fixing multiprocess test

permitio · Apr 24, 2024 · efe3bb9 · efe3bb9
1 parent 4c83f56
commit efe3bb9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/multiprocess_test.py b/tests/multiprocess_test.py
@@ -3,6 +3,7 @@
         Publishing-Client -> PubSubServer -> Subscribing->Client
 
 """
+
 import os
 import sys
 import pytest
@@ -35,11 +36,9 @@
 
 def setup_server():
     app = FastAPI()
-    router = APIRouter()
     # PubSub websocket endpoint
     endpoint = PubSubEndpoint()
-    endpoint.register_route(router, path="/pubsub")
-    app.include_router(router)
+    endpoint.register_route(app, path="/pubsub")
     uvicorn.run(app, port=PORT)