From 19664cd9cfacbfa36b819ca2d9fdb29bae1cc3db Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Thu, 25 Apr 2019 09:56:30 +0200
Subject: [PATCH] Call self._writer.wait_closed() on disconnect

https://docs.python.org/3/library/asyncio-stream.html#asyncio.StreamWriter.wait_closed
---
 telethon/network/connection/connection.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/telethon/network/connection/connection.py b/telethon/network/connection/connection.py
index 9a3f6c3e..3ec973b6 100644
--- a/telethon/network/connection/connection.py
+++ b/telethon/network/connection/connection.py
@@ -2,6 +2,7 @@ import abc
 import asyncio
 import socket
 import ssl as ssl_mod
+import sys
 
 from ...errors import InvalidChecksumError
 from ... import helpers
@@ -108,6 +109,8 @@ class Connection(abc.ABC):
 
         if self._writer:
             self._writer.close()
+            if sys.version_info >= (3, 7):
+                await self._writer.wait_closed()
 
     def send(self, data):
         """