From ce9aaa3fcac38b7d2b082ead7469ac49c9455dba Mon Sep 17 00:00:00 2001
From: Yury Pliner <yury.pliner@gmail.com>
Date: Sat, 5 Dec 2020 16:35:00 +0500
Subject: [PATCH] Fix TcpProxy for python 3.6

---
 tests/conftest.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index f3555a36..72b98303 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -434,7 +434,8 @@ async def disconnect(self):
         while self.connections:
             writer = self.connections.pop()
             writer.close()
-            await writer.wait_closed()
+            if hasattr(writer, "wait_closed"):
+                await writer.wait_closed()
 
     @staticmethod
     async def _pipe(