Update test.py

test.py

@@ -223,7 +223,8 @@ async def test_websocket(websocket: WebSocket):
         access_token = code
         messages=get_messages(access_token)
         print(len(messages))
-
+        await websocket.send_json({"total_messages":len(messages)})
+        await websocket.send_text("CompletedSendingTotalMessagesLength")
         attachments = []
         prev_data=""
         data_new={}