fix(async): Fix critical misbehavior

ansible · Jul 26, 2023 · 78e9d95 · 78e9d95
1 parent 70ff952
commit 78e9d95
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tacacs_plus/async_client.py b/tacacs_plus/async_client.py
@@ -112,8 +112,8 @@ async def flow_control(self, reader=None, writer=None):
             if not existing:
                 # if reader/writer were not provided -
                 # we need to properly close socket and writer
-                writer.close()
-                await writer.wait_closed()
+                yielded_writer.close()
+                await yielded_writer.wait_closed()
                 sock.close()
 
     async def send(self, body, req_type, seq_no=1, reader=None, writer=None):