Update test.py

test.py

@@ -315,7 +315,7 @@ async def test_websocket(websocket: WebSocket):
         await websocket.send_json({"message":"Successfully sent all the Data !"})
 
     await event_generator(data)
-    await websocket.close()
+    # await websocket.close()
     
 @app.get("/token")
 async def get_token(token: str = Depends(oauth2_scheme)):