Increase async server buffer size

Author: robsdedude

testkitbackend/__main__.py

@@ -34,15 +34,15 @@ def sync_main():
 
 
 def async_main():
-    server = AsyncServer(("0.0.0.0", 9876))
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(server.start())
-
-    try:
-        loop.run_forever()
-    finally:
-        server.stop()
-        loop.stop()
+    async def main():
+        server = AsyncServer(("0.0.0.0", 9876))
+        await server.start()
+        try:
+            await server.serve_forever()
+        finally:
+            server.stop()
+
+    asyncio.run(main())
 
 
 if __name__ == "__main__":

testkitbackend/server.py

@@ -53,9 +53,15 @@ async def _handler(reader, writer):
 
     async def start(self):
         self._server = await asyncio.start_server(
-            self._handler, host=self._address[0], port=self._address[1]
+            self._handler, host=self._address[0], port=self._address[1],
+            limit=float("inf")  # this is dirty but works (for now)
         )
 
+    async def serve_forever(self):
+        if not self._server:
+            raise RuntimeError("Server not started")
+        await self._server.serve_forever()
+
     def stop(self):
         if not self._server:
             raise RuntimeError("Try starting the server before stopping it ;)")