Update test.py

test.py

@@ -315,6 +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()
     
 @app.get("/token")
 async def get_token(token: str = Depends(oauth2_scheme)):