From 7e93d632700e0013772e5f45a3028c2512460e6f Mon Sep 17 00:00:00 2001 From: Kashif Khan Date: Mon, 17 Oct 2022 08:21:17 -0500 Subject: [PATCH] fix for async socket --- .../azure/eventhub/_pyamqp/aio/_transport_async.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/sdk/eventhub/azure-eventhub/azure/eventhub/_pyamqp/aio/_transport_async.py b/sdk/eventhub/azure-eventhub/azure/eventhub/_pyamqp/aio/_transport_async.py index 0fb9e0268b52..f295b205a78b 100644 --- a/sdk/eventhub/azure-eventhub/azure/eventhub/_pyamqp/aio/_transport_async.py +++ b/sdk/eventhub/azure-eventhub/azure/eventhub/_pyamqp/aio/_transport_async.py @@ -380,6 +380,7 @@ async def _read( async def _write(self, s): """Write a string out to the SSL socket fully.""" self.writer.write(s) + await self.writer.drain() async def close(self): if self.writer is not None: @@ -387,6 +388,7 @@ async def close(self): # see issue: https://github.com/encode/httpx/issues/914 self.writer.transport.abort() self.writer.close() + await self.writer.wait_closed() self.writer, self.reader = None, None self.sock = None self.connected = False