Update test.py

test.py

@@ -176,8 +176,9 @@ async def test_websocket(websocket: WebSocket):
     # code = data.get("message")
     # print(code)
     # code = raw_body.decode()
+    await event_generator(data)
     
-    def get_messages():
+    def get_messages(code:str):
         print()
         # print(code)
         access_token = code
@@ -225,10 +226,10 @@ async def test_websocket(websocket: WebSocket):
         print(filtered_data_list)
         return filtered_data_list
     
-    async def event_generator():
+    async def event_generator(code:str):
         print(code)
         access_token = code
-        messages=get_messages()
+        messages=get_messages(access_token)
         print(len(messages))
         await websocket.send(json.dumps({"total_messages": len(messages)}))