From 07c66d121aab15edc9c01ebab4149c7e69f444ab Mon Sep 17 00:00:00 2001 From: Luna Date: Tue, 26 Mar 2019 23:47:52 -0300 Subject: [PATCH] test_websocket: add a _close helper --- tests/test_websocket.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/test_websocket.py b/tests/test_websocket.py index 757e07e..76dac73 100644 --- a/tests/test_websocket.py +++ b/tests/test_websocket.py @@ -42,6 +42,10 @@ async def _json_send_op(conn, opcode, data=None): }) +async def _close(conn): + await conn.close(1000, 'test end') + + async def get_gw(test_cli) -> str: """Get the Gateway URL.""" gw_resp = await test_cli.get('/api/v6/gateway') @@ -128,7 +132,7 @@ async def test_ready_fields(test_cli): assert isinstance(data['guilds'], list) assert isinstance(data['session_id'], str) - await conn.close(1000, 'test end') + await _close(conn) @pytest.mark.asyncio @@ -157,3 +161,5 @@ async def test_heartbeat(test_cli): recv = await _json(conn) assert isinstance(recv, dict) assert recv['op'] == OP.HEARTBEAT_ACK + + await _close(conn)