From 0e57ad9633dc606e6eb111832bfa01e3149ecacf Mon Sep 17 00:00:00 2001
From: Luna <git@l4.pm>
Date: Tue, 26 Mar 2019 23:50:10 -0300
Subject: [PATCH] test_websocket: use _close on test_ready

---
 tests/test_websocket.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_websocket.py b/tests/test_websocket.py
index 76dac73..50216b5 100644
--- a/tests/test_websocket.py
+++ b/tests/test_websocket.py
@@ -72,7 +72,7 @@ async def test_gw(test_cli):
     assert isinstance(hello['d']['heartbeat_interval'], int)
     assert isinstance(hello['d']['_trace'], list)
 
-    await conn.close(1000, 'test end')
+    await _close(conn)
 
 
 @pytest.mark.asyncio
@@ -94,9 +94,10 @@ async def test_ready(test_cli):
     try:
         await _json(conn)
         assert True
-        await conn.close(1000, 'test end')
     except (Exception, websockets.ConnectionClosed):
         assert False
+    finally:
+        await _close(conn)
 
 
 @pytest.mark.asyncio