Add welcome message and PING handling

1 files changed, 13 insertions, 0 deletions

diff --git a/testserver.py b/testserver.py
index 1710a4a..894f832 100644
--- a/testserver.py
+++ b/testserver.py
@@ -9,6 +9,7 @@ logging.basicConfig(level = logging.INFO, format = '{asctime} {levelname} {messa
 class FakeIRCServerProtocol(asyncio.Protocol):
 	def __init__(self, loop = None):
 		self.loop = loop or asyncio.get_event_loop()
+		self.buffer = b''
 
 	def send(self, data):
 		logging.info(f'Send: {data!r}')
@@ -18,6 +19,7 @@ class FakeIRCServerProtocol(asyncio.Protocol):
 		logging.info(f'Connection from {transport.get_extra_info("peername")}')
 		self.transport = transport
 		self.connected = True
+		self.send(b'001 :Hello there!\r\n')
 		asyncio.create_task(self.pingloop())
 
 	async def pingloop(self):
@@ -27,6 +29,17 @@ class FakeIRCServerProtocol(asyncio.Protocol):
 
 	def data_received(self, data):
 		logging.info(f'Data received: {data!r}')
+		messages = data.split(b'\r\n')
+		if self.buffer:
+			self.message_received(self.buffer + messages[0])
+			messages = messages[1:]
+		for message in messages[:-1]:
+			self.message_received(message)
+		self.buffer = messages[-1]
+
+	def message_received(self, message):
+		if message.startswith(b'PING '):
+			self.send(b'PONG ' + message[5:] + b'\r\n')
 
 	def connection_lost(self, exc):
 		logging.info(f'Connection to {self.transport.get_extra_info("peername")} lost')