Game Coordinator now mostly (~90%) functional

1 files changed, 12 insertions, 3 deletions

diff --git a/resources/external/websocket_test.py b/resources/external/websocket_test.py
index 9c29fcf..25ebcff 100644
--- a/resources/external/websocket_test.py
+++ b/resources/external/websocket_test.py
@@ -3,16 +3,25 @@
 """Echo server using the asyncio API."""
 
 import asyncio
+import sys
 from websockets.asyncio.server import serve
 
 DEFAULT_PORT = 8181
 DEFAULT_IP = "127.0.0.1"
 
+async def delayed_func():
+    await asyncio.sleep(5)
+    print("after delay")
 
 async def echo(websocket):
     print("client connected")
-    async for message in websocket:
-        await websocket.send(message)
+    try:
+        async for message in websocket:
+            if message == "Hello world!":
+                asyncio.run(delayed_func())
+            await websocket.send(message)
+    finally:
+        print("client DC")
 
 
 async def main(ip, port):
@@ -35,6 +44,6 @@ if __name__ == "__main__":
                 PORT = int(args[arg_i])
             except:
                 print("Port must be an integer.")
-                return
+                quit()
 
     asyncio.run(main(IP, PORT))
\ No newline at end of file