Fix TcpProxy for python 3.6

aio-libs · Dec 6, 2020 · ce9aaa3 · ce9aaa3
1 parent 29cc1ed
commit ce9aaa3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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(