Misplacement of writer.drain()

src/server.py

@@ -105,12 +105,12 @@ class Server:
         """
         client_hello = await reader.readline()
         peer_is_sender = client_hello.decode('utf-8') == 'sender\n'
-        await writer.drain()
         if peer_is_sender:
             self._working = True
             logging.info("Sender is connecting...")
             # Send start signal to client
             writer.write("Start!\n".encode('utf-8'))
+            await writer.drain()
             await self.run_reader(reader=reader)
             logging.info("Incoming transmission ended")
         else:
@@ -119,6 +119,7 @@ class Server:
                 await asyncio.sleep(.5)
             # Send start signal to client
             writer.write("Start!\n".encode('utf-8'))
+            await writer.drain()
             await self.run_writer(writer=writer)
             logging.info("Outgoing transmission ended")
             self._working = False