Update test.py

test.py

@@ -5,7 +5,7 @@ from jose import jwt
 import webbrowser
 import base64
 import logging
-
+import time
 
 app = FastAPI()
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
@@ -243,8 +243,9 @@ async def test_websocket(websocket: WebSocket):
         messages=get_messages(access_token)
         print(len(messages))
         await websocket.send_json({"total_messages": len(messages)})
+        time.sleep(0.25)
         await websocket.send_json({"message":"Successfully sent all the Data !"})
-
+        time.sleep(0.25)
 
         attachments = []
         prev_data=""
@@ -315,7 +316,8 @@ async def test_websocket(websocket: WebSocket):
         await websocket.send_json({"message":"Successfully sent all the Data !"})
 
     await event_generator(data)
-    # await websocket.close()
+    logging.info("Closing connection")
+    await websocket.close()
     
 @app.get("/token")
 async def get_token(token: str = Depends(oauth2_scheme)):