Update test.py

test.py

@@ -317,7 +317,7 @@ async def test_websocket(websocket: WebSocket):
 
     await event_generator(data)
     logging.info("Closing connection")
-    # await websocket.close()
+    await websocket.close()
     
 @app.get("/token")
 async def get_token(token: str = Depends(oauth2_scheme)):