Fix exceptions during connections

Author: srittau

NEWS.md

@@ -1,3 +1,11 @@
+Changes in FakeSMTPd 0.2.1
+==========================
+
+Bug fixes
+---------
+
+* Log exceptions raised during a connection, instead of aborting.
+
 Changes in FakeSMTPd 0.2.0
 ==========================
 

fakesmtpd/connection.py

@@ -38,6 +38,14 @@ def __init__(self, reader: StreamReader, writer: StreamWriter,
 
     async def handle(self) -> None:
         logging.info("connection opened")
+        try:
+            await self._handle_connection()
+        except Exception as exc:
+            logging.warning(str(exc))
+        finally:
+            logging.info("connection closed")
+
+    async def _handle_connection(self) -> None:
         self._write_reply(SMTPStatus.SERVICE_READY,
                           "{} FakeSMTPd Service ready".format(getfqdn()))
         while not self.reader.at_eof():
@@ -58,7 +66,6 @@ async def handle(self) -> None:
                 elif code == SMTPStatus.SERVICE_CLOSING:
                     break
         self.writer.close()
-        logging.info("connection closed")
 
     def _handle_command_line(self, line: str) -> SMTPStatus:
         logging.debug(f"received command: {line}")