Merge pull request #110 from robertsmd/master

fix race condition on login. awaits login response

Author: areed1192

td/stream.py

@@ -636,7 +636,12 @@ async def _connect(self) -> websockets.WebSocketClientProtocol:
         # If we are connected then login.
         if is_connected:
             await self._send_message(login_request)
-            return self.connection
+            while True:
+                response = await self._receive_message(return_value=True)
+                responses = response.get('response')
+                for r in responses:
+                    if r.get('service') == 'ADMIN' and r.get('command') == 'LOGIN':
+                        return self.connection
 
     async def _check_connection(self) -> bool:
         """Determines if we have an active connection