From a1b4969ccfa9c17fce80ad638c16c55972c6c474 Mon Sep 17 00:00:00 2001 From: JustAnotherArchivist Date: Mon, 16 Dec 2019 00:24:26 +0000 Subject: [PATCH] Add welcome message and PING handling --- testserver.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) 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')