Make close behaviour consistent across new protocols and transports (#660)

This commit is contained in:
Steven B
2024-01-20 12:35:05 +00:00
committed by GitHub
parent e94cd118a4
commit 49cfef087c
4 changed files with 46 additions and 9 deletions

View File

@@ -265,10 +265,12 @@ class AesTransport(BaseTransport):
return await self.send_secure_passthrough(request)
async def close(self) -> None:
"""Close the transport."""
"""Mark the handshake and login as not done.
Since we likely lost the connection.
"""
self._handshake_done = False
self._login_token = None
await self._http_client.close()
class AesEncyptionSession: