From 733a537f58db10031b6ba8aa3f6701a3110b6b94 Mon Sep 17 00:00:00 2001
From: binaryDiv <binarydiv@gmail.com>
Date: Sat, 24 Nov 2018 02:10:33 +0100
Subject: [PATCH] Server: send test messages after init

---
 server/chatserver.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/server/chatserver.py b/server/chatserver.py
index f4e35e4..3456fd9 100755
--- a/server/chatserver.py
+++ b/server/chatserver.py
@@ -32,8 +32,13 @@ async def handle_client_init(websocket, message):
     """Handle client 'init' message."""
 
     print(f"< init: chat_id='{message['chat_id']}', nickname='{message['nickname']}'")
+    await send_client_init_response(websocket)
+    await send_client_previous_messages(websocket)
+
+
+async def send_client_init_response(websocket):
+    """Send an init response message to a newly connected client."""
 
-    # Send response to client
     response = json.dumps({
         'type': 'init'
     })
@@ -41,6 +46,20 @@ async def handle_client_init(websocket, message):
     await websocket.send(response)
 
 
+async def send_client_previous_messages(websocket):
+    """Sends previously written chat messages to a newly connected client."""
+
+    # For now: send test messages
+    for i in range(1, 4):
+        testmsg = json.dumps({
+            'type': 'message',
+            'from': 'Alice',
+            'text': f"Message number {i}, hello!"
+        })
+        print(f"> {testmsg}")
+        await websocket.send(testmsg)
+
+
 async def client_handler(websocket, path):
     """Handle client connection."""