From 044e6e8a85055fd3d63098118b3d7af409c577b7 Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Tue, 26 Jun 2018 10:43:38 +0200
Subject: [PATCH] Await updates task to finish upon disconnect

---
 telethon/client/telegrambaseclient.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/telethon/client/telegrambaseclient.py b/telethon/client/telegrambaseclient.py
index c083af27..5e9fe4f2 100644
--- a/telethon/client/telegrambaseclient.py
+++ b/telethon/client/telegrambaseclient.py
@@ -292,6 +292,9 @@ class TelegramBaseClient(abc.ABC):
         Disconnects from Telegram.
         """
         await self._sender.disconnect()
+        if self._updates_handle:
+            await self._updates_handle
+
         self.session.close()
 
     async def _switch_dc(self, new_dc):
@@ -410,11 +413,10 @@ class TelegramBaseClient(abc.ABC):
         """
         raise NotImplementedError
 
-    # Let people use client.invoke(SomeRequest()) instead client(...)
     async def invoke(self, *args, **kwargs):
         warnings.warn('client.invoke(...) is deprecated, '
                       'use client(...) instead')
-        return await self(*args, **kwargs)
+        self(*args, **kwargs)
 
     @abc.abstractmethod
     def _handle_update(self, update):